about summary refs log tree commit diff
path: root/src
diff options
context:
space:
mode:
authorNicholas Nethercote <nnethercote@mozilla.com>2019-03-29 08:32:13 +1100
committerNicholas Nethercote <nnethercote@mozilla.com>2019-04-01 07:55:25 +1100
commit606f3158bf7dd2708b1efdde2e6de34f8eaebd7b (patch)
tree19d61e538bb17a42f2a5a45ad54ade119976f0e4 /src
parent4c27fb19ba15a2e45485e601a79914c6280196b0 (diff)
downloadrust-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.rs22
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)
     }