Add List.prepend implementation for generic64

This commit is contained in:
Ahmad Sattar 2023-01-28 22:46:35 +01:00
parent 6bc3ba6407
commit 7fc24d5720
No known key found for this signature in database
GPG key ID: DE5DDDB4ACC89D39
3 changed files with 110 additions and 2 deletions

View file

@ -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