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
19void 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