1 //===--- RewriteTest.cpp - Rewriter playground ----------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This is a testbed.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/Rewrite/Frontend/Rewriters.h"
15 #include "clang/Lex/Preprocessor.h"
16 #include "clang/Rewrite/Core/TokenRewriter.h"
17 #include "llvm/Support/raw_ostream.h"
18 
DoRewriteTest(Preprocessor & PP,raw_ostream * OS)19 void clang::DoRewriteTest(Preprocessor &PP, raw_ostream* OS) {
20   SourceManager &SM = PP.getSourceManager();
21   const LangOptions &LangOpts = PP.getLangOpts();
22 
23   TokenRewriter Rewriter(SM.getMainFileID(), SM, LangOpts);
24 
25   // Throw <i> </i> tags around comments.
26   for (TokenRewriter::token_iterator I = Rewriter.token_begin(),
27        E = Rewriter.token_end(); I != E; ++I) {
28     if (I->isNot(tok::comment)) continue;
29 
30     Rewriter.AddTokenBefore(I, "<i>");
31     Rewriter.AddTokenAfter(I, "</i>");
32   }
33 
34 
35   // Print out the output.
36   for (TokenRewriter::token_iterator I = Rewriter.token_begin(),
37        E = Rewriter.token_end(); I != E; ++I)
38     *OS << PP.getSpelling(*I);
39 }
40