Always show one level of unions when printing layouts

This commit is contained in:
Ayaz Hafiz 2023-01-25 15:16:06 -06:00
parent 3382be6cb9
commit 741b1a1bd5
No known key found for this signature in database
GPG key ID: 0E2A37416A25EF58
5 changed files with 180 additions and 124 deletions

View file

@ -344,11 +344,9 @@ impl<'a> Proc<'a> {
let args_doc = self.args.iter().map(|(layout, symbol)| {
let arg_doc = symbol_to_doc(alloc, *symbol, pretty);
if pretty_print_ir_symbols() {
arg_doc.append(alloc.reflow(": ")).append(interner.to_doc(
*layout,
alloc,
Parens::NotNeeded,
))
arg_doc
.append(alloc.reflow(": "))
.append(interner.to_doc_top(*layout, alloc))
} else {
arg_doc
}
@ -359,7 +357,7 @@ impl<'a> Proc<'a> {
.text("procedure : ")
.append(symbol_to_doc(alloc, self.name.name(), pretty))
.append(" ")
.append(interner.to_doc(self.ret_layout, alloc, Parens::NotNeeded))
.append(interner.to_doc_top(self.ret_layout, alloc))
.append(alloc.hardline())
.append(alloc.text("procedure = "))
.append(symbol_to_doc(alloc, self.name.name(), pretty))
@ -2152,7 +2150,7 @@ impl<'a> Stmt<'a> {
.text("let ")
.append(symbol_to_doc(alloc, *symbol, pretty))
.append(" : ")
.append(interner.to_doc(*layout, alloc, Parens::NotNeeded))
.append(interner.to_doc_top(*layout, alloc))
.append(" = ")
.append(expr.to_doc(alloc, pretty))
.append(";")