language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-10-02 16:21:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

gen_wasm: add support for non-recursive Tags

Browse source
This commit is contained in:
Brian Carroll 2021-12-08 20:45:49 +00:00
parent f6b6d91c56
commit 6a801ebc7e
5 changed files with 106 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

66
compiler/gen_wasm/src/backend.rs
View file

@ -6,7 +6,7 @@ use roc_module::low_level::LowLevel;
use roc_module::symbol::{Interns, Symbol};
use roc_mono::gen_refcount::{RefcountProcGenerator, REFCOUNT_MAX};
use roc_mono::ir::{CallType, Expr, JoinPointId, Literal, Proc, Stmt};
use roc_mono::layout::{Builtin, Layout, LayoutIds};
use roc_mono::layout::{Builtin, Layout, LayoutIds, TagIdIntType, UnionLayout};
use crate::layout::{CallConv, ReturnMethod, WasmLayout};
use crate::low_level::{decode_low_level, LowlevelBuildResult};
@ -24,8 +24,8 @@ use crate::wasm_module::{
LinkingSubSection, LocalId, Signature, SymInfo, ValueType,
};
use crate::{
copy_memory, CopyMemoryConfig, Env, BUILTINS_IMPORT_MODULE_NAME, MEMORY_NAME, PTR_SIZE,
PTR_TYPE, STACK_POINTER_GLOBAL_ID, STACK_POINTER_NAME,
copy_memory, round_up_to_alignment, CopyMemoryConfig, Env, BUILTINS_IMPORT_MODULE_NAME,
MEMORY_NAME, PTR_SIZE, PTR_TYPE, STACK_POINTER_GLOBAL_ID, STACK_POINTER_NAME,
};
/// The memory address where the constants data will be loaded during module instantiation.
@ -633,7 +633,7 @@ impl<'a> WasmBackend<'a> {
Ok(())
}
Expr::Array { .. } => Err(format!("Expression is not yet implemented {:?}", 2)),
Expr::Array { .. } => Err(format!("Expression is not yet implemented {:?}", expr)),
Expr::EmptyArray => {
if let StoredValue::StackMemory { location, .. } = storage {
@ -653,10 +653,68 @@ impl<'a> WasmBackend<'a> {
}
}
Expr::Tag {
tag_layout,
tag_id,
arguments,
..
} => {
self.build_tag(tag_layout, *tag_id, arguments, storage);
Ok(())
}
x => Err(format!("Expression is not yet implemented {:?}", x)),
}
}
fn build_tag(
&mut self,
union_layout: &UnionLayout<'a>,
tag_id: TagIdIntType,
arguments: &'a [Symbol],
stored: &StoredValue,
) {
match union_layout {
UnionLayout::NonRecursive(tags) => {
let (local_id, offset) = if let StoredValue::StackMemory { location, .. } = stored {
location.local_and_offset(self.storage.stack_frame_pointer)
} else {
panic!("NonRecursive Tag should always be stored in StackMemory");
};
let mut field_offset = offset;
for field_symbol in arguments.iter() {
field_offset += self.storage.copy_value_to_memory(
&mut self.code_builder,
local_id,
field_offset,
*field_symbol,
);
}
let tag_field_layouts = &tags[tag_id as usize];
let alignment_bytes = Layout::Struct(tag_field_layouts).alignment_bytes(PTR_SIZE);
let tag_id_offset =
round_up_to_alignment(field_offset as i32, alignment_bytes as i32) as u32;
let tag_id_align = Align::from(alignment_bytes);
match tag_id_align {
Align::Bytes1 | Align::Bytes2 | Align::Bytes4 => {
self.code_builder.get_local(local_id);
self.code_builder.i32_const(tag_id as i32);
self.code_builder.i32_store(tag_id_align, tag_id_offset);
}
_ => {
self.code_builder.get_local(local_id);
self.code_builder.i64_const(tag_id as i64);
self.code_builder.i64_store(tag_id_align, tag_id_offset);
}
}
}
_ => unimplemented!("Tag with layout {:?}", union_layout),
}
}
fn build_low_level(
&mut self,
lowlevel: LowLevel,