language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-27 05:49:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

start correct list refcount function generation

Browse source
This commit is contained in:
Brendan Hansknecht 2024-07-07 10:53:39 -07:00
parent 61784629e7
commit 94fe0820d5
No known key found for this signature in database
GPG key ID: 0EA784685083E75B
1 changed files with 81 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

99
crates/compiler/mono/src/code_gen_help/refcount.rs
View file

@ -196,7 +196,14 @@ pub fn refcount_generic<'a>(
rc_return_stmt(root, ident_ids, ctx)
}
LayoutRepr::Builtin(Builtin::Str) => refcount_str(root, ident_ids, ctx),
LayoutRepr::Builtin(Builtin::List(_)) => refcount_list(root, ident_ids, ctx, structure),
LayoutRepr::Builtin(Builtin::List(element_layout)) => refcount_list(
root,
ident_ids,
ctx,
layout_interner,
element_layout,
structure,
),
LayoutRepr::Struct(field_layouts) => refcount_struct(
root,
ident_ids,
@ -944,42 +951,98 @@ fn refcount_list<'a>(
root: &mut CodeGenHelp<'a>,
ident_ids: &mut IdentIds,
ctx: &mut Context<'a>,
_structure: Symbol,
layout_interner: &mut STLayoutInterner<'a>,
element_layout: InLayout<'a>,
structure: Symbol,
) -> Stmt<'a> {
let arena = root.arena;
let layout_isize = root.layout_isize;
let rc_list_unit = root.create_symbol(ident_ids, "rc_list");
let ret_stmt = arena.alloc(rc_return_stmt(root, ident_ids, ctx));
let is_refcounted = root.create_symbol(ident_ids, "is_refcounted");
let is_refcounted_expr =
Expr::Literal(Literal::Bool(layout_interner.is_refcounted(element_layout)));
let is_refcounted_stmt = |next| Stmt::Let(is_refcounted, is_refcounted_expr, LAYOUT_BOOL, next);
let list = Symbol::ARG_1;
let rc_list_expr = match ctx.op {
HelperOp::IncN | HelperOp::Inc => {
// TODO: refcount_args is totally wrong here.
// Should be list, amount, elements_refcounted
let rc_list_args = refcount_args(root, ctx, list);
Expr::Call(Call {
match ctx.op {
HelperOp::IncN => {
let amount = Symbol::ARG_2;
let rc_list_expr = Expr::Call(Call {
call_type: CallType::LowLevel {
op: LowLevel::ListIncref,
update_mode: UpdateModeId::BACKEND_DUMMY,
},
arguments: rc_list_args,
})
arguments: root.arena.alloc([list, amount, is_refcounted]),
});
is_refcounted_stmt(arena.alloc(
//
Stmt::Let(rc_list_unit, rc_list_expr, LAYOUT_UNIT, ret_stmt),
))
}
HelperOp::Inc => {
let one = root.create_symbol(ident_ids, "one");
let one_expr = Expr::Literal(Literal::Int(0i128.to_ne_bytes()));
let one_stmt = |next| Stmt::Let(one, one_expr, layout_isize, next);
let rc_list_expr = Expr::Call(Call {
call_type: CallType::LowLevel {
op: LowLevel::ListIncref,
update_mode: UpdateModeId::BACKEND_DUMMY,
},
arguments: root.arena.alloc([list, one, is_refcounted]),
});
one_stmt(arena.alloc(
//
is_refcounted_stmt(arena.alloc(
//
Stmt::Let(rc_list_unit, rc_list_expr, LAYOUT_UNIT, ret_stmt),
)),
))
}
HelperOp::DecRef(_) | HelperOp::Dec => {
// TODO: refcount_args is totally wrong here.
// Should be list, alignment, element_width, elements_refcounted, element_def_fn
let rc_list_args = refcount_args(root, ctx, list);
Expr::Call(Call {
let (element_width, element_alignment) =
layout_interner.stack_size_and_alignment(element_layout);
let width = root.create_symbol(ident_ids, "width");
let width_expr = Expr::Literal(Literal::Int((element_width as u128).to_ne_bytes()));
let width_stmt = |next| Stmt::Let(width, width_expr, layout_isize, next);
let alignment = root.create_symbol(ident_ids, "alignment");
let alignment_expr =
Expr::Literal(Literal::Int((element_alignment as u128).to_ne_bytes()));
let alignment_stmt = |next| Stmt::Let(alignment, alignment_expr, LAYOUT_U32, next);
// let orig_op = ctx.op;
// ctx.op = HelperOp::IndirectDec;
// let dec_elem_fn =
// root.find_or_create_proc(ident_ids, ctx, layout_interner, element_layout);
// ctx.op = orig_op;
// TODO: How do I load a proc symbol to a pointer???
let rc_list_expr = Expr::Call(Call {
call_type: CallType::LowLevel {
op: LowLevel::ListDecref,
update_mode: UpdateModeId::BACKEND_DUMMY,
},
arguments: rc_list_args,
})
arguments: root.arena.alloc([list, alignment, width, is_refcounted]),
});
width_stmt(arena.alloc(
//
alignment_stmt(arena.alloc(
//
is_refcounted_stmt(arena.alloc(
//
Stmt::Let(rc_list_unit, rc_list_expr, LAYOUT_UNIT, ret_stmt),
)),
)),
))
}
_ => unreachable!(),
};
let rc_list_unit = root.create_symbol(ident_ids, "rc_list");
Stmt::Let(rc_list_unit, rc_list_expr, LAYOUT_UNIT, ret_stmt)
}
}
fn refcount_struct<'a>(