https://github.com/JuliaLang/julia
Tip revision: 68e9e56bd74ab58d895aee5b20089fae01bd0c37 authored by Tim Holy on 15 August 2023, 12:05:38 UTC
Add to build, .gitignore, NEWS
Add to build, .gitignore, NEWS
Tip revision: 68e9e56
method.c
// This file is a part of Julia. License is MIT: https://julialang.org/license
/*
Defining and adding methods
*/
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include "julia.h"
#include "julia_internal.h"
#include "julia_assert.h"
#ifdef __cplusplus
extern "C" {
#endif
extern jl_value_t *jl_builtin_getfield;
extern jl_value_t *jl_builtin_tuple;
jl_methtable_t *jl_kwcall_mt;
jl_method_t *jl_opaque_closure_method;
jl_method_t *jl_make_opaque_closure_method(jl_module_t *module, jl_value_t *name,
int nargs, jl_value_t *functionloc, jl_code_info_t *ci, int isva);
static void check_c_types(const char *where, jl_value_t *rt, jl_value_t *at)
{
if (jl_is_svec(rt))
jl_errorf("%s: missing return type", where);
JL_TYPECHKS(where, type, rt);
if (!jl_type_mappable_to_c(rt))
jl_errorf("%s: return type doesn't correspond to a C type", where);
JL_TYPECHKS(where, simplevector, at);
int i, l = jl_svec_len(at);
for (i = 0; i < l; i++) {
jl_value_t *ati = jl_svecref(at, i);
if (jl_is_vararg(ati))
jl_errorf("%s: Vararg not allowed for argument list", where);
JL_TYPECHKS(where, type, ati);
if (!jl_type_mappable_to_c(ati))
jl_errorf("%s: argument %d type doesn't correspond to a C type", where, i + 1);
}
}
// Resolve references to non-locally-defined variables to become references to global
// variables in `module` (unless the rvalue is one of the type parameters in `sparam_vals`).
static jl_value_t *resolve_globals(jl_value_t *expr, jl_module_t *module, jl_svec_t *sparam_vals,
int binding_effects, int eager_resolve)
{
if (jl_is_symbol(expr)) {
if (module == NULL)
return expr;
return jl_module_globalref(module, (jl_sym_t*)expr);
}
else if (jl_is_returnnode(expr)) {
jl_value_t *retval = jl_returnnode_value(expr);
if (retval) {
jl_value_t *val = resolve_globals(retval, module, sparam_vals, binding_effects, eager_resolve);
if (val != retval) {
JL_GC_PUSH1(&val);
expr = jl_new_struct(jl_returnnode_type, val);
JL_GC_POP();
}
}
return expr;
}
else if (jl_is_gotoifnot(expr)) {
jl_value_t *cond = resolve_globals(jl_gotoifnot_cond(expr), module, sparam_vals, binding_effects, eager_resolve);
if (cond != jl_gotoifnot_cond(expr)) {
intptr_t label = jl_gotoifnot_label(expr);
JL_GC_PUSH1(&cond);
expr = jl_new_struct_uninit(jl_gotoifnot_type);
set_nth_field(jl_gotoifnot_type, expr, 0, cond, 0);
jl_gotoifnot_label(expr) = label;
JL_GC_POP();
}
return expr;
}
else if (jl_is_expr(expr)) {
jl_expr_t *e = (jl_expr_t*)expr;
if (e->head == jl_global_sym && binding_effects) {
// execute the side-effects of "global x" decl immediately:
// creates uninitialized mutable binding in module for each global
jl_eval_global_expr(module, e, 1);
expr = jl_nothing;
}
if (jl_is_toplevel_only_expr(expr) || e->head == jl_const_sym ||
e->head == jl_coverageeffect_sym || e->head == jl_copyast_sym ||
e->head == jl_quote_sym || e->head == jl_inert_sym ||
e->head == jl_meta_sym || e->head == jl_inbounds_sym ||
e->head == jl_boundscheck_sym || e->head == jl_loopinfo_sym ||
e->head == jl_aliasscope_sym || e->head == jl_popaliasscope_sym ||
e->head == jl_inline_sym || e->head == jl_noinline_sym) {
// ignore these
}
else {
size_t i = 0, nargs = jl_array_len(e->args);
if (e->head == jl_opaque_closure_method_sym) {
if (nargs != 5) {
jl_error("opaque_closure_method: invalid syntax");
}
jl_value_t *name = jl_exprarg(e, 0);
jl_value_t *oc_nargs = jl_exprarg(e, 1);
int isva = jl_exprarg(e, 2) == jl_true;
jl_value_t *functionloc = jl_exprarg(e, 3);
jl_value_t *ci = jl_exprarg(e, 4);
if (!jl_is_code_info(ci)) {
jl_error("opaque_closure_method: lambda should be a CodeInfo");
} else if (!jl_is_long(oc_nargs)) {
jl_type_error("opaque_closure_method", (jl_value_t*)jl_long_type, oc_nargs);
}
jl_method_t *m = jl_make_opaque_closure_method(module, name, jl_unbox_long(oc_nargs), functionloc, (jl_code_info_t*)ci, isva);
return (jl_value_t*)m;
}
if (e->head == jl_cfunction_sym) {
JL_NARGS(cfunction method definition, 5, 5); // (type, func, rt, at, cc)
jl_value_t *typ = jl_exprarg(e, 0);
if (!jl_is_type(typ))
jl_error("first parameter to :cfunction must be a type");
if (typ == (jl_value_t*)jl_voidpointer_type) {
jl_value_t *a = jl_exprarg(e, 1);
JL_TYPECHK(cfunction method definition, quotenode, a);
*(jl_value_t**)a = jl_toplevel_eval(module, *(jl_value_t**)a);
jl_gc_wb(a, *(jl_value_t**)a);
}
jl_value_t *rt = jl_exprarg(e, 2);
jl_value_t *at = jl_exprarg(e, 3);
if (!jl_is_type(rt)) {
JL_TRY {
rt = jl_interpret_toplevel_expr_in(module, rt, NULL, sparam_vals);
}
JL_CATCH {
if (jl_typetagis(jl_current_exception(), jl_errorexception_type))
jl_error("could not evaluate cfunction return type (it might depend on a local variable)");
else
jl_rethrow();
}
jl_exprargset(e, 2, rt);
}
if (!jl_is_svec(at)) {
JL_TRY {
at = jl_interpret_toplevel_expr_in(module, at, NULL, sparam_vals);
}
JL_CATCH {
if (jl_typetagis(jl_current_exception(), jl_errorexception_type))
jl_error("could not evaluate cfunction argument type (it might depend on a local variable)");
else
jl_rethrow();
}
jl_exprargset(e, 3, at);
}
check_c_types("cfunction method definition", rt, at);
JL_TYPECHK(cfunction method definition, quotenode, jl_exprarg(e, 4));
JL_TYPECHK(cfunction method definition, symbol, *(jl_value_t**)jl_exprarg(e, 4));
return expr;
}
if (e->head == jl_foreigncall_sym) {
JL_NARGSV(ccall method definition, 5); // (fptr, rt, at, nreq, (cc, effects))
jl_value_t *rt = jl_exprarg(e, 1);
jl_value_t *at = jl_exprarg(e, 2);
if (!jl_is_type(rt)) {
JL_TRY {
rt = jl_interpret_toplevel_expr_in(module, rt, NULL, sparam_vals);
}
JL_CATCH {
if (jl_typetagis(jl_current_exception(), jl_errorexception_type))
jl_error("could not evaluate ccall return type (it might depend on a local variable)");
else
jl_rethrow();
}
jl_exprargset(e, 1, rt);
}
if (!jl_is_svec(at)) {
JL_TRY {
at = jl_interpret_toplevel_expr_in(module, at, NULL, sparam_vals);
}
JL_CATCH {
if (jl_typetagis(jl_current_exception(), jl_errorexception_type))
jl_error("could not evaluate ccall argument type (it might depend on a local variable)");
else
jl_rethrow();
}
jl_exprargset(e, 2, at);
}
check_c_types("ccall method definition", rt, at);
JL_TYPECHK(ccall method definition, long, jl_exprarg(e, 3));
JL_TYPECHK(ccall method definition, quotenode, jl_exprarg(e, 4));
jl_value_t *cc = jl_quotenode_value(jl_exprarg(e, 4));
if (!jl_is_symbol(cc)) {
JL_TYPECHK(ccall method definition, tuple, cc);
if (jl_nfields(cc) != 2) {
jl_error("In ccall calling convention, expected two argument tuple or symbol.");
}
JL_TYPECHK(ccall method definition, symbol, jl_get_nth_field(cc, 0));
JL_TYPECHK(ccall method definition, uint8, jl_get_nth_field(cc, 1));
}
jl_exprargset(e, 0, resolve_globals(jl_exprarg(e, 0), module, sparam_vals, binding_effects, 1));
i++;
}
if (e->head == jl_method_sym || e->head == jl_module_sym) {
i++;
}
for (; i < nargs; i++) {
// TODO: this should be making a copy, not mutating the source
jl_exprargset(e, i, resolve_globals(jl_exprarg(e, i), module, sparam_vals, binding_effects, eager_resolve));
}
if (e->head == jl_call_sym && jl_expr_nargs(e) == 3 &&
jl_is_globalref(jl_exprarg(e, 0)) &&
jl_is_globalref(jl_exprarg(e, 1)) &&
jl_is_quotenode(jl_exprarg(e, 2))) {
// replace module_expr.sym with GlobalRef(module, sym)
// for expressions pattern-matching to `getproperty(module_expr, :sym)` in a top-module
// (this is expected to help inference performance)
// TODO: this was broken by linear-IR
jl_value_t *s = jl_fieldref(jl_exprarg(e, 2), 0);
jl_value_t *me = jl_exprarg(e, 1);
jl_value_t *fe = jl_exprarg(e, 0);
jl_module_t *fe_mod = jl_globalref_mod(fe);
jl_sym_t *fe_sym = jl_globalref_name(fe);
jl_module_t *me_mod = jl_globalref_mod(me);
jl_sym_t *me_sym = jl_globalref_name(me);
if (fe_mod->istopmod && !strcmp(jl_symbol_name(fe_sym), "getproperty") && jl_is_symbol(s)) {
if (eager_resolve || jl_binding_resolved_p(me_mod, me_sym)) {
jl_binding_t *b = jl_get_binding(me_mod, me_sym);
if (b && b->constp) {
jl_value_t *v = jl_atomic_load_relaxed(&b->value);
if (v && jl_is_module(v))
return jl_module_globalref((jl_module_t*)v, (jl_sym_t*)s);
}
}
}
}
if (e->head == jl_call_sym && nargs > 0 &&
jl_is_globalref(jl_exprarg(e, 0))) {
// TODO: this hack should be deleted once llvmcall is fixed
jl_value_t *fe = jl_exprarg(e, 0);
jl_module_t *fe_mod = jl_globalref_mod(fe);
jl_sym_t *fe_sym = jl_globalref_name(fe);
if (jl_binding_resolved_p(fe_mod, fe_sym)) {
// look at some known called functions
jl_binding_t *b = jl_get_binding(fe_mod, fe_sym);
if (b && b->constp && jl_atomic_load_relaxed(&b->value) == jl_builtin_tuple) {
size_t j;
for (j = 1; j < nargs; j++) {
if (!jl_is_quotenode(jl_exprarg(e, j)))
break;
}
if (j == nargs) {
jl_value_t *val = NULL;
JL_TRY {
val = jl_interpret_toplevel_expr_in(module, (jl_value_t*)e, NULL, sparam_vals);
}
JL_CATCH {
}
if (val)
return val;
}
}
}
}
}
}
return expr;
}
JL_DLLEXPORT void jl_resolve_globals_in_ir(jl_array_t *stmts, jl_module_t *m, jl_svec_t *sparam_vals,
int binding_effects)
{
size_t i, l = jl_array_len(stmts);
for (i = 0; i < l; i++) {
jl_value_t *stmt = jl_array_ptr_ref(stmts, i);
jl_array_ptr_set(stmts, i, resolve_globals(stmt, m, sparam_vals, binding_effects, 0));
}
}
jl_value_t *expr_arg1(jl_value_t *expr) {
jl_array_t *args = ((jl_expr_t*)expr)->args;
return jl_array_ptr_ref(args, 0);
}
// copy a :lambda Expr into its CodeInfo representation,
// including popping of known meta nodes
static void jl_code_info_set_ir(jl_code_info_t *li, jl_expr_t *ir)
{
assert(jl_is_expr(ir));
jl_expr_t *bodyex = (jl_expr_t*)jl_exprarg(ir, 2);
jl_value_t *codelocs = jl_exprarg(ir, 3);
li->linetable = jl_exprarg(ir, 4);
size_t nlocs = jl_array_len(codelocs);
li->codelocs = (jl_value_t*)jl_alloc_array_1d(jl_array_int32_type, nlocs);
size_t j;
for (j = 0; j < nlocs; j++) {
jl_arrayset((jl_array_t*)li->codelocs, jl_box_int32(jl_unbox_long(jl_arrayref((jl_array_t*)codelocs, j))),
j);
}
assert(jl_is_expr(bodyex));
jl_array_t *body = bodyex->args;
li->code = body;
jl_gc_wb(li, li->code);
size_t n = jl_array_len(body);
jl_value_t **bd = (jl_value_t**)jl_array_ptr_data((jl_array_t*)li->code);
li->ssaflags = jl_alloc_array_1d(jl_array_uint8_type, n);
jl_gc_wb(li, li->ssaflags);
int inbounds_depth = 0; // number of stacked inbounds
// isempty(inline_flags): no user annotation
// last(inline_flags) == 1: inline region
// last(inline_flags) == 0: noinline region
arraylist_t *inline_flags = arraylist_new((arraylist_t*)malloc_s(sizeof(arraylist_t)), 0);
for (j = 0; j < n; j++) {
jl_value_t *st = bd[j];
int is_flag_stmt = 0;
// check :meta expression
if (jl_is_expr(st) && ((jl_expr_t*)st)->head == jl_meta_sym) {
size_t k, ins = 0, na = jl_expr_nargs(st);
jl_array_t *meta = ((jl_expr_t*)st)->args;
for (k = 0; k < na; k++) {
jl_value_t *ma = jl_array_ptr_ref(meta, k);
if (ma == (jl_value_t*)jl_inline_sym)
li->inlining = 1;
else if (ma == (jl_value_t*)jl_noinline_sym)
li->inlining = 2;
else if (ma == (jl_value_t*)jl_propagate_inbounds_sym)
li->propagate_inbounds = 1;
else if (ma == (jl_value_t*)jl_nospecializeinfer_sym)
li->nospecializeinfer = 1;
else if (ma == (jl_value_t*)jl_aggressive_constprop_sym)
li->constprop = 1;
else if (ma == (jl_value_t*)jl_no_constprop_sym)
li->constprop = 2;
else if (jl_is_expr(ma) && ((jl_expr_t*)ma)->head == jl_purity_sym) {
if (jl_expr_nargs(ma) == 7) {
li->purity.overrides.ipo_consistent = jl_unbox_bool(jl_exprarg(ma, 0));
li->purity.overrides.ipo_effect_free = jl_unbox_bool(jl_exprarg(ma, 1));
li->purity.overrides.ipo_nothrow = jl_unbox_bool(jl_exprarg(ma, 2));
li->purity.overrides.ipo_terminates_globally = jl_unbox_bool(jl_exprarg(ma, 3));
li->purity.overrides.ipo_terminates_locally = jl_unbox_bool(jl_exprarg(ma, 4));
li->purity.overrides.ipo_notaskstate = jl_unbox_bool(jl_exprarg(ma, 5));
li->purity.overrides.ipo_inaccessiblememonly = jl_unbox_bool(jl_exprarg(ma, 6));
}
}
else
jl_array_ptr_set(meta, ins++, ma);
}
if (ins == 0)
bd[j] = jl_nothing;
else
jl_array_del_end(meta, na - ins);
}
// check other flag expressions
else if (jl_is_expr(st) && ((jl_expr_t*)st)->head == jl_inbounds_sym) {
is_flag_stmt = 1;
jl_value_t *arg1 = expr_arg1(st);
if (arg1 == (jl_value_t*)jl_true) // push
inbounds_depth += 1;
else if (arg1 == (jl_value_t*)jl_false) // clear
inbounds_depth = 0;
else if (inbounds_depth > 0) // pop
inbounds_depth -= 1;
bd[j] = jl_nothing;
}
else if (jl_is_expr(st) && ((jl_expr_t*)st)->head == jl_inline_sym) {
is_flag_stmt = 1;
jl_value_t *arg1 = expr_arg1(st);
if (arg1 == (jl_value_t*)jl_true) // enter inline region
arraylist_push(inline_flags, (void*)1);
else { // exit inline region
assert(arg1 == (jl_value_t*)jl_false);
arraylist_pop(inline_flags);
}
bd[j] = jl_nothing;
}
else if (jl_is_expr(st) && ((jl_expr_t*)st)->head == jl_noinline_sym) {
is_flag_stmt = 1;
jl_value_t *arg1 = expr_arg1(st);
if (arg1 == (jl_value_t*)jl_true) // enter noinline region
arraylist_push(inline_flags, (void*)0);
else { // exit noinline region
assert(arg1 == (jl_value_t*)jl_false);
arraylist_pop(inline_flags);
}
bd[j] = jl_nothing;
}
else if (jl_is_expr(st) && ((jl_expr_t*)st)->head == jl_return_sym) {
jl_array_ptr_set(body, j, jl_new_struct(jl_returnnode_type, jl_exprarg(st, 0)));
}
else if (jl_is_expr(st) && (((jl_expr_t*)st)->head == jl_foreigncall_sym || ((jl_expr_t*)st)->head == jl_cfunction_sym)) {
li->has_fcall = 1;
}
if (is_flag_stmt)
jl_array_uint8_set(li->ssaflags, j, 0);
else {
uint8_t flag = 0;
if (inbounds_depth > 0)
flag |= 1 << 0;
if (inline_flags->len > 0) {
void* inline_flag = inline_flags->items[inline_flags->len - 1];
flag |= 1 << (inline_flag ? 1 : 2);
}
jl_array_uint8_set(li->ssaflags, j, flag);
}
}
assert(inline_flags->len == 0); // malformed otherwise
arraylist_free(inline_flags);
free(inline_flags);
jl_array_t *vinfo = (jl_array_t*)jl_exprarg(ir, 1);
jl_array_t *vis = (jl_array_t*)jl_array_ptr_ref(vinfo, 0);
size_t nslots = jl_array_len(vis);
jl_value_t *ssavalue_types = jl_array_ptr_ref(vinfo, 2);
assert(jl_is_long(ssavalue_types));
size_t nssavalue = jl_unbox_long(ssavalue_types);
li->slotnames = jl_alloc_array_1d(jl_array_symbol_type, nslots);
jl_gc_wb(li, li->slotnames);
li->slotflags = jl_alloc_array_1d(jl_array_uint8_type, nslots);
jl_gc_wb(li, li->slotflags);
li->ssavaluetypes = jl_box_long(nssavalue);
jl_gc_wb(li, li->ssavaluetypes);
// Flags that need to be copied to slotflags
const uint8_t vinfo_mask = 8 | 16 | 32 | 64;
int i;
for (i = 0; i < nslots; i++) {
jl_value_t *vi = jl_array_ptr_ref(vis, i);
jl_sym_t *name = (jl_sym_t*)jl_array_ptr_ref(vi, 0);
assert(jl_is_symbol(name));
char *str = jl_symbol_name(name);
if (i > 0 && name != jl_unused_sym) {
if (str[0] == '#') {
// convention for renamed variables: #...#original_name
char *nxt = strchr(str + 1, '#');
if (nxt)
name = jl_symbol(nxt+1);
else if (str[1] == 's') // compiler-generated temporaries, #sXXX
name = jl_empty_sym;
}
}
jl_array_ptr_set(li->slotnames, i, name);
jl_array_uint8_set(li->slotflags, i, vinfo_mask & jl_unbox_long(jl_array_ptr_ref(vi, 2)));
}
}
JL_DLLEXPORT jl_method_instance_t *jl_new_method_instance_uninit(void)
{
jl_task_t *ct = jl_current_task;
jl_method_instance_t *mi =
(jl_method_instance_t*)jl_gc_alloc(ct->ptls, sizeof(jl_method_instance_t),
jl_method_instance_type);
mi->def.value = NULL;
mi->specTypes = NULL;
mi->sparam_vals = jl_emptysvec;
jl_atomic_store_relaxed(&mi->uninferred, NULL);
mi->backedges = NULL;
mi->callbacks = NULL;
jl_atomic_store_relaxed(&mi->cache, NULL);
mi->inInference = 0;
mi->cache_with_orig = 0;
jl_atomic_store_relaxed(&mi->precompiled, 0);
return mi;
}
JL_DLLEXPORT jl_code_info_t *jl_new_code_info_uninit(void)
{
jl_task_t *ct = jl_current_task;
jl_code_info_t *src =
(jl_code_info_t*)jl_gc_alloc(ct->ptls, sizeof(jl_code_info_t),
jl_code_info_type);
src->code = NULL;
src->codelocs = NULL;
src->ssavaluetypes = NULL;
src->ssaflags = NULL;
src->method_for_inference_limit_heuristics = jl_nothing;
src->linetable = jl_nothing;
src->slotflags = NULL;
src->slotnames = NULL;
src->slottypes = jl_nothing;
src->parent = (jl_method_instance_t*)jl_nothing;
src->rettype = (jl_value_t*)jl_any_type;
src->min_world = 1;
src->max_world = ~(size_t)0;
src->inferred = 0;
src->propagate_inbounds = 0;
src->has_fcall = 0;
src->nospecializeinfer = 0;
src->edges = jl_nothing;
src->constprop = 0;
src->inlining = 0;
src->purity.bits = 0;
src->inlining_cost = UINT16_MAX;
return src;
}
jl_code_info_t *jl_new_code_info_from_ir(jl_expr_t *ir)
{
jl_code_info_t *src = NULL;
JL_GC_PUSH1(&src);
src = jl_new_code_info_uninit();
jl_code_info_set_ir(src, ir);
JL_GC_POP();
return src;
}
void jl_add_function_to_lineinfo(jl_code_info_t *ci, jl_value_t *func)
{
// func may contain jl_symbol (function name), jl_method_t, or jl_method_instance_t
jl_array_t *li = (jl_array_t*)ci->linetable;
size_t i, n = jl_array_len(li);
jl_value_t *rt = NULL, *lno = NULL, *inl = NULL;
JL_GC_PUSH3(&rt, &lno, &inl);
for (i = 0; i < n; i++) {
jl_value_t *ln = jl_array_ptr_ref(li, i);
assert(jl_typetagis(ln, jl_lineinfonode_type));
jl_value_t *mod = jl_fieldref_noalloc(ln, 0);
jl_value_t *file = jl_fieldref_noalloc(ln, 2);
lno = jl_fieldref(ln, 3);
inl = jl_fieldref(ln, 4);
// respect a given linetable if available
jl_value_t *ln_func = jl_fieldref_noalloc(ln, 1);
if (jl_is_symbol(ln_func) && (jl_sym_t*)ln_func == jl_symbol("none") && jl_is_int32(inl) && jl_unbox_int32(inl) == 0)
ln_func = func;
rt = jl_new_struct(jl_lineinfonode_type, mod, ln_func, file, lno, inl);
jl_array_ptr_set(li, i, rt);
}
JL_GC_POP();
}
// invoke (compiling if necessary) the jlcall function pointer for a method template
static jl_value_t *jl_call_staged(jl_method_t *def, jl_value_t *generator,
size_t world, jl_svec_t *sparam_vals, jl_value_t **args, uint32_t nargs)
{
size_t n_sparams = jl_svec_len(sparam_vals);
jl_value_t **gargs;
size_t totargs = 2 + n_sparams + def->nargs;
JL_GC_PUSHARGS(gargs, totargs);
gargs[0] = jl_box_ulong(world);
gargs[1] = jl_box_long(def->line);
gargs[1] = jl_new_struct(jl_linenumbernode_type, gargs[1], def->file);
memcpy(&gargs[2], jl_svec_data(sparam_vals), n_sparams * sizeof(void*));
memcpy(&gargs[2 + n_sparams], args, (def->nargs - def->isva) * sizeof(void*));
if (def->isva)
gargs[totargs - 1] = jl_f_tuple(NULL, &args[def->nargs - 1], nargs - def->nargs + 1);
jl_value_t *code = jl_apply_generic(generator, gargs, totargs);
JL_GC_POP();
return code;
}
// Lower `ex` into Julia IR, and (if it expands into a CodeInfo) resolve global-variable
// references in light of the provided type parameters.
// Like `jl_expand`, if there is an error expanding the provided expression, the return value
// will be an error expression (an `Expr` with `error_sym` as its head), which should be eval'd
// in the caller's context.
JL_DLLEXPORT jl_code_info_t *jl_expand_and_resolve(jl_value_t *ex, jl_module_t *module,
jl_svec_t *sparam_vals) {
jl_code_info_t *func = (jl_code_info_t*)jl_expand((jl_value_t*)ex, module);
JL_GC_PUSH1(&func);
if (jl_is_code_info(func)) {
jl_array_t *stmts = (jl_array_t*)func->code;
jl_resolve_globals_in_ir(stmts, module, sparam_vals, 1);
}
JL_GC_POP();
return func;
}
// Return a newly allocated CodeInfo for the function signature
// effectively described by the tuple (specTypes, env, Method) inside linfo
JL_DLLEXPORT jl_code_info_t *jl_code_for_staged(jl_method_instance_t *linfo, size_t world)
{
jl_value_t *uninferred = jl_atomic_load_relaxed(&linfo->uninferred);
if (uninferred) {
assert(jl_is_code_info(uninferred)); // make sure this did not get `nothing` put here
return (jl_code_info_t*)jl_copy_ast((jl_value_t*)uninferred);
}
JL_TIMING(STAGED_FUNCTION, STAGED_FUNCTION);
jl_value_t *tt = linfo->specTypes;
jl_method_t *def = linfo->def.method;
jl_timing_show_method_instance(linfo, JL_TIMING_DEFAULT_BLOCK);
jl_value_t *generator = def->generator;
assert(generator != NULL);
assert(jl_is_method(def));
jl_code_info_t *func = NULL;
jl_value_t *ex = NULL;
JL_GC_PUSH2(&ex, &func);
jl_task_t *ct = jl_current_task;
int last_lineno = jl_lineno;
int last_in = ct->ptls->in_pure_callback;
size_t last_age = ct->world_age;
JL_TRY {
ct->ptls->in_pure_callback = 1;
ct->world_age = def->primary_world;
// invoke code generator
jl_tupletype_t *ttdt = (jl_tupletype_t*)jl_unwrap_unionall(tt);
ex = jl_call_staged(def, generator, world, linfo->sparam_vals, jl_svec_data(ttdt->parameters), jl_nparams(ttdt));
// do some post-processing
if (jl_is_code_info(ex)) {
func = (jl_code_info_t*)ex;
jl_array_t *stmts = (jl_array_t*)func->code;
jl_resolve_globals_in_ir(stmts, def->module, linfo->sparam_vals, 1);
}
else {
// Lower the user's expression and resolve references to the type parameters
func = jl_expand_and_resolve(ex, def->module, linfo->sparam_vals);
if (!jl_is_code_info(func)) {
if (jl_is_expr(func) && ((jl_expr_t*)func)->head == jl_error_sym) {
ct->ptls->in_pure_callback = 0;
jl_toplevel_eval(def->module, (jl_value_t*)func);
}
jl_error("The function body AST defined by this @generated function is not pure. This likely means it contains a closure, a comprehension or a generator.");
}
}
jl_add_function_to_lineinfo(func, (jl_value_t*)def->name);
// If this generated function has an opaque closure, cache it for
// correctness of method identity
for (int i = 0; i < jl_array_len(func->code); ++i) {
jl_value_t *stmt = jl_array_ptr_ref(func->code, i);
if (jl_is_expr(stmt) && ((jl_expr_t*)stmt)->head == jl_new_opaque_closure_sym) {
if (jl_options.incremental && jl_generating_output())
jl_error("Impossible to correctly handle OpaqueClosure inside @generated returned during precompile process.");
jl_value_t *uninferred = jl_copy_ast((jl_value_t*)func);
jl_value_t *old = NULL;
if (jl_atomic_cmpswap(&linfo->uninferred, &old, uninferred)) {
jl_gc_wb(linfo, uninferred);
}
else {
assert(jl_is_code_info(old));
func = (jl_code_info_t*)old;
}
break;
}
}
ct->ptls->in_pure_callback = last_in;
jl_lineno = last_lineno;
ct->world_age = last_age;
}
JL_CATCH {
ct->ptls->in_pure_callback = last_in;
jl_lineno = last_lineno;
jl_rethrow();
}
JL_GC_POP();
return func;
}
JL_DLLEXPORT jl_code_info_t *jl_copy_code_info(jl_code_info_t *src)
{
jl_task_t *ct = jl_current_task;
jl_code_info_t *newsrc =
(jl_code_info_t*)jl_gc_alloc(ct->ptls, sizeof(jl_code_info_t),
jl_code_info_type);
*newsrc = *src;
return newsrc;
}
// return a new lambda-info that has some extra static parameters merged in
jl_method_instance_t *jl_get_specialized(jl_method_t *m, jl_value_t *types, jl_svec_t *sp)
{
assert((size_t)jl_subtype_env_size(m->sig) == jl_svec_len(sp) || sp == jl_emptysvec);
jl_method_instance_t *new_linfo = jl_new_method_instance_uninit();
new_linfo->def.method = m;
new_linfo->specTypes = types;
new_linfo->sparam_vals = sp;
return new_linfo;
}
JL_DLLEXPORT void jl_method_set_source(jl_method_t *m, jl_code_info_t *src)
{
uint8_t j;
uint8_t called = 0;
int gen_only = 0;
for (j = 1; j < m->nargs && j <= sizeof(m->nospecialize) * 8; j++) {
jl_value_t *ai = jl_array_ptr_ref(src->slotnames, j);
if (ai == (jl_value_t*)jl_unused_sym) {
// TODO: enable this. currently it triggers a bug on arguments like
// ::Type{>:Missing}
//int sn = j-1;
//m->nospecialize |= (1 << sn);
continue;
}
if (j <= 8) {
if (jl_array_uint8_ref(src->slotflags, j) & 64)
called |= (1 << (j - 1));
}
}
m->called = called;
m->nospecializeinfer = src->nospecializeinfer;
m->constprop = src->constprop;
m->purity.bits = src->purity.bits;
jl_add_function_to_lineinfo(src, (jl_value_t*)m->name);
jl_array_t *copy = NULL;
jl_svec_t *sparam_vars = jl_outer_unionall_vars(m->sig);
JL_GC_PUSH3(©, &sparam_vars, &src);
assert(jl_typetagis(src->code, jl_array_any_type));
jl_array_t *stmts = (jl_array_t*)src->code;
size_t i, n = jl_array_len(stmts);
copy = jl_alloc_vec_any(n);
for (i = 0; i < n; i++) {
jl_value_t *st = jl_array_ptr_ref(stmts, i);
if (jl_is_expr(st) && ((jl_expr_t*)st)->head == jl_meta_sym) {
size_t nargs = jl_expr_nargs(st);
if (nargs >= 1 && jl_exprarg(st, 0) == (jl_value_t*)jl_nospecialize_sym) {
if (nargs == 1) // bare `@nospecialize` is special: it prevents specialization on all args
m->nospecialize = -1;
size_t j;
for (j = 1; j < nargs; j++) {
jl_value_t *aj = jl_exprarg(st, j);
if (!jl_is_slotnumber(aj) && !jl_is_argument(aj))
continue;
int sn = (int)jl_slot_number(aj) - 2;
if (sn < 0) // @nospecialize on self is valid but currently ignored
continue;
if (sn > (m->nargs - 2)) {
jl_error("@nospecialize annotation applied to a non-argument");
}
if (sn >= sizeof(m->nospecialize) * 8) {
jl_printf(JL_STDERR,
"WARNING: @nospecialize annotation only supported on the first %d arguments.\n",
(int)(sizeof(m->nospecialize) * 8));
continue;
}
m->nospecialize |= (1 << sn);
}
st = jl_nothing;
}
else if (nargs >= 1 && jl_exprarg(st, 0) == (jl_value_t*)jl_specialize_sym) {
if (nargs == 1) // bare `@specialize` is special: it causes specialization on all args
m->nospecialize = 0;
for (j = 1; j < nargs; j++) {
jl_value_t *aj = jl_exprarg(st, j);
if (!jl_is_slotnumber(aj) && !jl_is_argument(aj))
continue;
int sn = (int)jl_slot_number(aj) - 2;
if (sn < 0) // @specialize on self is valid but currently ignored
continue;
if (sn > (m->nargs - 2)) {
jl_error("@specialize annotation applied to a non-argument");
}
if (sn >= sizeof(m->nospecialize) * 8) {
jl_printf(JL_STDERR,
"WARNING: @specialize annotation only supported on the first %d arguments.\n",
(int)(sizeof(m->nospecialize) * 8));
continue;
}
m->nospecialize &= ~(1 << sn);
}
st = jl_nothing;
}
else if (nargs == 2 && jl_exprarg(st, 0) == (jl_value_t*)jl_generated_sym) {
if (m->generator != NULL)
jl_error("duplicate @generated function body");
jl_value_t *gexpr = jl_exprarg(st, 1);
// the frontend would put (new (core GeneratedFunctionStub) funcname argnames sp) here, for example
m->generator = jl_toplevel_eval(m->module, gexpr);
jl_gc_wb(m, m->generator);
st = jl_nothing;
}
else if (nargs == 1 && jl_exprarg(st, 0) == (jl_value_t*)jl_generated_only_sym) {
gen_only = 1;
st = jl_nothing;
}
else if (nargs == 2 && jl_exprarg(st, 0) == (jl_value_t*)jl_symbol("nkw")) {
m->nkw = jl_unbox_long(jl_exprarg(st, 1));
st = jl_nothing;
}
}
else {
st = resolve_globals(st, m->module, sparam_vars, 1, 0);
}
jl_array_ptr_set(copy, i, st);
}
src = jl_copy_code_info(src);
src->code = copy;
jl_gc_wb(src, copy);
m->slot_syms = jl_compress_argnames(src->slotnames);
jl_gc_wb(m, m->slot_syms);
if (gen_only)
m->source = NULL;
else
m->source = (jl_value_t*)jl_compress_ir(m, src);
jl_gc_wb(m, m->source);
JL_GC_POP();
}
JL_DLLEXPORT jl_method_t *jl_new_method_uninit(jl_module_t *module)
{
jl_task_t *ct = jl_current_task;
jl_method_t *m =
(jl_method_t*)jl_gc_alloc(ct->ptls, sizeof(jl_method_t), jl_method_type);
jl_atomic_store_relaxed(&m->specializations, (jl_value_t*)jl_emptysvec);
jl_atomic_store_relaxed(&m->speckeyset, (jl_array_t*)jl_an_empty_vec_any);
m->sig = NULL;
m->slot_syms = NULL;
m->roots = NULL;
m->root_blocks = NULL;
m->nroots_sysimg = 0;
m->ccallable = NULL;
m->module = module;
m->external_mt = NULL;
m->source = NULL;
jl_atomic_store_relaxed(&m->unspecialized, NULL);
m->generator = NULL;
m->name = NULL;
m->file = jl_empty_sym;
m->line = 0;
m->called = 0xff;
m->nospecialize = module->nospecialize;
m->nkw = 0;
jl_atomic_store_relaxed(&m->invokes, jl_nothing);
m->recursion_relation = NULL;
m->isva = 0;
m->nargs = 0;
m->primary_world = 1;
m->deleted_world = ~(size_t)0;
m->is_for_opaque_closure = 0;
m->nospecializeinfer = 0;
m->constprop = 0;
m->purity.bits = 0;
m->max_varargs = UINT8_MAX;
JL_MUTEX_INIT(&m->writelock, "method->writelock");
return m;
}
// backedges ------------------------------------------------------------------
// Use this in a `while` loop to iterate over the backedges in a MethodInstance.
// `*invokesig` will be NULL if the call was made by ordinary dispatch, otherwise
// it will be the signature supplied in an `invoke` call.
// If you don't need `invokesig`, you can set it to NULL on input.
// Initialize iteration with `i = 0`. Returns `i` for the next backedge to be extracted.
int get_next_edge(jl_array_t *list, int i, jl_value_t** invokesig, jl_method_instance_t **caller) JL_NOTSAFEPOINT
{
jl_value_t *item = jl_array_ptr_ref(list, i);
if (jl_is_method_instance(item)) {
// Not an `invoke` call, it's just the MethodInstance
if (invokesig != NULL)
*invokesig = NULL;
*caller = (jl_method_instance_t*)item;
return i + 1;
}
assert(jl_is_type(item));
// An `invoke` call, it's a (sig, MethodInstance) pair
if (invokesig != NULL)
*invokesig = item;
*caller = (jl_method_instance_t*)jl_array_ptr_ref(list, i + 1);
if (*caller)
assert(jl_is_method_instance(*caller));
return i + 2;
}
int set_next_edge(jl_array_t *list, int i, jl_value_t *invokesig, jl_method_instance_t *caller)
{
if (invokesig)
jl_array_ptr_set(list, i++, invokesig);
jl_array_ptr_set(list, i++, caller);
return i;
}
void push_edge(jl_array_t *list, jl_value_t *invokesig, jl_method_instance_t *caller)
{
if (invokesig)
jl_array_ptr_1d_push(list, invokesig);
jl_array_ptr_1d_push(list, (jl_value_t*)caller);
return;
}
// method definition ----------------------------------------------------------
jl_method_t *jl_make_opaque_closure_method(jl_module_t *module, jl_value_t *name,
int nargs, jl_value_t *functionloc, jl_code_info_t *ci, int isva)
{
jl_method_t *m = jl_new_method_uninit(module);
JL_GC_PUSH1(&m);
// TODO: Maybe have a signature of (parent method, stmt#)?
m->sig = (jl_value_t*)jl_anytuple_type;
m->isva = isva;
m->is_for_opaque_closure = 1;
if (name == jl_nothing) {
m->name = jl_symbol("opaque closure");
} else {
assert(jl_is_symbol(name));
m->name = (jl_sym_t*)name;
}
m->nargs = nargs + 1;
assert(jl_is_linenode(functionloc));
jl_value_t *file = jl_linenode_file(functionloc);
m->file = jl_is_symbol(file) ? (jl_sym_t*)file : jl_empty_sym;
m->line = jl_linenode_line(functionloc);
jl_method_set_source(m, ci);
JL_GC_POP();
return m;
}
// empty generic function def
JL_DLLEXPORT jl_value_t *jl_generic_function_def(jl_sym_t *name,
jl_module_t *module,
_Atomic(jl_value_t*) *bp,
jl_binding_t *bnd)
{
jl_value_t *gf = NULL;
assert(name && bp);
if (bnd && jl_atomic_load_relaxed(&bnd->value) != NULL && !bnd->constp)
jl_errorf("cannot define function %s; it already has a value", jl_symbol_name(name));
gf = jl_atomic_load_relaxed(bp);
if (gf != NULL) {
if (!jl_is_datatype_singleton((jl_datatype_t*)jl_typeof(gf)) && !jl_is_type(gf))
jl_errorf("cannot define function %s; it already has a value", jl_symbol_name(name));
}
if (bnd)
bnd->constp = 1; // XXX: use jl_declare_constant and jl_checked_assignment
if (gf == NULL) {
gf = (jl_value_t*)jl_new_generic_function(name, module);
jl_atomic_store(bp, gf); // TODO: fix constp assignment data race
if (bnd) jl_gc_wb(bnd, gf);
}
return gf;
}
static jl_methtable_t *nth_methtable(jl_value_t *a JL_PROPAGATES_ROOT, int n) JL_NOTSAFEPOINT
{
if (jl_is_datatype(a)) {
if (n == 0) {
jl_methtable_t *mt = ((jl_datatype_t*)a)->name->mt;
if (mt != NULL)
return mt;
}
else if (jl_is_tuple_type(a)) {
if (jl_nparams(a) >= n)
return nth_methtable(jl_tparam(a, n - 1), 0);
}
}
else if (jl_is_typevar(a)) {
return nth_methtable(((jl_tvar_t*)a)->ub, n);
}
else if (jl_is_unionall(a)) {
return nth_methtable(((jl_unionall_t*)a)->body, n);
}
else if (jl_is_uniontype(a)) {
jl_uniontype_t *u = (jl_uniontype_t*)a;
jl_methtable_t *m1 = nth_methtable(u->a, n);
if ((jl_value_t*)m1 != jl_nothing) {
jl_methtable_t *m2 = nth_methtable(u->b, n);
if (m1 == m2)
return m1;
}
}
return (jl_methtable_t*)jl_nothing;
}
// get the MethodTable for dispatch, or `nothing` if cannot be determined
JL_DLLEXPORT jl_methtable_t *jl_method_table_for(jl_value_t *argtypes JL_PROPAGATES_ROOT) JL_NOTSAFEPOINT
{
return nth_methtable(argtypes, 1);
}
jl_methtable_t *jl_kwmethod_table_for(jl_value_t *argtypes JL_PROPAGATES_ROOT) JL_NOTSAFEPOINT
{
jl_methtable_t *kwmt = nth_methtable(argtypes, 3);
if ((jl_value_t*)kwmt == jl_nothing)
return NULL;
return kwmt;
}
JL_DLLEXPORT jl_methtable_t *jl_method_get_table(jl_method_t *method JL_PROPAGATES_ROOT) JL_NOTSAFEPOINT
{
return method->external_mt ? (jl_methtable_t*)method->external_mt : jl_method_table_for(method->sig);
}
jl_array_t *jl_all_methods JL_GLOBALLY_ROOTED;
JL_DLLEXPORT jl_method_t* jl_method_def(jl_svec_t *argdata,
jl_methtable_t *mt,
jl_code_info_t *f,
jl_module_t *module)
{
// argdata is svec(svec(types...), svec(typevars...), functionloc)
jl_svec_t *atypes = (jl_svec_t*)jl_svecref(argdata, 0);
jl_svec_t *tvars = (jl_svec_t*)jl_svecref(argdata, 1);
jl_value_t *functionloc = jl_svecref(argdata, 2);
assert(jl_is_svec(atypes));
assert(jl_is_svec(tvars));
size_t nargs = jl_svec_len(atypes);
assert(nargs > 0);
int isva = jl_is_vararg(jl_svecref(atypes, nargs - 1));
if (!jl_is_type(jl_svecref(atypes, 0)) || (isva && nargs == 1))
jl_error("function type in method definition is not a type");
jl_sym_t *name;
jl_method_t *m = NULL;
jl_value_t *argtype = NULL;
JL_GC_PUSH3(&f, &m, &argtype);
size_t i, na = jl_svec_len(atypes);
argtype = jl_apply_tuple_type(atypes);
if (!jl_is_datatype(argtype))
jl_error("invalid type in method definition (Union{})");
jl_methtable_t *external_mt = mt;
if (!mt)
mt = jl_method_table_for(argtype);
if ((jl_value_t*)mt == jl_nothing)
jl_error("Method dispatch is unimplemented currently for this method signature");
if (mt->frozen)
jl_error("cannot add methods to a builtin function");
assert(jl_is_linenode(functionloc));
jl_sym_t *file = (jl_sym_t*)jl_linenode_file(functionloc);
if (!jl_is_symbol(file))
file = jl_empty_sym;
int32_t line = jl_linenode_line(functionloc);
// TODO: derive our debug name from the syntax instead of the type
jl_methtable_t *kwmt = mt == jl_kwcall_mt ? jl_kwmethod_table_for(argtype) : mt;
// if we have a kwcall, try to derive the name from the callee argument method table
name = (kwmt ? kwmt : mt)->name;
if (kwmt == jl_type_type_mt || kwmt == jl_nonfunction_mt || external_mt) {
// our value for `name` is bad, try to guess what the syntax might have had,
// like `jl_static_show_func_sig` might have come up with
jl_datatype_t *dt = jl_nth_argument_datatype(argtype, mt == jl_kwcall_mt ? 3 : 1);
if (dt != NULL) {
name = dt->name->name;
if (jl_is_type_type((jl_value_t*)dt)) {
dt = (jl_datatype_t*)jl_argument_datatype(jl_tparam0(dt));
if ((jl_value_t*)dt != jl_nothing) {
name = dt->name->name;
}
}
}
}
if (!jl_is_code_info(f)) {
// this occurs when there is a closure being added to an out-of-scope function
// the user should only do this at the toplevel
// the result is that the closure variables get interpolated directly into the IR
f = jl_new_code_info_from_ir((jl_expr_t*)f);
}
for (i = 0; i < na; i++) {
jl_value_t *elt = jl_svecref(atypes, i);
int isvalid = jl_is_type(elt) || jl_is_typevar(elt) || jl_is_vararg(elt);
if (elt == jl_bottom_type || (jl_is_vararg(elt) && jl_unwrap_vararg(elt) == jl_bottom_type))
isvalid = 0;
if (!isvalid) {
jl_sym_t *argname = (jl_sym_t*)jl_array_ptr_ref(f->slotnames, i);
if (argname == jl_unused_sym)
jl_exceptionf(jl_argumenterror_type,
"invalid type for argument number %d in method definition for %s at %s:%d",
i,
jl_symbol_name(name),
jl_symbol_name(file),
line);
else
jl_exceptionf(jl_argumenterror_type,
"invalid type for argument %s in method definition for %s at %s:%d",
jl_symbol_name(argname),
jl_symbol_name(name),
jl_symbol_name(file),
line);
}
if (jl_is_vararg(elt) && i < na-1)
jl_exceptionf(jl_argumenterror_type,
"Vararg on non-final argument in method definition for %s at %s:%d",
jl_symbol_name(name),
jl_symbol_name(file),
line);
}
for (i = jl_svec_len(tvars); i > 0; i--) {
jl_value_t *tv = jl_svecref(tvars, i - 1);
if (!jl_is_typevar(tv))
jl_type_error("method signature", (jl_value_t*)jl_tvar_type, tv);
if (!jl_has_typevar(argtype, (jl_tvar_t*)tv)) // deprecate this to an error in v2
jl_printf(JL_STDERR,
"WARNING: method definition for %s at %s:%d declares type variable %s but does not use it.\n",
jl_symbol_name(name),
jl_symbol_name(file),
line,
jl_symbol_name(((jl_tvar_t*)tv)->name));
argtype = jl_new_struct(jl_unionall_type, tv, argtype);
}
if (jl_has_free_typevars(argtype)) {
jl_exceptionf(jl_argumenterror_type,
"method definition for %s at %s:%d has free type variables",
jl_symbol_name(name),
jl_symbol_name(file),
line);
}
m = jl_new_method_uninit(module);
m->external_mt = (jl_value_t*)external_mt;
if (external_mt)
jl_gc_wb(m, external_mt);
m->sig = argtype;
m->name = name;
m->isva = isva;
m->nargs = nargs;
m->file = file;
m->line = line;
jl_method_set_source(m, f);
#ifdef RECORD_METHOD_ORDER
if (jl_all_methods == NULL)
jl_all_methods = jl_alloc_vec_any(0);
#endif
if (jl_all_methods != NULL) {
while (jl_array_len(jl_all_methods) < m->primary_world)
jl_array_ptr_1d_push(jl_all_methods, NULL);
jl_array_ptr_1d_push(jl_all_methods, (jl_value_t*)m);
}
jl_method_table_insert(mt, m, NULL);
if (jl_newmeth_tracer)
jl_call_tracer(jl_newmeth_tracer, (jl_value_t*)m);
JL_GC_POP();
return m;
}
// root blocks
// This section handles method roots. Roots are GC-preserved items needed to
// represent lowered, type-inferred, and/or compiled code. These items are
// stored in a flat list (`m.roots`), and during serialization and
// deserialization of code we replace C-pointers to these items with a
// relocatable reference. We use a bipartite reference, `(key, index)` pair,
// where `key` identifies the module that added the root and `index` numbers
// just those roots with the same `key`.
//
// During precompilation (serialization), we save roots that were added to
// methods that are tagged with this package's module-key, even for "external"
// methods not owned by a module currently being precompiled. During
// deserialization, we load the new roots and append them to the method. When
// code is deserialized (see ircode.c), we replace the bipartite reference with
// the pointer to the memory address in the current session. The bipartite
// reference allows us to cache both roots and references in precompilation .ji
// files using a naming scheme that is independent of which packages are loaded
// in arbitrary order.
//
// To track the module-of-origin for each root, methods also have a
// `root_blocks` field that uses run-length encoding (RLE) storing `key` and the
// (absolute) integer index within `roots` at which a block of roots with that
// key begins. This makes it possible to look up an individual `(key, index)`
// pair fairly efficiently. A given `key` may possess more than one block; the
// `index` continues to increment regardless of block boundaries.
//
// Roots with `key = 0` are considered to be of unknown origin, and
// CodeInstances referencing such roots will remain unserializable unless all
// such roots were added at the time of system image creation. To track this
// additional data, we use two fields:
//
// - methods have an `nroots_sysimg` field to count the number of roots defined
// at the time of writing the system image (such occur first in the list of
// roots). These are the cases with `key = 0` that do not prevent
// serialization.
// - CodeInstances have a `relocatability` field which when 1 indicates that
// every root is "safe," meaning it was either added at sysimg creation or is
// tagged with a non-zero `key`. Even a single unsafe root will cause this to
// have value 0.
// Get the key of the current (final) block of roots
static uint64_t current_root_id(jl_array_t *root_blocks)
{
if (!root_blocks)
return 0;
assert(jl_is_array(root_blocks));
size_t nx2 = jl_array_len(root_blocks);
if (nx2 == 0)
return 0;
uint64_t *blocks = (uint64_t*)jl_array_data(root_blocks);
return blocks[nx2-2];
}
// Add a new block of `len` roots with key `modid` (module id)
static void add_root_block(jl_array_t *root_blocks, uint64_t modid, size_t len)
{
assert(jl_is_array(root_blocks));
jl_array_grow_end(root_blocks, 2);
uint64_t *blocks = (uint64_t*)jl_array_data(root_blocks);
int nx2 = jl_array_len(root_blocks);
blocks[nx2-2] = modid;
blocks[nx2-1] = len;
}
// Allocate storage for roots
static void prepare_method_for_roots(jl_method_t *m, uint64_t modid)
{
if (!m->roots) {
m->roots = jl_alloc_vec_any(0);
jl_gc_wb(m, m->roots);
}
if (!m->root_blocks && modid != 0) {
m->root_blocks = jl_alloc_array_1d(jl_array_uint64_type, 0);
jl_gc_wb(m, m->root_blocks);
}
}
// Add a single root with owner `mod` to a method
JL_DLLEXPORT void jl_add_method_root(jl_method_t *m, jl_module_t *mod, jl_value_t* root)
{
JL_GC_PUSH2(&m, &root);
uint64_t modid = 0;
if (mod) {
assert(jl_is_module(mod));
modid = mod->build_id.lo;
}
assert(jl_is_method(m));
prepare_method_for_roots(m, modid);
if (current_root_id(m->root_blocks) != modid)
add_root_block(m->root_blocks, modid, jl_array_len(m->roots));
jl_array_ptr_1d_push(m->roots, root);
JL_GC_POP();
}
// Add a list of roots with key `modid` to a method
void jl_append_method_roots(jl_method_t *m, uint64_t modid, jl_array_t* roots)
{
JL_GC_PUSH2(&m, &roots);
assert(jl_is_method(m));
assert(jl_is_array(roots));
prepare_method_for_roots(m, modid);
add_root_block(m->root_blocks, modid, jl_array_len(m->roots));
jl_array_ptr_1d_append(m->roots, roots);
JL_GC_POP();
}
// given the absolute index i of a root, retrieve its relocatable reference
// returns 1 if the root is relocatable
int get_root_reference(rle_reference *rr, jl_method_t *m, size_t i)
{
if (!m->root_blocks) {
rr->key = 0;
rr->index = i;
return i < m->nroots_sysimg;
}
rle_index_to_reference(rr, i, (uint64_t*)jl_array_data(m->root_blocks), jl_array_len(m->root_blocks), 0);
if (rr->key)
return 1;
return i < m->nroots_sysimg;
}
// get a root, given its key and index relative to the key
// this is the relocatable way to get a root from m->roots
jl_value_t *lookup_root(jl_method_t *m, uint64_t key, int index)
{
if (!m->root_blocks) {
assert(key == 0);
return jl_array_ptr_ref(m->roots, index);
}
rle_reference rr = {key, index};
size_t i = rle_reference_to_index(&rr, (uint64_t*)jl_array_data(m->root_blocks), jl_array_len(m->root_blocks), 0);
return jl_array_ptr_ref(m->roots, i);
}
// Count the number of roots added by module with id `key`
int nroots_with_key(jl_method_t *m, uint64_t key)
{
size_t nroots = 0;
if (m->roots)
nroots = jl_array_len(m->roots);
if (!m->root_blocks)
return key == 0 ? nroots : 0;
uint64_t *rletable = (uint64_t*)jl_array_data(m->root_blocks);
size_t j, nblocks2 = jl_array_len(m->root_blocks);
int nwithkey = 0;
for (j = 0; j < nblocks2; j+=2) {
if (rletable[j] == key)
nwithkey += (j+3 < nblocks2 ? rletable[j+3] : nroots) - rletable[j+1];
}
return nwithkey;
}
#ifdef __cplusplus
}
#endif