1 //===--- FileIndexRecord.cpp - Index data per file --------------*- 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 #include "FileIndexRecord.h"
11 #include "clang/AST/ASTContext.h"
12 #include "clang/AST/DeclTemplate.h"
13 #include "clang/Basic/SourceManager.h"
14 #include "llvm/ADT/SmallString.h"
15 #include "llvm/Support/Path.h"
16
17 using namespace clang;
18 using namespace clang::index;
19
20 ArrayRef<DeclOccurrence>
getDeclOccurrencesSortedByOffset() const21 FileIndexRecord::getDeclOccurrencesSortedByOffset() const {
22 if (!IsSorted) {
23 llvm::stable_sort(Decls,
24 [](const DeclOccurrence &A, const DeclOccurrence &B) {
25 return A.Offset < B.Offset;
26 });
27 IsSorted = true;
28 }
29 return Decls;
30 }
31
addDeclOccurence(SymbolRoleSet Roles,unsigned Offset,const Decl * D,ArrayRef<SymbolRelation> Relations)32 void FileIndexRecord::addDeclOccurence(SymbolRoleSet Roles, unsigned Offset,
33 const Decl *D,
34 ArrayRef<SymbolRelation> Relations) {
35 assert(D->isCanonicalDecl() &&
36 "Occurrences should be associated with their canonical decl");
37 IsSorted = false;
38 Decls.emplace_back(Roles, Offset, D, Relations);
39 }
40
addMacroOccurence(SymbolRoleSet Roles,unsigned Offset,const IdentifierInfo * Name,const MacroInfo * MI)41 void FileIndexRecord::addMacroOccurence(SymbolRoleSet Roles, unsigned Offset,
42 const IdentifierInfo *Name,
43 const MacroInfo *MI) {
44 IsSorted = false;
45 Decls.emplace_back(Roles, Offset, Name, MI);
46 }
47
removeHeaderGuardMacros()48 void FileIndexRecord::removeHeaderGuardMacros() {
49 auto It =
50 std::remove_if(Decls.begin(), Decls.end(), [](const DeclOccurrence &D) {
51 if (const auto *MI = D.DeclOrMacro.dyn_cast<const MacroInfo *>())
52 return MI->isUsedForHeaderGuard();
53 return false;
54 });
55 Decls.erase(It, Decls.end());
56 }
57
print(llvm::raw_ostream & OS,SourceManager & SM) const58 void FileIndexRecord::print(llvm::raw_ostream &OS, SourceManager &SM) const {
59 OS << "DECLS BEGIN ---\n";
60 for (auto &DclInfo : Decls) {
61 if (const auto *D = DclInfo.DeclOrMacro.dyn_cast<const Decl *>()) {
62 SourceLocation Loc = SM.getFileLoc(D->getLocation());
63 PresumedLoc PLoc = SM.getPresumedLoc(Loc);
64 OS << llvm::sys::path::filename(PLoc.getFilename()) << ':'
65 << PLoc.getLine() << ':' << PLoc.getColumn();
66
67 if (const auto *ND = dyn_cast<NamedDecl>(D)) {
68 OS << ' ' << ND->getDeclName();
69 }
70 } else {
71 const auto *MI = DclInfo.DeclOrMacro.get<const MacroInfo *>();
72 SourceLocation Loc = SM.getFileLoc(MI->getDefinitionLoc());
73 PresumedLoc PLoc = SM.getPresumedLoc(Loc);
74 OS << llvm::sys::path::filename(PLoc.getFilename()) << ':'
75 << PLoc.getLine() << ':' << PLoc.getColumn();
76 OS << ' ' << DclInfo.MacroName->getName();
77 }
78
79 OS << '\n';
80 }
81 OS << "DECLS END ---\n";
82 }
83