Home | History | Annotate | Line # | Download | only in Index 
      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 static 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 
     39 void 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 
     47 void FileIndexRecord::addMacroOccurence(SymbolRoleSet Roles, unsigned Offset,
     48                                         const IdentifierInfo *Name,
     49                                         const MacroInfo *MI) {
     50   addOccurrence(Decls, DeclOccurrence(Roles, Offset, Name, MI));
     51 }
     52 
     53 void 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 
     63 void 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