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