1 //===-- ClangExternalASTSourceCallbacks.cpp -------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "Plugins/ExpressionParser/Clang/ClangExternalASTSourceCallbacks.h"
10 #include "Plugins/TypeSystem/Clang/TypeSystemClang.h"
11 
12 #include "clang/AST/Decl.h"
13 #include "clang/AST/DeclObjC.h"
14 
15 using namespace lldb_private;
16 
17 char ClangExternalASTSourceCallbacks::ID;
18 
19 void ClangExternalASTSourceCallbacks::CompleteType(clang::TagDecl *tag_decl) {
20   m_ast.CompleteTagDecl(tag_decl);
21 }
22 
23 void ClangExternalASTSourceCallbacks::CompleteType(
24     clang::ObjCInterfaceDecl *objc_decl) {
25   m_ast.CompleteObjCInterfaceDecl(objc_decl);
26 }
27 
28 bool ClangExternalASTSourceCallbacks::layoutRecordType(
29     const clang::RecordDecl *Record, uint64_t &Size, uint64_t &Alignment,
30     llvm::DenseMap<const clang::FieldDecl *, uint64_t> &FieldOffsets,
31     llvm::DenseMap<const clang::CXXRecordDecl *, clang::CharUnits> &BaseOffsets,
32     llvm::DenseMap<const clang::CXXRecordDecl *, clang::CharUnits>
33         &VirtualBaseOffsets) {
34   return m_ast.LayoutRecordType(Record, Size, Alignment, FieldOffsets,
35                                 BaseOffsets, VirtualBaseOffsets);
36 }
37 
38 void ClangExternalASTSourceCallbacks::FindExternalLexicalDecls(
39     const clang::DeclContext *decl_ctx,
40     llvm::function_ref<bool(clang::Decl::Kind)> IsKindWeWant,
41     llvm::SmallVectorImpl<clang::Decl *> &decls) {
42   if (decl_ctx) {
43     clang::TagDecl *tag_decl = llvm::dyn_cast<clang::TagDecl>(
44         const_cast<clang::DeclContext *>(decl_ctx));
45     if (tag_decl)
46       CompleteType(tag_decl);
47   }
48 }
49 
50 bool ClangExternalASTSourceCallbacks::FindExternalVisibleDeclsByName(
51     const clang::DeclContext *DC, clang::DeclarationName Name) {
52   llvm::SmallVector<clang::NamedDecl *, 4> decls;
53   // Objective-C methods are not added into the LookupPtr when they originate
54   // from an external source. SetExternalVisibleDeclsForName() adds them.
55   if (auto *oid = llvm::dyn_cast<clang::ObjCInterfaceDecl>(DC)) {
56     clang::ObjCContainerDecl::method_range noload_methods(oid->noload_decls());
57     for (auto *omd : noload_methods)
58       if (omd->getDeclName() == Name)
59         decls.push_back(omd);
60   }
61   return !SetExternalVisibleDeclsForName(DC, Name, decls).empty();
62 }
63 
64 OptionalClangModuleID
65 ClangExternalASTSourceCallbacks::RegisterModule(clang::Module *module) {
66   m_modules.push_back(module);
67   unsigned id = m_modules.size();
68   m_ids.insert({module, id});
69   return OptionalClangModuleID(id);
70 }
71 
72 llvm::Optional<clang::ASTSourceDescriptor>
73 ClangExternalASTSourceCallbacks::getSourceDescriptor(unsigned id) {
74   if (clang::Module *module = getModule(id))
75     return {*module};
76   return {};
77 }
78 
79 clang::Module *ClangExternalASTSourceCallbacks::getModule(unsigned id) {
80   if (id && id <= m_modules.size())
81     return m_modules[id - 1];
82   return nullptr;
83 }
84 
85 OptionalClangModuleID
86 ClangExternalASTSourceCallbacks::GetIDForModule(clang::Module *module) {
87   return OptionalClangModuleID(m_ids[module]);
88 }
89