1b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner//===--- RewriteTest.cpp - Rewriter playground ----------------------------===// 2b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// 3b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// The LLVM Compiler Infrastructure 4b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// 5b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// This file is distributed under the University of Illinois Open Source 6b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// License. See LICENSE.TXT for details. 7b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// 8b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner//===----------------------------------------------------------------------===// 9b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// 10b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// This is a testbed. 11b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner// 12b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner//===----------------------------------------------------------------------===// 13b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner 14305c613af6cfc40e519c75d9d2c84c6fa9a841c0Ted Kremenek#include "clang/Rewrite/Frontend/Rewriters.h" 15b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner#include "clang/Lex/Preprocessor.h" 16305c613af6cfc40e519c75d9d2c84c6fa9a841c0Ted Kremenek#include "clang/Rewrite/Core/TokenRewriter.h" 17f54fce8ff8818cde6b421265f709319f5fea0844Eli Friedman#include "llvm/Support/raw_ostream.h" 18b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner 195f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnervoid clang::DoRewriteTest(Preprocessor &PP, raw_ostream* OS) { 20b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner SourceManager &SM = PP.getSourceManager(); 214e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie const LangOptions &LangOpts = PP.getLangOpts(); 22b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner 23cff9cc95de367a3aea885a7f8fee304fe2707b92Chris Lattner TokenRewriter Rewriter(SM.getMainFileID(), SM, LangOpts); 2499bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner 2599bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner // Throw <i> </i> tags around comments. 2699bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner for (TokenRewriter::token_iterator I = Rewriter.token_begin(), 2799bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner E = Rewriter.token_end(); I != E; ++I) { 2899bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner if (I->isNot(tok::comment)) continue; 2999bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner 3099bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner Rewriter.AddTokenBefore(I, "<i>"); 3199bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner Rewriter.AddTokenAfter(I, "</i>"); 32b13c5eef3634e32075511d11ded4b699c17fb0d7Chris Lattner } 331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3599bd46c018ece10d6541a4b6bf0dbe97ad162477Chris Lattner // Print out the output. 36cff9cc95de367a3aea885a7f8fee304fe2707b92Chris Lattner for (TokenRewriter::token_iterator I = Rewriter.token_begin(), 37cff9cc95de367a3aea885a7f8fee304fe2707b92Chris Lattner E = Rewriter.token_end(); I != E; ++I) 38f54fce8ff8818cde6b421265f709319f5fea0844Eli Friedman *OS << PP.getSpelling(*I); 3985b489739fbc2d1ab430aea311ab381b4ee55848Matthijs Kooijman} 40