From d12ad77e40134bd763f882b022c9203ceec4d76a Mon Sep 17 00:00:00 2001
From: Chris Lattner <sabre@nondot.org>
Date: Wed, 21 Jun 2006 03:49:01 +0000
Subject: [PATCH] Add really early support for emitting # line directives, and
 emitting the right number of newlines between tokens when needed.  This
 reduces the delta of the gcc -E output from 12198 differences to 6764.  It
 still needs to emit filenames on #line directives, track filename switches,
 and track entry/exit of include files.

llvm-svn: 38559
---
 clang/Driver/clang.cpp | 61 +++++++++++++++++++++++++++---------------
 1 file changed, 40 insertions(+), 21 deletions(-)

diff --git a/clang/Driver/clang.cpp b/clang/Driver/clang.cpp
index 78fbba05d96f..44ac2ded60e1 100644
--- a/clang/Driver/clang.cpp
+++ b/clang/Driver/clang.cpp
@@ -588,11 +588,49 @@ static void ReadPrologFiles(Preprocessor &PP, std::vector<char> &Buf) {
 // Preprocessed output mode.
 //===----------------------------------------------------------------------===//
 
+/// HandleFirstTokOnLine - When emitting a preprocessed file in -E mode, this
+/// is called for the first token on each new line.
+static void HandleFirstTokOnLine(LexerToken &Tok, Preprocessor &PP,
+                                 unsigned &CurLine) {
+  // Figure out what line we went to and insert the appropriate number of
+  // newline characters.
+  unsigned LineNo = PP.getSourceManager().getLineNumber(Tok.getLocation());
+  
+  // If this line is "close enough" to the original line, just print newlines,
+  // otherwise print a #line directive.
+  if (LineNo-CurLine < 8) {
+    for (; CurLine != LineNo; ++CurLine)
+      std::cout << "\n";
+  } else {
+    // FIXME: filename too.
+    std::cout << "\n# " << LineNo << "\n";
+    CurLine = LineNo;
+  }
+  
+  // Print out space characters so that the first token on a line is
+  // indented for easy reading.
+  unsigned ColNo = 
+    PP.getSourceManager().getColumnNumber(Tok.getLocation());
+  
+  // This hack prevents stuff like:
+  // #define HASH #
+  // HASH define foo bar
+  // From having the # character end up at column 1, which makes it so it
+  // is not handled as a #define next time through the preprocessor if in
+  // -fpreprocessed mode.
+  if (ColNo <= 1 && Tok.getKind() == tok::hash)
+    std::cout << ' ';
+  
+  // Otherwise, indent the appropriate number of spaces.
+  for (; ColNo > 1; --ColNo)
+    std::cout << ' ';
+}
+
 /// DoPrintPreprocessedInput - This implements -E mode.
 void DoPrintPreprocessedInput(Preprocessor &PP) {
   LexerToken Tok;
   char Buffer[256];
-  bool isFirstToken = true;
+  unsigned CurLine = 1;
   do {
     PP.Lex(Tok);
 
@@ -602,29 +640,10 @@ void DoPrintPreprocessedInput(Preprocessor &PP) {
     // FIXME: For some tests, this fails just because there is no col# info from
     // macro expansions!
     if (Tok.isAtStartOfLine()) {
-      if (!isFirstToken)
-        std::cout << "\n";
-      // Print out space characters so that the first token on a line is
-      // indented for easy reading.
-      unsigned ColNo = 
-        PP.getSourceManager().getColumnNumber(Tok.getLocation());
-      
-      // This hack prevents stuff like:
-      // #define HASH #
-      // HASH define foo bar
-      // From having the # character end up at column 1, which makes it so it
-      // is not handled as a #define next time through the preprocessor if in
-      // -fpreprocessed mode.
-      if (ColNo <= 1 && Tok.getKind() == tok::hash)
-        std::cout << ' ';
-      
-      for (; ColNo > 1; --ColNo)
-        std::cout << ' ';
-      
+      HandleFirstTokOnLine(Tok, PP, CurLine);
     } else if (Tok.hasLeadingSpace()) {
       std::cout << ' ';
     }
-    isFirstToken = false;    
     
     if (Tok.getLength() < 256) {
       unsigned Len = PP.getSpelling(Tok, Buffer);