diff options
| author | Nicholas Nethercote <nnethercote@mozilla.com> | 2019-03-29 08:32:13 +1100 |
|---|---|---|
| committer | Nicholas Nethercote <nnethercote@mozilla.com> | 2019-04-01 07:55:25 +1100 |
| commit | 606f3158bf7dd2708b1efdde2e6de34f8eaebd7b (patch) | |
| tree | 19d61e538bb17a42f2a5a45ad54ade119976f0e4 /src | |
| parent | 4c27fb19ba15a2e45485e601a79914c6280196b0 (diff) | |
| download | rust-606f3158bf7dd2708b1efdde2e6de34f8eaebd7b.tar.gz rust-606f3158bf7dd2708b1efdde2e6de34f8eaebd7b.zip | |
Optimize indentation in the pretty printer.
Currently the pretty-printer calls `write!` for every space of indentation. On some workloads the indentation level can exceed 100, and a faster implementation reduces instruction counts by up to 7% on a few workloads.
Diffstat (limited to 'src')
| -rw-r--r-- | src/libsyntax/print/pp.rs | 22 |
1 files changed, 19 insertions, 3 deletions
diff --git a/src/libsyntax/print/pp.rs b/src/libsyntax/print/pp.rs index d8a8cbb655b..45eb6995a76 100644 --- a/src/libsyntax/print/pp.rs +++ b/src/libsyntax/print/pp.rs @@ -300,6 +300,8 @@ impl Default for BufEntry { } } +const SPACES: [u8; 128] = [b' '; 128]; + impl<'a> Printer<'a> { pub fn last_token(&mut self) -> Token { self.buf[self.right].token.clone() @@ -580,10 +582,24 @@ impl<'a> Printer<'a> { debug!("print String({})", s); // assert!(len <= space); self.space -= len; - while self.pending_indentation > 0 { - write!(self.out, " ")?; - self.pending_indentation -= 1; + + // Write the pending indent. A more concise way of doing this would be: + // + // write!(self.out, "{: >n$}", "", n = self.pending_indentation as usize)?; + // + // But that is significantly slower than using `SPACES`. This code is + // sufficiently hot, and indents can get sufficiently large, that the + // difference is significant on some workloads. + let spaces_len = SPACES.len() as isize; + while self.pending_indentation >= spaces_len { + self.out.write_all(&SPACES)?; + self.pending_indentation -= spaces_len; } + if self.pending_indentation > 0 { + self.out.write_all(&SPACES[0..self.pending_indentation as usize])?; + self.pending_indentation = 0; + } + write!(self.out, "{}", s) } |
