mirror of
https://github.com/roc-lang/roc.git
synced 2025-09-27 13:59:08 +00:00
Add List.prepend
implementation for generic64
This commit is contained in:
parent
6bc3ba6407
commit
7fc24d5720
3 changed files with 110 additions and 2 deletions
|
@ -1758,6 +1758,93 @@ impl<
|
|||
self.free_symbol(&Symbol::DEV_TMP3);
|
||||
}
|
||||
|
||||
fn build_list_prepend(
|
||||
&mut self,
|
||||
dst: &Symbol,
|
||||
args: &'a [Symbol],
|
||||
arg_layouts: &[InLayout<'a>],
|
||||
ret_layout: &InLayout<'a>,
|
||||
) {
|
||||
let list = args[0];
|
||||
let list_layout = arg_layouts[0];
|
||||
let elem = args[1];
|
||||
let elem_layout = arg_layouts[1];
|
||||
|
||||
// List alignment argument (u32).
|
||||
let u32_layout = Layout::U32;
|
||||
let list_alignment = self.layout_interner.alignment_bytes(*ret_layout);
|
||||
self.load_literal(
|
||||
&Symbol::DEV_TMP,
|
||||
&u32_layout,
|
||||
&Literal::Int((list_alignment as i128).to_ne_bytes()),
|
||||
);
|
||||
|
||||
// Have to pass the input element by pointer, so put it on the stack and load it's address.
|
||||
self.storage_manager
|
||||
.ensure_symbol_on_stack(&mut self.buf, &elem);
|
||||
let (new_elem_offset, _) = self.storage_manager.stack_offset_and_size(&elem);
|
||||
|
||||
// Load address of input element into register.
|
||||
let reg = self
|
||||
.storage_manager
|
||||
.claim_general_reg(&mut self.buf, &Symbol::DEV_TMP2);
|
||||
ASM::add_reg64_reg64_imm32(&mut self.buf, reg, CC::BASE_PTR_REG, new_elem_offset);
|
||||
|
||||
// Load element_witdh argument (usize).
|
||||
let u64_layout = Layout::U64;
|
||||
let elem_stack_size = self.layout_interner.stack_size(elem_layout);
|
||||
self.load_literal(
|
||||
&Symbol::DEV_TMP3,
|
||||
&u64_layout,
|
||||
&Literal::Int((elem_stack_size as i128).to_ne_bytes()),
|
||||
);
|
||||
|
||||
// Setup the return location.
|
||||
let base_offset = self
|
||||
.storage_manager
|
||||
.claim_stack_area(dst, self.layout_interner.stack_size(*ret_layout));
|
||||
|
||||
let lowlevel_args = bumpalo::vec![
|
||||
in self.env.arena;
|
||||
list,
|
||||
// alignment
|
||||
Symbol::DEV_TMP,
|
||||
// element
|
||||
Symbol::DEV_TMP2,
|
||||
// element_width
|
||||
Symbol::DEV_TMP3,
|
||||
];
|
||||
let lowlevel_arg_layouts = bumpalo::vec![
|
||||
in self.env.arena;
|
||||
list_layout,
|
||||
u32_layout,
|
||||
u64_layout,
|
||||
u64_layout,
|
||||
];
|
||||
|
||||
self.build_fn_call(
|
||||
&Symbol::DEV_TMP4,
|
||||
bitcode::LIST_PREPEND.to_string(),
|
||||
&lowlevel_args,
|
||||
&lowlevel_arg_layouts,
|
||||
ret_layout,
|
||||
);
|
||||
self.free_symbol(&Symbol::DEV_TMP);
|
||||
self.free_symbol(&Symbol::DEV_TMP2);
|
||||
self.free_symbol(&Symbol::DEV_TMP3);
|
||||
|
||||
// Return list value from fn call
|
||||
self.storage_manager.copy_symbol_to_stack_offset(
|
||||
self.layout_interner,
|
||||
&mut self.buf,
|
||||
base_offset,
|
||||
&Symbol::DEV_TMP4,
|
||||
ret_layout,
|
||||
);
|
||||
|
||||
self.free_symbol(&Symbol::DEV_TMP4);
|
||||
}
|
||||
|
||||
fn build_ptr_cast(&mut self, dst: &Symbol, src: &Symbol) {
|
||||
let dst_reg = self.storage_manager.claim_general_reg(&mut self.buf, dst);
|
||||
self.storage_manager
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue