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 17using namespace clang; 18using namespace clang::index; 19 20static void addOccurrence(std::vector<DeclOccurrence> &Decls, 21 DeclOccurrence Info) { 22 auto IsNextOccurence = [&]() -> bool { 23 if (Decls.empty()) 24 return true; 25 auto &Last = Decls.back(); 26 return Last.Offset < Info.Offset; 27 }; 28 29 if (IsNextOccurence()) { 30 Decls.push_back(std::move(Info)); 31 return; 32 } 33 34 // We keep Decls in order as we need to access them in this order in all cases. 35 auto It = llvm::upper_bound(Decls, Info); 36 Decls.insert(It, std::move(Info)); 37} 38 39void FileIndexRecord::addDeclOccurence(SymbolRoleSet Roles, unsigned Offset, 40 const Decl *D, 41 ArrayRef<SymbolRelation> Relations) { 42 assert(D->isCanonicalDecl() && 43 "Occurrences should be associated with their canonical decl"); 44 addOccurrence(Decls, DeclOccurrence(Roles, Offset, D, Relations)); 45} 46 47void FileIndexRecord::addMacroOccurence(SymbolRoleSet Roles, unsigned Offset, 48 const IdentifierInfo *Name, 49 const MacroInfo *MI) { 50 addOccurrence(Decls, DeclOccurrence(Roles, Offset, Name, MI)); 51} 52 53void FileIndexRecord::removeHeaderGuardMacros() { 54 auto It = 55 std::remove_if(Decls.begin(), Decls.end(), [](const DeclOccurrence &D) { 56 if (const auto *MI = D.DeclOrMacro.dyn_cast<const MacroInfo *>()) 57 return MI->isUsedForHeaderGuard(); 58 return false; 59 }); 60 Decls.erase(It, Decls.end()); 61} 62 63void FileIndexRecord::print(llvm::raw_ostream &OS, SourceManager &SM) const { 64 OS << "DECLS BEGIN ---\n"; 65 for (auto &DclInfo : Decls) { 66 if (const auto *D = DclInfo.DeclOrMacro.dyn_cast<const Decl *>()) { 67 SourceLocation Loc = SM.getFileLoc(D->getLocation()); 68 PresumedLoc PLoc = SM.getPresumedLoc(Loc); 69 OS << llvm::sys::path::filename(PLoc.getFilename()) << ':' 70 << PLoc.getLine() << ':' << PLoc.getColumn(); 71 72 if (const auto *ND = dyn_cast<NamedDecl>(D)) { 73 OS << ' ' << ND->getDeclName(); 74 } 75 } else { 76 const auto *MI = DclInfo.DeclOrMacro.get<const MacroInfo *>(); 77 SourceLocation Loc = SM.getFileLoc(MI->getDefinitionLoc()); 78 PresumedLoc PLoc = SM.getPresumedLoc(Loc); 79 OS << llvm::sys::path::filename(PLoc.getFilename()) << ':' 80 << PLoc.getLine() << ':' << PLoc.getColumn(); 81 OS << ' ' << DclInfo.MacroName->getName(); 82 } 83 84 OS << '\n'; 85 } 86 OS << "DECLS END ---\n"; 87} 88