mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-24 22:08:57 -04:00
0621cb2e7d
The rewrite facility's footprint is small so it's not worth going to these lengths to support disabling at configure time, particularly since key compiler features now depend on it. Meanwhile the Objective-C rewriters have been moved under the ENABLE_CLANG_ARCMT umbrella for now as they're comparatively heavy and still potentially worth excluding from lightweight builds. Tests are now passing with any combination of feature flags. The flags historically haven't been tested by LLVM's build servers so caveat emptor. llvm-svn: 213171
95 lines
3.2 KiB
C++
95 lines
3.2 KiB
C++
//===--- 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/Rewrite/Frontend/ASTConsumers.h"
|
|
#include "clang/AST/ASTConsumer.h"
|
|
#include "clang/AST/ASTContext.h"
|
|
#include "clang/AST/Decl.h"
|
|
#include "clang/Basic/Diagnostic.h"
|
|
#include "clang/Basic/FileManager.h"
|
|
#include "clang/Basic/SourceManager.h"
|
|
#include "clang/Lex/Preprocessor.h"
|
|
#include "clang/Rewrite/Core/HTMLRewrite.h"
|
|
#include "clang/Rewrite/Core/Rewriter.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;
|
|
raw_ostream *Out;
|
|
Preprocessor &PP;
|
|
bool SyntaxHighlight, HighlightMacros;
|
|
|
|
public:
|
|
HTMLPrinter(raw_ostream *OS, Preprocessor &pp,
|
|
bool _SyntaxHighlight, bool _HighlightMacros)
|
|
: Out(OS), PP(pp), SyntaxHighlight(_SyntaxHighlight),
|
|
HighlightMacros(_HighlightMacros) {}
|
|
|
|
void Initialize(ASTContext &context) override;
|
|
void HandleTranslationUnit(ASTContext &Ctx) override;
|
|
};
|
|
}
|
|
|
|
ASTConsumer* clang::CreateHTMLPrinter(raw_ostream *OS,
|
|
Preprocessor &PP,
|
|
bool SyntaxHighlight,
|
|
bool HighlightMacros) {
|
|
return new HTMLPrinter(OS, PP, SyntaxHighlight, HighlightMacros);
|
|
}
|
|
|
|
void HTMLPrinter::Initialize(ASTContext &context) {
|
|
R.setSourceMgr(context.getSourceManager(), context.getLangOpts());
|
|
}
|
|
|
|
void HTMLPrinter::HandleTranslationUnit(ASTContext &Ctx) {
|
|
if (PP.getDiagnostics().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 (SyntaxHighlight) html::SyntaxHighlight(R, FID, PP);
|
|
if (HighlightMacros) 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);
|
|
}
|