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

Extract some common wrapper logic

Browse source
This commit is contained in:
Richard Feldman 2022-08-05 13:49:00 -04:00
parent efa31ae747
commit e9966e6551
No known key found for this signature in database
GPG key ID: 7E4127D1E4241798
1 changed files with 18 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

32
crates/glue/src/types.rs
View file

@ -534,7 +534,10 @@ pub enum RocTagUnion {
/// A recursive tag union with just one constructor
/// Optimization: No need to store a tag ID (the payload is "unwrapped")
/// e.g. `RoseTree a : [Tree a (List (RoseTree a))]`
NonNullableUnwrapped { name: String, content: TypeId },
NonNullableUnwrapped {
name: String,
content: TypeId,
},
/// A recursive tag union that has an empty variant
/// Optimization: Represent the empty variant as null pointer => no memory usage & fast comparison
@ -1016,7 +1019,7 @@ fn add_tag_union<'a>(
})
.collect();
let typ = match layout {
let tag_union_type = match layout {
Layout::Union(union_layout) => {
use UnionLayout::*;
@ -1030,12 +1033,12 @@ fn add_tag_union<'a>(
.max(1);
let discriminant_offset = union_layout.tag_id_offset(env.target).unwrap();
RocType::TagUnion(RocTagUnion::NonRecursive {
RocTagUnion::NonRecursive {
name: name.clone(),
tags,
discriminant_size,
discriminant_offset,
})
}
}
// A recursive tag union (general case)
// e.g. `Expr : [Sym Str, Add Expr Expr]`
@ -1044,12 +1047,12 @@ fn add_tag_union<'a>(
Discriminant::from_number_of_tags(tags.len()).stack_size();
let discriminant_offset = union_layout.tag_id_offset(env.target).unwrap();
RocType::TagUnion(RocTagUnion::Recursive {
RocTagUnion::Recursive {
name: name.clone(),
tags,
discriminant_size,
discriminant_offset,
})
}
}
// A recursive tag union with just one constructor
// Optimization: No need to store a tag ID (the payload is "unwrapped")
@ -1091,41 +1094,42 @@ fn add_tag_union<'a>(
let (non_null_tag, non_null_payload) = non_null;
RocType::TagUnion(RocTagUnion::NullableUnwrapped {
RocTagUnion::NullableUnwrapped {
name: name.clone(),
null_tag,
non_null_tag,
non_null_payload: non_null_payload.unwrap(),
null_represents_first_tag,
})
}
}
}
}
Layout::Builtin(Builtin::Int(int_width)) => RocType::TagUnion(RocTagUnion::Enumeration {
Layout::Builtin(Builtin::Int(int_width)) => RocTagUnion::Enumeration {
name: name.clone(),
tags: tags.into_iter().map(|(tag_name, _)| tag_name).collect(),
size: int_width.stack_size(),
}),
},
Layout::Builtin(_)
| Layout::Struct { .. }
| Layout::Boxed(_)
| Layout::LambdaSet(_)
| Layout::RecursivePointer => {
// These must be single-tag unions. Generate ordinary nonrecursive
// tag unions for them, and let Rust do the unwrapping.
// These must be single-tag wrappers. Generate ordinary nonrecursive
// tag unions for them, and let the generator do any unwrapping.
//
// This should be a very rare use case, and it's not worth overcomplicating
// the rest of glue to make it do something different.
RocType::TagUnion(RocTagUnion::NonRecursive {
RocTagUnion::NonRecursive {
name: name.clone(),
tags,
// These actually have no discriminant, since there's only one tag.
discriminant_size: 1,
discriminant_offset: 0,
})
}
}
};
let typ = RocType::TagUnion(tag_union_type);
let type_id = types.add_named(name, typ, layout);
if is_recursive {