Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //===- ASTImporterLookupTable.h - ASTImporter specific lookup--*- C++ -*---===// | 
| 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 | //  This file defines the ASTImporterLookupTable class which implements a | ||
| 10 | //  lookup procedure for the import mechanism. | ||
| 11 | // | ||
| 12 | //===----------------------------------------------------------------------===// | ||
| 13 | |||
| 14 | #ifndef LLVM_CLANG_AST_ASTIMPORTERLOOKUPTABLE_H | ||
| 15 | #define LLVM_CLANG_AST_ASTIMPORTERLOOKUPTABLE_H | ||
| 16 | |||
| 17 | #include "clang/AST/DeclBase.h" // lookup_result | ||
| 18 | #include "clang/AST/DeclarationName.h" | ||
| 19 | #include "llvm/ADT/DenseMap.h" | ||
| 20 | #include "llvm/ADT/SetVector.h" | ||
| 21 | |||
| 22 | namespace clang { | ||
| 23 | |||
| 24 | class NamedDecl; | ||
| 25 | class DeclContext; | ||
| 26 | |||
| 27 | // There are certain cases when normal C/C++ lookup (localUncachedLookup) | ||
| 28 | // does not find AST nodes. E.g.: | ||
| 29 | // Example 1: | ||
| 30 | //   template <class T> | ||
| 31 | //   struct X { | ||
| 32 | //     friend void foo(); // this is never found in the DC of the TU. | ||
| 33 | //   }; | ||
| 34 | // Example 2: | ||
| 35 | //   // The fwd decl to Foo is not found in the lookupPtr of the DC of the | ||
| 36 | //   // translation unit decl. | ||
| 37 | //   // Here we could find the node by doing a traverse throught the list of | ||
| 38 | //   // the Decls in the DC, but that would not scale. | ||
| 39 | //   struct A { struct Foo *p; }; | ||
| 40 | // This is a severe problem because the importer decides if it has to create a | ||
| 41 | // new Decl or not based on the lookup results. | ||
| 42 | // To overcome these cases we need an importer specific lookup table which | ||
| 43 | // holds every node and we are not interested in any C/C++ specific visibility | ||
| 44 | // considerations. Simply, we must know if there is an existing Decl in a | ||
| 45 | // given DC. Once we found it then we can handle any visibility related tasks. | ||
| 46 | class ASTImporterLookupTable { | ||
| 47 | |||
| 48 |   // We store a list of declarations for each name. | ||
| 49 |   // And we collect these lists for each DeclContext. | ||
| 50 |   // We could have a flat map with (DeclContext, Name) tuple as key, but a two | ||
| 51 |   // level map seems easier to handle. | ||
| 52 | using DeclList = llvm::SmallSetVector<NamedDecl *, 2>; | ||
| 53 | using NameMap = llvm::SmallDenseMap<DeclarationName, DeclList, 4>; | ||
| 54 | using DCMap = llvm::DenseMap<DeclContext *, NameMap>; | ||
| 55 | |||
| 56 | void add(DeclContext *DC, NamedDecl *ND); | ||
| 57 | void remove(DeclContext *DC, NamedDecl *ND); | ||
| 58 | |||
| 59 |   DCMap LookupTable; | ||
| 60 | |||
| 61 | public: | ||
| 62 | ASTImporterLookupTable(TranslationUnitDecl &TU); | ||
| 63 | void add(NamedDecl *ND); | ||
| 64 | void remove(NamedDecl *ND); | ||
| 65 |   // Sometimes a declaration is created first with a temporarily value of decl | ||
| 66 |   // context (often the translation unit) and later moved to the final context. | ||
| 67 |   // This happens for declarations that are created before the final declaration | ||
| 68 |   // context. In such cases the lookup table needs to be updated. | ||
| 69 |   // (The declaration is in these cases not added to the temporary decl context, | ||
| 70 |   // only its parent is set.) | ||
| 71 |   // FIXME: It would be better to not add the declaration to the temporary | ||
| 72 |   // context at all in the lookup table, but this requires big change in | ||
| 73 |   // ASTImporter. | ||
| 74 |   // The function should be called when the old context is definitely different | ||
| 75 |   // from the new. | ||
| 76 | void update(NamedDecl *ND, DeclContext *OldDC); | ||
| 77 |   // Same as 'update' but allow if 'ND' is not in the table or the old context | ||
| 78 |   // is the same as the new. | ||
| 79 |   // FIXME: The old redeclaration context is not handled. | ||
| 80 | void updateForced(NamedDecl *ND, DeclContext *OldDC); | ||
| 81 | using LookupResult = DeclList; | ||
| 82 | LookupResult lookup(DeclContext *DC, DeclarationName Name) const; | ||
| 83 |   // Check if the `ND` is within the lookup table (with its current name) in | ||
| 84 |   // context `DC`. This is intended for debug purposes when the DeclContext of a | ||
| 85 |   // NamedDecl is changed. | ||
| 86 | bool contains(DeclContext *DC, NamedDecl *ND) const; | ||
| 87 | void dump(DeclContext *DC) const; | ||
| 88 | void dump() const; | ||
| 89 | }; | ||
| 90 | |||
| 91 | } // namespace clang | ||
| 92 | |||
| 93 | #endif // LLVM_CLANG_AST_ASTIMPORTERLOOKUPTABLE_H |