mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
76 lines
2.4 KiB
C++
76 lines
2.4 KiB
C++
//===--- GenClangDecl.cpp - Swift IRGen for imported Clang declarations ---===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "IRGenModule.h"
|
|
#include "clang/AST/Decl.h"
|
|
#include "clang/AST/DeclGroup.h"
|
|
#include "clang/AST/RecursiveASTVisitor.h"
|
|
#include "clang/CodeGen/ModuleBuilder.h"
|
|
#include "llvm/ADT/SmallPtrSet.h"
|
|
|
|
using namespace swift;
|
|
using namespace irgen;
|
|
|
|
namespace {
|
|
class ClangDeclRefFinder
|
|
: public clang::RecursiveASTVisitor<ClangDeclRefFinder> {
|
|
std::function<void(const clang::DeclRefExpr *)> callback;
|
|
public:
|
|
template <typename Fn>
|
|
explicit ClangDeclRefFinder(Fn fn) : callback(fn) {}
|
|
|
|
bool VisitDeclRefExpr(clang::DeclRefExpr *DRE) {
|
|
callback(DRE);
|
|
return true;
|
|
}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
void IRGenModule::emitClangDecl(clang::Decl *decl) {
|
|
auto valueDecl = dyn_cast<clang::ValueDecl>(decl);
|
|
if (!valueDecl || valueDecl->isExternallyVisible()) {
|
|
ClangCodeGen->HandleTopLevelDecl(clang::DeclGroupRef(decl));
|
|
return;
|
|
}
|
|
|
|
if (!GlobalClangDecls.insert(decl->getCanonicalDecl()).second)
|
|
return;
|
|
SmallVector<const clang::Decl *, 8> stack;
|
|
stack.push_back(decl);
|
|
|
|
ClangDeclRefFinder refFinder([&](const clang::DeclRefExpr *DRE) {
|
|
const clang::ValueDecl *D = DRE->getDecl();
|
|
if (!D->hasLinkage() || D->isExternallyVisible())
|
|
return;
|
|
if (!GlobalClangDecls.insert(D->getCanonicalDecl()).second)
|
|
return;
|
|
stack.push_back(D);
|
|
});
|
|
|
|
while (!stack.empty()) {
|
|
auto *next = const_cast<clang::Decl *>(stack.pop_back_val());
|
|
if (auto fn = dyn_cast<clang::FunctionDecl>(next)) {
|
|
const clang::FunctionDecl *definition;
|
|
if (fn->hasBody(definition)) {
|
|
refFinder.TraverseDecl(const_cast<clang::FunctionDecl *>(definition));
|
|
next = const_cast<clang::FunctionDecl *>(definition);
|
|
}
|
|
}
|
|
ClangCodeGen->HandleTopLevelDecl(clang::DeclGroupRef(next));
|
|
}
|
|
}
|
|
|
|
void IRGenModule::finalizeClangCodeGen() {
|
|
ClangCodeGen->HandleTranslationUnit(
|
|
*const_cast<clang::ASTContext *>(ClangASTContext));
|
|
}
|