Print function argument layouts in pretty mode

This commit is contained in:
Ayaz Hafiz 2022-08-13 08:46:07 -07:00
parent 991b99250c
commit 9ffe6eb6dd
No known key found for this signature in database
GPG key ID: 0E2A37416A25EF58

View file

@ -343,10 +343,16 @@ impl<'a> Proc<'a> {
D::Doc: Clone,
A: Clone,
{
let args_doc = self
.args
.iter()
.map(|(_, symbol)| symbol_to_doc(alloc, *symbol));
let args_doc = self.args.iter().map(|(layout, symbol)| {
let arg_doc = symbol_to_doc(alloc, *symbol);
if pretty_print_ir_symbols() {
arg_doc
.append(alloc.reflow(": "))
.append(layout.to_doc(alloc, Parens::NotNeeded))
} else {
arg_doc
}
});
if pretty_print_ir_symbols() {
alloc