maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Some cleanups to the HTMLRewrite API. Added support for printing out line 

numbers (more work to be done on aesthetics).

llvm-svn: 48512
This commit is contained in:
Ted Kremenek 2008-03-18 23:08:51 +00:00
parent 43784cc27d
commit 5792cebb08
3 changed files with 124 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
clang/Driver/HTMLPrint.cpp
View File

@ -49,9 +49,10 @@ HTMLPrinter::~HTMLPrinter() {
html::EscapeText(R, FileID);
html::AddLineNumbers(R, FileID);
// html::InsertTag(R, html::HEAD, StartLoc, EndLoc, true);
// html::InsertTag(R, html::BODY, StartLoc, EndLoc, true);
// html::InsertTag(R, html::PRE, StartLoc, EndLoc);
html::InsertTag(R, html::PRE, StartLoc, EndLoc, 0, 0, true);
html::InsertTag(R, html::BODY, StartLoc, EndLoc, NULL, "\n", true);
html::InsertTag(R, html::HEAD, StartLoc, StartLoc, 0, 0, true);
html::InsertTag(R, html::HTML, StartLoc, EndLoc, NULL, "\n", true);
// Emit the HTML.

18
clang/include/clang/Rewrite/HTMLRewrite.h
View File

@ -23,14 +23,26 @@ class Rewriter;
namespace html {
enum Tags { PRE, HEAD, BODY };
// Basic operations.
enum Tags { BODY,
DIV,
HEAD,
HTML,
PRE,
SPAN };
void EscapeText(Rewriter& R, unsigned FileID, bool EscapeSpaces = false);
void InsertTag(Rewriter& R, Tags tag,
SourceLocation OpenLoc, SourceLocation CloseLoc,
bool NewlineOpen = false, bool NewlineClose = true,
bool OutermostTag = false);
const char* Attributes = NULL, const char* Content = NULL,
bool Newline = false,
bool OpenInsertBefore = true, bool CloseInsertAfter = true);
// High-level operations.
void AddLineNumbers(Rewriter& R, unsigned FileID);
} // end html namespace
} // end clang namespace

128
clang/lib/Rewrite/HTMLRewrite.cpp
View File

@ -20,6 +20,10 @@
using namespace clang;
//===----------------------------------------------------------------------===//
// Basic operations.
//===----------------------------------------------------------------------===//
void html::EscapeText(Rewriter& R, unsigned FileID, bool EscapeSpaces) {
const llvm::MemoryBuffer *Buf = R.getSourceMgr().getBuffer(FileID);
@ -46,46 +50,124 @@ void html::EscapeText(Rewriter& R, unsigned FileID, bool EscapeSpaces) {
}
}
static void TagOpen(std::ostringstream& os, const char* TagStr,
const char* Attr, const char* Content) {
os << '<' << TagStr;
if (Attr) os << ' ' << Attr;
os << '>';
if (Content) os << Content;
}
static void TagClose(std::ostringstream& os, const char* TagStr) {
os << "</" << TagStr << ">";
}
void html::InsertTag(Rewriter& R, html::Tags tag,
SourceLocation B, SourceLocation E,
bool NewlineOpen, bool NewlineClose, bool OutermostTag) {
const char* Attributes,
const char* Content, bool Newline,
bool OpenInsertBefore, bool CloseInsertAfter) {
const char* TagStr = 0;
switch (tag) {
default: break;
case PRE: TagStr = "pre"; break;
case HEAD: TagStr = "head"; break;
case BODY: TagStr = "body"; break;
case DIV: TagStr = "div"; break;
case HEAD: TagStr = "head"; break;
case HTML: TagStr = "html"; break;
case PRE: TagStr = "pre"; break;
case SPAN: TagStr = "span"; break;
}
assert (TagStr && "Tag not supported.");
{ // Generate the opening tag.
std::ostringstream os;
os << '<' << TagStr << '>';
if (NewlineOpen) os << '\n';
// Generate the opening tag. We also generate the closing
// tag of the start and end SourceLocations are the same.
{
std::ostringstream os;
TagOpen(os, TagStr, Attributes, Content);
if (B == E) {
TagClose(os, TagStr);
if (Newline) os << '\n';
}
const char* s = os.str().c_str();
unsigned n = os.str().size();
if (OutermostTag)
R.InsertTextBefore(B, s, n);
if (OpenInsertBefore)
R.InsertTextBefore(B, os.str().c_str(), os.str().size());
else
R.InsertTextAfter(B, s, n);
R.InsertTextAfter(B, os.str().c_str(), os.str().size());
}
{ // Generate the closing tag.
std::ostringstream os;
os << "</" << TagStr << '>';
if (NewlineClose) os << '\n';
// Generate the closing tag if the start and end SourceLocations
// are different.
if (B != E) {
std::ostringstream os;
TagClose(os, TagStr);
if (Newline) os << '\n';
const char* s = os.str().c_str();
unsigned n = os.str().size();
if (OutermostTag)
R.InsertTextAfter(E, s, n);
if (CloseInsertAfter)
R.InsertTextAfter(E, os.str().c_str(), os.str().size());
else
R.InsertTextBefore(E, s, n);
R.InsertTextBefore(E, os.str().c_str(), os.str().size());
}
}
//===----------------------------------------------------------------------===//
// High-level operations.
//===----------------------------------------------------------------------===//
static void AddLineNumber(Rewriter& R, unsigned LineNo,
SourceLocation B, SourceLocation E) {
// Add two "div" tags: one to contain the line number, and the other
// to contain the content of the line.
std::ostringstream os;
os << LineNo;
html::InsertTag(R, html::SPAN, B, E, "class=Line");
html::InsertTag(R, html::SPAN, B, B, "class=Num", os.str().c_str());
}
void html::AddLineNumbers(Rewriter& R, unsigned FileID) {
const llvm::MemoryBuffer *Buf = R.getSourceMgr().getBuffer(FileID);
const char* FileBeg = Buf->getBufferStart();
const char* FileEnd = Buf->getBufferEnd();
const char* C = FileBeg;
assert (C <= FileEnd);
unsigned LineNo = 0;
unsigned FilePos = 0;
while (C != FileEnd) {
++LineNo;
unsigned LineStartPos = FilePos;
unsigned LineEndPos = FileEnd - FileBeg;
assert (FilePos <= LineEndPos);
assert (C < FileEnd);
// Scan until the newline (or end-of-file).
for ( ; C != FileEnd ; ++C, ++FilePos)
if (*C == '\n') {
LineEndPos = FilePos;
break;
}
AddLineNumber(R, LineNo,
SourceLocation::getFileLoc(FileID, LineStartPos),
SourceLocation::getFileLoc(FileID, LineEndPos));
if (C != FileEnd) {
++C;
++FilePos;
}
}
}