| //===--- HTMLPrint.cpp - Source code -> HTML pretty-printing --------------===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // Pretty-printing of source code to HTML. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "clang/Frontend/ASTConsumers.h" |
| #include "clang/AST/ASTConsumer.h" |
| #include "clang/AST/Decl.h" |
| #include "clang/Rewrite/Rewriter.h" |
| #include "clang/Rewrite/HTMLRewrite.h" |
| #include "clang/Basic/Diagnostic.h" |
| #include "clang/Basic/SourceManager.h" |
| #include "clang/Basic/FileManager.h" |
| #include "clang/AST/ASTContext.h" |
| #include "llvm/Support/MemoryBuffer.h" |
| #include "llvm/Support/raw_ostream.h" |
| using namespace clang; |
| |
| //===----------------------------------------------------------------------===// |
| // Functional HTML pretty-printing. |
| //===----------------------------------------------------------------------===// |
| |
| namespace { |
| class HTMLPrinter : public ASTConsumer { |
| Rewriter R; |
| llvm::raw_ostream *Out; |
| Diagnostic &Diags; |
| Preprocessor *PP; |
| PreprocessorFactory *PPF; |
| public: |
| HTMLPrinter(llvm::raw_ostream *OS, Diagnostic &D, Preprocessor *pp, |
| PreprocessorFactory* ppf) |
| : Out(OS), Diags(D), PP(pp), PPF(ppf) {} |
| virtual ~HTMLPrinter(); |
| |
| void Initialize(ASTContext &context); |
| }; |
| } |
| |
| ASTConsumer* clang::CreateHTMLPrinter(llvm::raw_ostream *OS, |
| Diagnostic &D, Preprocessor *PP, |
| PreprocessorFactory* PPF) { |
| |
| return new HTMLPrinter(OS, D, PP, PPF); |
| } |
| |
| void HTMLPrinter::Initialize(ASTContext &context) { |
| R.setSourceMgr(context.getSourceManager(), context.getLangOptions()); |
| } |
| |
| HTMLPrinter::~HTMLPrinter() { |
| if (Diags.hasErrorOccurred()) |
| return; |
| |
| // Format the file. |
| FileID FID = R.getSourceMgr().getMainFileID(); |
| const FileEntry* Entry = R.getSourceMgr().getFileEntryForID(FID); |
| const char* Name; |
| // In some cases, in particular the case where the input is from stdin, |
| // there is no entry. Fall back to the memory buffer for a name in those |
| // cases. |
| if (Entry) |
| Name = Entry->getName(); |
| else |
| Name = R.getSourceMgr().getBuffer(FID)->getBufferIdentifier(); |
| |
| html::AddLineNumbers(R, FID); |
| html::AddHeaderFooterInternalBuiltinCSS(R, FID, Name); |
| |
| // If we have a preprocessor, relex the file and syntax highlight. |
| // We might not have a preprocessor if we come from a deserialized AST file, |
| // for example. |
| |
| if (PP) html::SyntaxHighlight(R, FID, *PP); |
| if (PPF) html::HighlightMacros(R, FID, *PP); |
| html::EscapeText(R, FID, false, true); |
| |
| // Emit the HTML. |
| const RewriteBuffer &RewriteBuf = R.getEditBuffer(FID); |
| char *Buffer = (char*)malloc(RewriteBuf.size()); |
| std::copy(RewriteBuf.begin(), RewriteBuf.end(), Buffer); |
| Out->write(Buffer, RewriteBuf.size()); |
| free(Buffer); |
| } |