Home | History | Annotate | Line # | Download | only in AST 
      1 //===- ASTImporterSharedState.h - ASTImporter specific state --*- C++ -*---===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 //
     10 //  This file defines the ASTImporter specific state, which may be shared
     11 //  amongst several ASTImporter objects.
     12 //
     13 //===----------------------------------------------------------------------===//
     14 
     15 #ifndef LLVM_CLANG_AST_ASTIMPORTERSHAREDSTATE_H
     16 #define LLVM_CLANG_AST_ASTIMPORTERSHAREDSTATE_H
     17 
     18 #include "clang/AST/ASTImporterLookupTable.h"
     19 #include "clang/AST/Decl.h"
     20 #include "llvm/ADT/DenseMap.h"
     21 // FIXME We need this because of ImportError.
     22 #include "clang/AST/ASTImporter.h"
     23 
     24 namespace clang {
     25 
     26 class TranslationUnitDecl;
     27 
     28 /// Importer specific state, which may be shared amongst several ASTImporter
     29 /// objects.
     30 class ASTImporterSharedState {
     31 
     32   /// Pointer to the import specific lookup table.
     33   std::unique_ptr<ASTImporterLookupTable> LookupTable;
     34 
     35   /// Mapping from the already-imported declarations in the "to"
     36   /// context to the error status of the import of that declaration.
     37   /// This map contains only the declarations that were not correctly
     38   /// imported. The same declaration may or may not be included in
     39   /// ImportedFromDecls. This map is updated continuously during imports and
     40   /// never cleared (like ImportedFromDecls).
     41   llvm::DenseMap<Decl *, ImportError> ImportErrors;
     42 
     43   // FIXME put ImportedFromDecls here!
     44   // And from that point we can better encapsulate the lookup table.
     45 
     46 public:
     47   ASTImporterSharedState() = default;
     48 
     49   ASTImporterSharedState(TranslationUnitDecl &ToTU) {
     50     LookupTable = std::make_unique<ASTImporterLookupTable>(ToTU);
     51   }
     52 
     53   ASTImporterLookupTable *getLookupTable() { return LookupTable.get(); }
     54 
     55   void addDeclToLookup(Decl *D) {
     56     if (LookupTable)
     57       if (auto *ND = dyn_cast<NamedDecl>(D))
     58         LookupTable->add(ND);
     59   }
     60 
     61   void removeDeclFromLookup(Decl *D) {
     62     if (LookupTable)
     63       if (auto *ND = dyn_cast<NamedDecl>(D))
     64         LookupTable->remove(ND);
     65   }
     66 
     67   llvm::Optional<ImportError> getImportDeclErrorIfAny(Decl *ToD) const {
     68     auto Pos = ImportErrors.find(ToD);
     69     if (Pos != ImportErrors.end())
     70       return Pos->second;
     71     else
     72       return Optional<ImportError>();
     73   }
     74 
     75   void setImportDeclError(Decl *To, ImportError Error) {
     76     ImportErrors[To] = Error;
     77   }
     78 };
     79 
     80 } // namespace clang
     81 #endif // LLVM_CLANG_AST_ASTIMPORTERSHAREDSTATE_H
     82