HTMLRewrite.cpp revision a11d61793341fea195c29a0dab3fbd74f2b39a8c
16a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek//== HTMLRewrite.cpp - Translate source code into prettified HTML --*- C++ -*-// 26a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// 36a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// The LLVM Compiler Infrastructure 46a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// 56a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// This file is distributed under the University of Illinois Open Source 66a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// License. See LICENSE.TXT for details. 76a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// 86a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek//===----------------------------------------------------------------------===// 96a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// 106a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// This file defines the HTMLRewriter clas, which is used to translate the 116a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// text of a source file into prettified HTML. 126a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek// 136a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek//===----------------------------------------------------------------------===// 146a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek 15339b9c27759d7b6a53e2370f83f66e78b3254595Ted Kremenek#include "clang/Lex/Preprocessor.h" 166a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek#include "clang/Rewrite/Rewriter.h" 176a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek#include "clang/Rewrite/HTMLRewrite.h" 183245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner#include "clang/Lex/Preprocessor.h" 196a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek#include "clang/Basic/SourceManager.h" 2057df3b950061c73d13d3116f747e79d7955a216aChris Lattner#include "llvm/ADT/SmallString.h" 21339b9c27759d7b6a53e2370f83f66e78b3254595Ted Kremenek#include "llvm/ADT/OwningPtr.h" 226a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek#include "llvm/Support/MemoryBuffer.h" 23a95d3750441ac8ad03e36af8e6e74039c9a3109dTed Kremenek#include "llvm/Support/raw_ostream.h" 246a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenekusing namespace clang; 256a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek 269402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner 275ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner/// HighlightRange - Highlight a range in the source code with the specified 285ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner/// start/end tags. B/E must be in the same file. This ensures that 295ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner/// start/end tags are placed at the start/end of each line if the range is 305ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner/// multiline. 315ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattnervoid html::HighlightRange(Rewriter &R, SourceLocation B, SourceLocation E, 325ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner const char *StartTag, const char *EndTag) { 335ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner SourceManager &SM = R.getSourceMgr(); 34f7cf85b330bedd2877e1371fb0a83e99751ae162Chris Lattner B = SM.getInstantiationLoc(B); 35f7cf85b330bedd2877e1371fb0a83e99751ae162Chris Lattner E = SM.getInstantiationLoc(E); 36a11d61793341fea195c29a0dab3fbd74f2b39a8cChris Lattner FileID FID = SM.getFileID(B); 37a11d61793341fea195c29a0dab3fbd74f2b39a8cChris Lattner assert(SM.getFileID(E) == FID && "B/E not in the same file!"); 385ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner 395ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner unsigned BOffset = SM.getFullFilePos(B); 405ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner unsigned EOffset = SM.getFullFilePos(E); 415ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner 425ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner // Include the whole end token in the range. 435ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner EOffset += Lexer::MeasureTokenLength(E, R.getSourceMgr()); 445ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner 452b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner HighlightRange(R.getEditBuffer(FID), BOffset, EOffset, 462b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner SM.getBufferData(FID).first, StartTag, EndTag); 475ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner} 485ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner 495ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner/// HighlightRange - This is the same as the above method, but takes 505ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner/// decomposed file locations. 515ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattnervoid html::HighlightRange(RewriteBuffer &RB, unsigned B, unsigned E, 525ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner const char *BufferStart, 535ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner const char *StartTag, const char *EndTag) { 549402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // Insert the tag at the absolute start/end of the range. 555ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner RB.InsertTextAfter(B, StartTag, strlen(StartTag)); 565ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner RB.InsertTextBefore(E, EndTag, strlen(EndTag)); 575ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner 589402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // Scan the range to see if there is a \r or \n. If so, and if the line is 599402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // not blank, insert tags on that line as well. 609402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner bool HadOpenTag = true; 619402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner 629402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner unsigned LastNonWhiteSpace = B; 639402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner for (unsigned i = B; i != E; ++i) { 649402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner switch (BufferStart[i]) { 659402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner case '\r': 669402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner case '\n': 679402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // Okay, we found a newline in the range. If we have an open tag, we need 689402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // to insert a close tag at the first non-whitespace before the newline. 699402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner if (HadOpenTag) 709402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner RB.InsertTextBefore(LastNonWhiteSpace+1, EndTag, strlen(EndTag)); 719402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner 729402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // Instead of inserting an open tag immediately after the newline, we 739402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // wait until we see a non-whitespace character. This prevents us from 749402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // inserting tags around blank lines, and also allows the open tag to 759402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // be put *after* whitespace on a non-blank line. 769402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner HadOpenTag = false; 779402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner break; 789402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner case '\0': 799402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner case ' ': 809402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner case '\t': 819402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner case '\f': 829402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner case '\v': 839402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // Ignore whitespace. 849402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner break; 859402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner 869402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner default: 879402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // If there is no tag open, do it now. 889402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner if (!HadOpenTag) { 899402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner RB.InsertTextAfter(i, StartTag, strlen(StartTag)); 909402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner HadOpenTag = true; 919402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner } 929402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner 939402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner // Remember this character. 949402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner LastNonWhiteSpace = i; 959402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner break; 969402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner } 979402b57a0dca4058fe56d7fd84e97fc496421125Chris Lattner } 985ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner} 995ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner 1002b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattnervoid html::EscapeText(Rewriter &R, FileID FID, 101fa5be3617294f0e3c341f0ecb6b2076478b1b5acTed Kremenek bool EscapeSpaces, bool ReplaceTabs) { 1026a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek 1032b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner const llvm::MemoryBuffer *Buf = R.getSourceMgr().getBuffer(FID); 1046a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek const char* C = Buf->getBufferStart(); 1056a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek const char* FileEnd = Buf->getBufferEnd(); 1066a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek 1076a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek assert (C <= FileEnd); 1086a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek 1092b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner RewriteBuffer &RB = R.getEditBuffer(FID); 1105c176f7a9ba9f7084b903393845be24f85e091daChris Lattner 1115c176f7a9ba9f7084b903393845be24f85e091daChris Lattner unsigned ColNo = 0; 1126a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek for (unsigned FilePos = 0; C != FileEnd ; ++C, ++FilePos) { 1136a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek switch (*C) { 1145c176f7a9ba9f7084b903393845be24f85e091daChris Lattner default: ++ColNo; break; 1155c176f7a9ba9f7084b903393845be24f85e091daChris Lattner case '\n': 1165c176f7a9ba9f7084b903393845be24f85e091daChris Lattner case '\r': 1175c176f7a9ba9f7084b903393845be24f85e091daChris Lattner ColNo = 0; 1185c176f7a9ba9f7084b903393845be24f85e091daChris Lattner break; 119735271479ac57c27f744806859efd5b001dea248Chris Lattner 120735271479ac57c27f744806859efd5b001dea248Chris Lattner case ' ': 121735271479ac57c27f744806859efd5b001dea248Chris Lattner if (EscapeSpaces) 122735271479ac57c27f744806859efd5b001dea248Chris Lattner RB.ReplaceText(FilePos, 1, " ", 6); 1235c176f7a9ba9f7084b903393845be24f85e091daChris Lattner ++ColNo; 124735271479ac57c27f744806859efd5b001dea248Chris Lattner break; 125f3d8d19caf6f1a21785eee8d62c45ef5a0a0e72eChris Lattner case '\f': 126f3d8d19caf6f1a21785eee8d62c45ef5a0a0e72eChris Lattner RB.ReplaceText(FilePos, 1, "<hr>", 4); 127f3d8d19caf6f1a21785eee8d62c45ef5a0a0e72eChris Lattner ColNo = 0; 128f3d8d19caf6f1a21785eee8d62c45ef5a0a0e72eChris Lattner break; 129f3d8d19caf6f1a21785eee8d62c45ef5a0a0e72eChris Lattner 1305c176f7a9ba9f7084b903393845be24f85e091daChris Lattner case '\t': { 131735271479ac57c27f744806859efd5b001dea248Chris Lattner if (!ReplaceTabs) 13249cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek break; 1335c176f7a9ba9f7084b903393845be24f85e091daChris Lattner unsigned NumSpaces = 8-(ColNo&7); 134735271479ac57c27f744806859efd5b001dea248Chris Lattner if (EscapeSpaces) 1358aa06aca8b2d3771a5405d789b2e704149045dd4Chris Lattner RB.ReplaceText(FilePos, 1, " " 1365c176f7a9ba9f7084b903393845be24f85e091daChris Lattner " ", 6*NumSpaces); 137735271479ac57c27f744806859efd5b001dea248Chris Lattner else 1385c176f7a9ba9f7084b903393845be24f85e091daChris Lattner RB.ReplaceText(FilePos, 1, " ", NumSpaces); 1395c176f7a9ba9f7084b903393845be24f85e091daChris Lattner ColNo += NumSpaces; 140735271479ac57c27f744806859efd5b001dea248Chris Lattner break; 1415c176f7a9ba9f7084b903393845be24f85e091daChris Lattner } 142735271479ac57c27f744806859efd5b001dea248Chris Lattner case '<': 143735271479ac57c27f744806859efd5b001dea248Chris Lattner RB.ReplaceText(FilePos, 1, "<", 4); 1445c176f7a9ba9f7084b903393845be24f85e091daChris Lattner ++ColNo; 145735271479ac57c27f744806859efd5b001dea248Chris Lattner break; 146735271479ac57c27f744806859efd5b001dea248Chris Lattner 147735271479ac57c27f744806859efd5b001dea248Chris Lattner case '>': 148735271479ac57c27f744806859efd5b001dea248Chris Lattner RB.ReplaceText(FilePos, 1, ">", 4); 1495c176f7a9ba9f7084b903393845be24f85e091daChris Lattner ++ColNo; 150735271479ac57c27f744806859efd5b001dea248Chris Lattner break; 151735271479ac57c27f744806859efd5b001dea248Chris Lattner 152735271479ac57c27f744806859efd5b001dea248Chris Lattner case '&': 153735271479ac57c27f744806859efd5b001dea248Chris Lattner RB.ReplaceText(FilePos, 1, "&", 5); 1545c176f7a9ba9f7084b903393845be24f85e091daChris Lattner ++ColNo; 155735271479ac57c27f744806859efd5b001dea248Chris Lattner break; 1566a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek } 1576a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek } 1586a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek} 1596a34083e9f74a45e2f79c9fab66f177809a5db66Ted Kremenek 160fa5be3617294f0e3c341f0ecb6b2076478b1b5acTed Kremenekstd::string html::EscapeText(const std::string& s, bool EscapeSpaces, 161fa5be3617294f0e3c341f0ecb6b2076478b1b5acTed Kremenek bool ReplaceTabs) { 162053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek 163053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek unsigned len = s.size(); 164a95d3750441ac8ad03e36af8e6e74039c9a3109dTed Kremenek std::string Str; 165a95d3750441ac8ad03e36af8e6e74039c9a3109dTed Kremenek llvm::raw_string_ostream os(Str); 166053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek 167053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek for (unsigned i = 0 ; i < len; ++i) { 168053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek 169053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek char c = s[i]; 170053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek switch (c) { 1718570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner default: 1728570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner os << c; break; 1738570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner 1748570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner case ' ': 1758570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner if (EscapeSpaces) os << " "; 1768570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner else os << ' '; 1778570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner break; 1788570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner 1798570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner case '\t': 18063366303f6cf02ea65ac9ad9000e3e8467699234Nico Weber if (ReplaceTabs) { 18163366303f6cf02ea65ac9ad9000e3e8467699234Nico Weber if (EscapeSpaces) 18263366303f6cf02ea65ac9ad9000e3e8467699234Nico Weber for (unsigned i = 0; i < 4; ++i) 18363366303f6cf02ea65ac9ad9000e3e8467699234Nico Weber os << " "; 18463366303f6cf02ea65ac9ad9000e3e8467699234Nico Weber else 18563366303f6cf02ea65ac9ad9000e3e8467699234Nico Weber for (unsigned i = 0; i < 4; ++i) 18663366303f6cf02ea65ac9ad9000e3e8467699234Nico Weber os << " "; 18763366303f6cf02ea65ac9ad9000e3e8467699234Nico Weber } 1888570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner else 1898570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner os << c; 1908570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner 191053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek break; 1928570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner 1938570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner case '<': os << "<"; break; 1948570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner case '>': os << ">"; break; 1958570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner case '&': os << "&"; break; 196053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek } 197053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek } 198053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek 199053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek return os.str(); 200053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek} 201053ef593fa9d2b890645a914eee203231fb34458Ted Kremenek 2028570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattnerstatic void AddLineNumber(RewriteBuffer &RB, unsigned LineNo, 2038570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner unsigned B, unsigned E) { 20457df3b950061c73d13d3116f747e79d7955a216aChris Lattner llvm::SmallString<100> Str; 20557df3b950061c73d13d3116f747e79d7955a216aChris Lattner Str += "<tr><td class=\"num\" id=\"LN"; 20657df3b950061c73d13d3116f747e79d7955a216aChris Lattner Str.append_uint(LineNo); 20757df3b950061c73d13d3116f747e79d7955a216aChris Lattner Str += "\">"; 20857df3b950061c73d13d3116f747e79d7955a216aChris Lattner Str.append_uint(LineNo); 20957df3b950061c73d13d3116f747e79d7955a216aChris Lattner Str += "</td><td class=\"line\">"; 21057df3b950061c73d13d3116f747e79d7955a216aChris Lattner 21149cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek if (B == E) { // Handle empty lines. 21257df3b950061c73d13d3116f747e79d7955a216aChris Lattner Str += " </td></tr>"; 2138570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner RB.InsertTextBefore(B, &Str[0], Str.size()); 21457df3b950061c73d13d3116f747e79d7955a216aChris Lattner } else { 2158570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner RB.InsertTextBefore(B, &Str[0], Str.size()); 2168570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner RB.InsertTextBefore(E, "</td></tr>", strlen("</td></tr>")); 21749cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek } 218b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek} 219b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 2202b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattnervoid html::AddLineNumbers(Rewriter& R, FileID FID) { 221b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 2222b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner const llvm::MemoryBuffer *Buf = R.getSourceMgr().getBuffer(FID); 223b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek const char* FileBeg = Buf->getBufferStart(); 224b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek const char* FileEnd = Buf->getBufferEnd(); 225b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek const char* C = FileBeg; 2262b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner RewriteBuffer &RB = R.getEditBuffer(FID); 227b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 228b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek assert (C <= FileEnd); 229b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 230b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek unsigned LineNo = 0; 231b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek unsigned FilePos = 0; 232b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 233b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek while (C != FileEnd) { 234b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 235b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek ++LineNo; 236b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek unsigned LineStartPos = FilePos; 237b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek unsigned LineEndPos = FileEnd - FileBeg; 238b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 239b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek assert (FilePos <= LineEndPos); 240b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek assert (C < FileEnd); 241b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 242b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek // Scan until the newline (or end-of-file). 243b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 24449cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek while (C != FileEnd) { 24549cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek char c = *C; 24649cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek ++C; 24749cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek 24849cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek if (c == '\n') { 24949cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek LineEndPos = FilePos++; 250b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek break; 251b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek } 25249cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek 25349cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek ++FilePos; 25449cd6354d5373245dd2e69ca7b7113e6a795d36eTed Kremenek } 255b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek 2568570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner AddLineNumber(RB, LineNo, LineStartPos, LineEndPos); 257d6c1360c2bf234c73572a865f119d0518aca8154Ted Kremenek } 258d6c1360c2bf234c73572a865f119d0518aca8154Ted Kremenek 2598570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner // Add one big table tag that surrounds all of the code. 2608570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner RB.InsertTextBefore(0, "<table class=\"code\">\n", 2618570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner strlen("<table class=\"code\">\n")); 262d6c1360c2bf234c73572a865f119d0518aca8154Ted Kremenek 2638570f0b0fde7ca812f8d37f52305f3df4dd2ce01Chris Lattner RB.InsertTextAfter(FileEnd - FileBeg, "</table>", strlen("</table>")); 264b485cd1e0a5a1e942d0e682b9b1c4bc9df111528Ted Kremenek} 265ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek 2662b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattnervoid html::AddHeaderFooterInternalBuiltinCSS(Rewriter& R, FileID FID, 267f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek const char *title) { 268ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek 2692b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner const llvm::MemoryBuffer *Buf = R.getSourceMgr().getBuffer(FID); 270ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek const char* FileStart = Buf->getBufferStart(); 271ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek const char* FileEnd = Buf->getBufferEnd(); 272ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek 2732b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner SourceLocation StartLoc = R.getSourceMgr().getLocForStartOfFile(FID); 2742b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner SourceLocation EndLoc = StartLoc.getFileLocWithOffset(FileEnd-FileStart); 275ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek 276a95d3750441ac8ad03e36af8e6e74039c9a3109dTed Kremenek std::string s; 277a95d3750441ac8ad03e36af8e6e74039c9a3109dTed Kremenek llvm::raw_string_ostream os(s); 278f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek os << "<!doctype html>\n" // Use HTML 5 doctype 279f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek "<html>\n<head>\n"; 280f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek 281f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek if (title) 282f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek os << "<title>" << html::EscapeText(title) << "</title>\n"; 283f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek 284f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek os << "<style type=\"text/css\">\n" 28570bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " body { color:#000000; background-color:#ffffff }\n" 28670bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " body { font-family:Helvetica, sans-serif; font-size:10pt }\n" 2874b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " h1 { font-size:14pt }\n" 288f5016260522e449e9bacdb0e5a87ad3932d9fbd4Ted Kremenek " .code { border-collapse:collapse; width:100%; }\n" 28970bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .code { font-family: \"Andale Mono\", monospace; font-size:10pt }\n" 29070bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .code { line-height: 1.2em }\n" 291f5016260522e449e9bacdb0e5a87ad3932d9fbd4Ted Kremenek " .comment { color: green; font-style: oblique }\n" 292f5016260522e449e9bacdb0e5a87ad3932d9fbd4Ted Kremenek " .keyword { color: blue }\n" 293cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek " .string_literal { color: red }\n" 294f5016260522e449e9bacdb0e5a87ad3932d9fbd4Ted Kremenek " .directive { color: darkmagenta }\n" 2956f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner // Macro expansions. 29607339a63b46e38c954fcccbef721c609d0c2040eTed Kremenek " .expansion { display: none; }\n" 29707339a63b46e38c954fcccbef721c609d0c2040eTed Kremenek " .macro:hover .expansion { display: block; border: 2px solid #FF0000; " 298dc5be47542e6d4a28d20abf9c0f0a0edd72939b6Chris Lattner "padding: 2px; background-color:#FFF0F0; font-weight: normal; " 2996f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner " -webkit-border-radius:5px; -webkit-box-shadow:1px 1px 7px #000; " 3008aa06aca8b2d3771a5405d789b2e704149045dd4Chris Lattner "position: absolute; top: -1em; left:10em; z-index: 1 } \n" 301f5016260522e449e9bacdb0e5a87ad3932d9fbd4Ted Kremenek " .macro { color: darkmagenta; background-color:LemonChiffon;" 3026f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner // Macros are position: relative to provide base for expansions. 3036f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner " position: relative }\n" 30470bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .num { width:2.5em; padding-right:2ex; background-color:#eeeeee }\n" 30570bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .num { text-align:right; font-size: smaller }\n" 30670bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .num { color:#444444 }\n" 30770bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .line { padding-left: 1ex; border-left: 3px solid #ccc }\n" 30870bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .line { white-space: pre }\n" 30970bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .msg { background-color:#fff8b4; color:#000000 }\n" 31070bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .msg { -webkit-box-shadow:1px 1px 7px #000 }\n" 31170bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .msg { -webkit-border-radius:5px }\n" 31270bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .msg { font-family:Helvetica, sans-serif; font-size: smaller }\n" 31370bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .msg { font-weight: bold }\n" 31470bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .msg { float:left }\n" 31570bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .msg { padding:0.5em 1ex 0.5em 1ex }\n" 31670bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .msg { margin-top:10px; margin-bottom:10px }\n" 317170db7c38d363f6ca34b502f3ff1bb17e5eba6b7Ted Kremenek " .msg { max-width:60em; word-wrap: break-word; white-space: pre-wrap;}\n" 31870bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .mrange { background-color:#dfddf3 }\n" 31970bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .mrange { border-bottom:1px solid #6F9DBE }\n" 32070bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek " .PathIndex { font-weight: bold }\n" 3214b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " table.simpletable {\n" 3224b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " padding: 5px;\n" 3234b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " font-size:12pt;\n" 3244b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " margin:20px;\n" 3254b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " border-collapse: collapse; border-spacing: 0px;\n" 3264b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " }\n" 3274b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " td.rowname {\n" 3284b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " text-align:right; font-weight:bold; color:#444444;\n" 3294b0f81323b518429203051bbcd4864bbf4b000a9Ted Kremenek " padding-right:2ex; }\n" 330f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek "</style>\n</head>\n<body>"; 33170bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek 332f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek // Generate header 333f6f593fae2f0531b4bc06891941f7fbba5217618Ted Kremenek R.InsertStrBefore(StartLoc, os.str()); 334ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek // Generate footer 335ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek 33670bcba6030a76edf46c4f941ad9a5297a1f98c47Ted Kremenek R.InsertCStrAfter(EndLoc, "</body></html>\n"); 337ad0a203130dc5d1fb7231b88767174511424fa98Ted Kremenek} 3383245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner 3393245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner/// SyntaxHighlight - Relex the specified FileID and annotate the HTML with 3403245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner/// information about keywords, macro expansions etc. This uses the macro 3413245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner/// table state from the end of the file, so it won't be perfectly perfect, 3423245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner/// but it will be reasonably close. 3432b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattnervoid html::SyntaxHighlight(Rewriter &R, FileID FID, Preprocessor &PP) { 3442b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner RewriteBuffer &RB = R.getEditBuffer(FID); 3453245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner 346a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner const SourceManager &SourceMgr = PP.getSourceManager(); 347025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner Lexer L(FID, SourceMgr, PP.getLangOptions()); 348025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner const char *BufferStart = L.getBufferStart(); 349a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner 3503245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner // Inform the preprocessor that we want to retain comments as tokens, so we 3513245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner // can highlight them. 352678c6358c8d4e368c78629099142397c63c1ee35Chris Lattner L.SetCommentRetentionState(true); 3533245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner 354c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Lex all the tokens in raw mode, to avoid entering #includes or expanding 355c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // macros. 3563245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner Token Tok; 357590f0cc643274267d4d41125b62557e1d87886c3Chris Lattner L.LexFromRawLexer(Tok); 35874ea3e5c57c087c046223096a97ea4e365f85eb6Chris Lattner 35974ea3e5c57c087c046223096a97ea4e365f85eb6Chris Lattner while (Tok.isNot(tok::eof)) { 36074ea3e5c57c087c046223096a97ea4e365f85eb6Chris Lattner // Since we are lexing unexpanded tokens, all tokens are from the main 36174ea3e5c57c087c046223096a97ea4e365f85eb6Chris Lattner // FileID. 36274ea3e5c57c087c046223096a97ea4e365f85eb6Chris Lattner unsigned TokOffs = SourceMgr.getFullFilePos(Tok.getLocation()); 3633245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner unsigned TokLen = Tok.getLength(); 3643245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner switch (Tok.getKind()) { 365a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner default: break; 366a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner case tok::identifier: { 367a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner // Fill in Result.IdentifierInfo, looking up the identifier in the 368a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner // identifier table. 369a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner IdentifierInfo *II = PP.LookUpIdentifierInfo(Tok, BufferStart+TokOffs); 370a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner 371a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner // If this is a pp-identifier, for a keyword, highlight it as such. 3725ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner if (II->getTokenID() != tok::identifier) 3735ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner HighlightRange(RB, TokOffs, TokOffs+TokLen, BufferStart, 3745ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner "<span class='keyword'>", "</span>"); 375c4586c234edd8df0477a895aebcbc3eb220aed6bChris Lattner break; 376a745e8c52839d9c8dd0fd8d5276b4eab182ec7f2Chris Lattner } 3773245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner case tok::comment: 3785ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner HighlightRange(RB, TokOffs, TokOffs+TokLen, BufferStart, 3795ef3e2c45f13fccdb0d7bbcf24c1beee8eee6f64Chris Lattner "<span class='comment'>", "</span>"); 3803245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner break; 381cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek case tok::wide_string_literal: 382cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek // Chop off the L prefix 383cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek ++TokOffs; 384cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek --TokLen; 385cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek // FALL THROUGH. 386cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek case tok::string_literal: 387cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek HighlightRange(RB, TokOffs, TokOffs+TokLen, BufferStart, 388cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek "<span class='string_literal'>", "</span>"); 389cc1b8532a113fa3216096757b3b4e203867bd5e0Ted Kremenek break; 3905deb96d06583abb751463427457d46041af262d0Chris Lattner case tok::hash: { 39174ea3e5c57c087c046223096a97ea4e365f85eb6Chris Lattner // If this is a preprocessor directive, all tokens to end of line are too. 3925deb96d06583abb751463427457d46041af262d0Chris Lattner if (!Tok.isAtStartOfLine()) 3935deb96d06583abb751463427457d46041af262d0Chris Lattner break; 3945deb96d06583abb751463427457d46041af262d0Chris Lattner 3955deb96d06583abb751463427457d46041af262d0Chris Lattner // Eat all of the tokens until we get to the next one at the start of 3965deb96d06583abb751463427457d46041af262d0Chris Lattner // line. 3975deb96d06583abb751463427457d46041af262d0Chris Lattner unsigned TokEnd = TokOffs+TokLen; 398590f0cc643274267d4d41125b62557e1d87886c3Chris Lattner L.LexFromRawLexer(Tok); 3995deb96d06583abb751463427457d46041af262d0Chris Lattner while (!Tok.isAtStartOfLine() && Tok.isNot(tok::eof)) { 4005deb96d06583abb751463427457d46041af262d0Chris Lattner TokEnd = SourceMgr.getFullFilePos(Tok.getLocation())+Tok.getLength(); 401590f0cc643274267d4d41125b62557e1d87886c3Chris Lattner L.LexFromRawLexer(Tok); 40274ea3e5c57c087c046223096a97ea4e365f85eb6Chris Lattner } 4035deb96d06583abb751463427457d46041af262d0Chris Lattner 4045deb96d06583abb751463427457d46041af262d0Chris Lattner // Find end of line. This is a hack. 4055deb96d06583abb751463427457d46041af262d0Chris Lattner HighlightRange(RB, TokOffs, TokEnd, BufferStart, 4065deb96d06583abb751463427457d46041af262d0Chris Lattner "<span class='directive'>", "</span>"); 4075deb96d06583abb751463427457d46041af262d0Chris Lattner 4085deb96d06583abb751463427457d46041af262d0Chris Lattner // Don't skip the next token. 4095deb96d06583abb751463427457d46041af262d0Chris Lattner continue; 4105deb96d06583abb751463427457d46041af262d0Chris Lattner } 4113245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner } 4123245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner 413590f0cc643274267d4d41125b62557e1d87886c3Chris Lattner L.LexFromRawLexer(Tok); 41474ea3e5c57c087c046223096a97ea4e365f85eb6Chris Lattner } 4153245a0a1c7a4fd74fca845b2edba275bb126d773Chris Lattner} 416c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 417c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner/// HighlightMacros - This uses the macro table state from the end of the 418c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner/// file, to reexpand macros and insert (into the HTML) information about the 419c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner/// macro expansions. This won't be perfectly perfect, but it will be 420c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner/// reasonably close. 4212b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattnervoid html::HighlightMacros(Rewriter &R, FileID FID, Preprocessor& PP) { 422339b9c27759d7b6a53e2370f83f66e78b3254595Ted Kremenek 4232b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner RewriteBuffer &RB = R.getEditBuffer(FID); 424c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 425c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Inform the preprocessor that we don't want comments. 426fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek PP.SetCommentRetentionState(false, false); 427c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 428c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Start parsing the specified input file. 429fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek PP.EnterMainSourceFile(); 430c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 431c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Lex all the tokens. 432fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek const SourceManager &SourceMgr = PP.getSourceManager(); 433c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner Token Tok; 434fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek PP.Lex(Tok); 435c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner while (Tok.isNot(tok::eof)) { 436c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Ignore non-macro tokens. 437c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner if (!Tok.getLocation().isMacroID()) { 438fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek PP.Lex(Tok); 439c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner continue; 440c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner } 441c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 442f7cf85b330bedd2877e1371fb0a83e99751ae162Chris Lattner // Ignore tokens whose instantiation location was not the main file. 443f7cf85b330bedd2877e1371fb0a83e99751ae162Chris Lattner SourceLocation LLoc = SourceMgr.getInstantiationLoc(Tok.getLocation()); 4442b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner std::pair<FileID, unsigned> LLocInfo = 445c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner SourceMgr.getDecomposedFileLoc(LLoc); 446c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 4472b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner if (LLocInfo.first != FID) { 448fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek PP.Lex(Tok); 449c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner continue; 450c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner } 451c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 452c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Okay, we have the first token of a macro expansion: highlight the 453c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // instantiation. 454c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 455c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Get the size of current macro call itself. 456c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // FIXME: This should highlight the args of a function-like 457c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // macro, using a heuristic. 458c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner unsigned TokLen = Lexer::MeasureTokenLength(LLoc, SourceMgr); 459c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 460c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner unsigned TokOffs = LLocInfo.second; 4616f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner // Highlight the macro invocation itself. 462c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner RB.InsertTextAfter(TokOffs, "<span class='macro'>", 463c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner strlen("<span class='macro'>")); 464c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner RB.InsertTextBefore(TokOffs+TokLen, "</span>", strlen("</span>")); 465c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 466fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek std::string Expansion = PP.getSpelling(Tok); 4676f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner unsigned LineLen = Expansion.size(); 4686f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner 469c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Okay, eat this token, getting the next one. 470fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek PP.Lex(Tok); 471c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 472c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // Skip all the rest of the tokens that are part of this macro 473c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // instantiation. It would be really nice to pop up a window with all the 474c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner // spelling of the tokens or something. 475c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner while (!Tok.is(tok::eof) && 476f7cf85b330bedd2877e1371fb0a83e99751ae162Chris Lattner SourceMgr.getInstantiationLoc(Tok.getLocation()) == LLoc) { 4776f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner // Insert a newline if the macro expansion is getting large. 4786f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner if (LineLen > 60) { 4796f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner Expansion += "<br>"; 4806f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner LineLen = 0; 4816f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner } 4826f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner 4836f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner LineLen -= Expansion.size(); 4849227c6953497be34281354f949d6f4cd34a696ccChris Lattner // Escape any special characters in the token text. 485fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek Expansion += ' ' + EscapeText(PP.getSpelling(Tok)); 4866f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner LineLen += Expansion.size(); 487fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek PP.Lex(Tok); 4886f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner } 4899227c6953497be34281354f949d6f4cd34a696ccChris Lattner 4906f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner // Insert the information about the expansion inside the macro span. 4916f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner Expansion = "<span class='expansion'>" + Expansion + "</span>"; 4926f46be279f4bc8fc24611f060258bcfbe1c175c4Chris Lattner RB.InsertTextBefore(TokOffs+TokLen, Expansion.c_str(), Expansion.size()); 493c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner } 494c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner} 495c54d50a4180520370c12dd7d06d035263d357d56Chris Lattner 4962b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattnervoid html::HighlightMacros(Rewriter &R, FileID FID, 497fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek PreprocessorFactory &PPF) { 498fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek 499fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek llvm::OwningPtr<Preprocessor> PP(PPF.CreatePreprocessor()); 5002b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner HighlightMacros(R, FID, *PP); 501fb58609c5b3f0d665ea58e37ef39eb7ea74758a4Ted Kremenek} 502