1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
//===-- ASTStateSerialization.cpp - Persists TU-specific state across TUs -===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "RefactoringContinuations.h"
#include "clang/AST/RecursiveASTVisitor.h"
using namespace clang;
using namespace clang::tooling;
using namespace clang::tooling::detail;
namespace {
class USRToDeclConverter
: public clang::RecursiveASTVisitor<USRToDeclConverter> {
llvm::StringMap<const Decl *> &USRs;
unsigned NumFound = 0;
public:
USRToDeclConverter(llvm::StringMap<const Decl *> &USRs) : USRs(USRs) {}
bool isDone() const { return NumFound == USRs.size(); }
bool VisitNamedDecl(const NamedDecl *D) {
std::string USR = rename::getUSRForDecl(D);
auto It = USRs.find(USR);
if (It == USRs.end() || It->second)
return true;
It->second = D;
++NumFound;
return NumFound != USRs.size();
}
};
} // end anonymous namespace
const Decl *PersistentToASTSpecificStateConverter::lookupDecl(StringRef USR) {
if (USR.empty())
return nullptr;
auto It = ConvertedDeclRefs.find(USR);
if (It != ConvertedDeclRefs.end())
return It->second;
// FIXME: If we ever need to convert a PersistentDeclRef through the ASTQuery,
// we have to support conversion without coalesced conversion.
assert(false && "Persistent decl refs should be converted all at once");
return nullptr;
}
void PersistentToASTSpecificStateConverter::runCoalescedConversions() {
USRToDeclConverter Converter(ConvertedDeclRefs);
for (Decl *D : Context.getTranslationUnitDecl()->decls()) {
Converter.TraverseDecl(D);
if (Converter.isDone())
break;
}
}
FileID
PersistentToASTSpecificStateConverter::convert(const PersistentFileID &Ref) {
FileManager &FM = Context.getSourceManager().getFileManager();
llvm::ErrorOr<const FileEntry *> Entry = FM.getFile(Ref.Filename);
if (!Entry)
return FileID();
return Context.getSourceManager().translateFile(*Entry);
}
|
