add scafolding to play around with and bring up the code rewriter.

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@42855 91177308-0d34-0410-b5e6-96231b3b80d8

3 files changed, 51 insertions, 0 deletions

diff --git a/Driver/ASTConsumers.h b/Driver/ASTConsumers.h
index 4265672b95..90b4f9735a 100644
--- a/Driver/ASTConsumers.h
+++ b/Driver/ASTConsumers.h
@@ -27,6 +27,7 @@ ASTConsumer *CreateLiveVarAnalyzer();
 ASTConsumer *CreateDeadStoreChecker(Diagnostic &Diags);
 ASTConsumer *CreateUnitValsChecker(Diagnostic &Diags);
 ASTConsumer *CreateLLVMEmitter(Diagnostic &Diags);
+ASTConsumer *CreateCodeRewriterTest();
 
 } // end clang namespace
 
diff --git a/Driver/RewriteTest.cpp b/Driver/RewriteTest.cpp
new file mode 100644
index 0000000000..ef1a9b0822
--- /dev/null
+++ b/Driver/RewriteTest.cpp
@@ -0,0 +1,43 @@
+//===--- RewriteTest.cpp - Playground for the code rewriter ---------------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file was developed by Chris Lattner and is distributed under the
+// University of Illinois Open Source License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// Hacks and fun related to the code rewriter.
+//
+//===----------------------------------------------------------------------===//
+
+#include "ASTConsumers.h"
+#include "clang/AST/AST.h"
+#include "clang/AST/ASTConsumer.h"
+
+using namespace clang;
+
+
+namespace {
+  class ASTViewer : public ASTConsumer {
+    SourceManager *SM;
+  public:
+    void Initialize(ASTContext &Context, unsigned MainFileID) {
+      SM = &Context.SourceMgr;
+    }
+    
+    virtual void HandleTopLevelDecl(Decl *D);
+  };
+}
+
+ASTConsumer *clang::CreateCodeRewriterTest() { return new ASTViewer(); }
+
+
+
+
+void ASTViewer::HandleTopLevelDecl(Decl *D) {
+  if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
+    if (ND->getName())
+      printf("%s\n", ND->getName());
+  
+}
diff --git a/Driver/clang.cpp b/Driver/clang.cpp
index 9d8f71288f..6d9b6b0e46 100644
--- a/Driver/clang.cpp
+++ b/Driver/clang.cpp
@@ -49,6 +49,7 @@ static llvm::cl::opt<bool>
 Stats("stats", llvm::cl::desc("Print performance metrics and statistics"));
 
 enum ProgActions {
+  RewriteTest,                  // Rewriter testing stuff.
   EmitLLVM,                     // Emit a .ll file.
   ASTPrint,                     // Parse ASTs and print them.
   ASTDump,                      // Parse ASTs and dump them.
@@ -101,6 +102,8 @@ ProgAction(llvm::cl::desc("Choose output type:"), llvm::cl::ZeroOrMore,
                         "Flag warnings of uses of unitialized variables."),
              clEnumValN(EmitLLVM, "emit-llvm",
                         "Build ASTs then convert to LLVM, emit .ll file"),
+             clEnumValN(RewriteTest, "rewrite-test",
+                        "Playground for the code rewriter"),
              clEnumValEnd));
 
 static llvm::cl::opt<bool>
@@ -796,6 +799,10 @@ static void ProcessInputFile(Preprocessor &PP, unsigned MainFileID,
   case EmitLLVM:
     Consumer = CreateLLVMEmitter(PP.getDiagnostics());
     break;
+    
+  case RewriteTest:
+    Consumer = CreateCodeRewriterTest();
+    break;
   }
   
   if (Consumer) {