Added the beginning of a set of rewriter functions for pretty-printing source

code as HTML.  Added a new driver option "--emit-html" to dump the source
of the main input file as HTML.

llvm-svn: 48505

clang/Driver/ASTConsumers.h

@@ -53,6 +53,9 @@ ASTConsumer *CreateCodeRewriterTest(const std::string& InFile,
                                     Diagnostic &Diags,
                                     const LangOptions &LOpts);
 
+ASTConsumer* CreateHTMLPrinter();
+
+
 ASTConsumer *CreateSerializationTest(Diagnostic &Diags,
                                      FileManager& FMgr, 
                                      const LangOptions &LOpts);

clang/Driver/clang.cpp

@@ -63,6 +63,7 @@ enum ProgActions {
   EmitLLVM,                     // Emit a .ll file.
   EmitBC,                       // Emit a .bc file.
   SerializeAST,                 // Emit a .ast file.
+  EmitHTML,                     // Translate input source into HTML.
   ASTPrint,                     // Parse ASTs and print them.
   ASTDump,                      // Parse ASTs and dump them.
   ASTView,                      // Parse ASTs and view them in Graphviz.
@@ -100,6 +101,8 @@ ProgAction(llvm::cl::desc("Choose output type:"), llvm::cl::ZeroOrMore,
                         "Run parser and perform semantic analysis"),
              clEnumValN(ParsePrintCallbacks, "parse-print-callbacks",
                         "Run parser and print each callback invoked"),
+             clEnumValN(EmitHTML, "emit-html",
+                        "Output input source as HTML"),
              clEnumValN(ASTPrint, "ast-print",
                         "Build ASTs and then pretty-print them"),
              clEnumValN(ASTDump, "ast-dump",
@@ -994,7 +997,10 @@ static ASTConsumer* CreateASTConsumer(const std::string& InFile,
       return CreateASTDumper();
       
     case ASTView:
-      return CreateASTViewer();      
+      return CreateASTViewer();   
+      
+    case EmitHTML:
+      return CreateHTMLPrinter();
       
     case ParseCFGDump:
     case ParseCFGView:

clang/include/clang/Rewrite/HTMLRewrite.h

@@ -0,0 +1,37 @@
+//==- HTMLRewrite.h - Translate source code into prettified HTML ---*- C++ -*-//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+//  This file defines a set of functions used for translating source code
+//  into beautified HTML.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_CLANG_HTMLREWRITER_H
+#define LLVM_CLANG_HTMLREWRITER_H
+
+#include "clang/Basic/SourceLocation.h"
+
+namespace clang {
+  
+class Rewriter;
+  
+namespace html {
+  
+  enum Tags { PRE, HEAD, BODY };
+  
+  void EscapeText(Rewriter& R, unsigned FileID, bool EscapeSpaces = false);
+
+  void InsertTag(Rewriter& R, Tags tag,
+                 SourceLocation OpenLoc, SourceLocation CloseLoc,
+                 bool NewlineOpen = false, bool NewlineClose = true);
+
+} // end html namespace
+} // end clang namespace
+
+#endif

clang/lib/Rewrite/HTMLRewrite.cpp

@@ -0,0 +1,77 @@
+//== HTMLRewrite.cpp - Translate source code into prettified HTML --*- C++ -*-//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+//  This file defines the HTMLRewriter clas, which is used to translate the
+//  text of a source file into prettified HTML.
+//
+//===----------------------------------------------------------------------===//
+
+#include "clang/Rewrite/Rewriter.h"
+#include "clang/Rewrite/HTMLRewrite.h"
+#include "clang/Basic/SourceManager.h"
+#include "llvm/Support/MemoryBuffer.h"
+#include <sstream>
+
+using namespace clang;
+
+void html::EscapeText(Rewriter& R, unsigned FileID, bool EscapeSpaces) {
+  
+  const llvm::MemoryBuffer *Buf = R.getSourceMgr().getBuffer(FileID);
+  const char* C = Buf->getBufferStart();
+  const char* FileEnd = Buf->getBufferEnd();
+  
+  assert (C <= FileEnd);
+  
+  for (unsigned FilePos = 0; C != FileEnd ; ++C, ++FilePos) {
+    
+    SourceLocation Loc = SourceLocation::getFileLoc(FileID, FilePos);
+  
+    switch (*C) {
+      default: break;
+        
+      case ' ':
+        if (EscapeSpaces) R.ReplaceText(Loc, 1, "&#32;", 5);
+        break;
+
+      case '<': R.ReplaceText(Loc, 1, "&lt;", 4); break;
+      case '>': R.ReplaceText(Loc, 1, "&gt;", 4); break;
+      case '&': R.ReplaceText(Loc, 1, "&amp;", 5); break;
+    }
+  }
+}
+
+void html::InsertTag(Rewriter& R, html::Tags tag,
+                     SourceLocation B, SourceLocation E,
+                     bool NewlineOpen, bool NewlineClose) {
+  
+  const char* TagStr = 0;
+  
+  switch (tag) {
+    default: break;      
+    case PRE: TagStr = "pre"; break;
+    case HEAD: TagStr = "head"; break;
+    case BODY: TagStr = "body"; break;
+  }
+  
+  assert (TagStr && "Tag not supported.");
+
+  { // Generate the opening tag.
+    std::ostringstream os;  
+    os << '<' << TagStr << '>';
+    if (NewlineOpen) os << '\n';
+    R.InsertTextAfter(B, os.str().c_str(), os.str().size());
+  }
+  
+  { // Generate the closing tag.
+    std::ostringstream os;  
+    os << "</" << TagStr << '>';
+    if (NewlineClose) os << '\n';
+    R.InsertTextBefore(E, os.str().c_str(), os.str().size());
+  }
+}