text: //===--- FileIndexRecord.cpp - Index data per file --------------*- C++ -*-===//
  //
  // 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 "FileIndexRecord.h"
  #include "clang/AST/ASTContext.h"
  #include "clang/AST/DeclTemplate.h"
  #include "clang/Basic/SourceManager.h"
  #include "llvm/ADT/SmallString.h"
  #include "llvm/Support/Path.h"
  
  using namespace clang;
  using namespace clang::index;
  
  ArrayRef<DeclOccurrence>
  FileIndexRecord::getDeclOccurrencesSortedByOffset() const {
    if (!IsSorted) {
      llvm::stable_sort(Decls,
                        [](const DeclOccurrence &A, const DeclOccurrence &B) {
                          return A.Offset < B.Offset;
                        });
      IsSorted = true;
    }
    return Decls;
  }
  
  void FileIndexRecord::addDeclOccurence(SymbolRoleSet Roles, unsigned Offset,
                                         const Decl *D,
                                         ArrayRef<SymbolRelation> Relations) {
    assert(D->isCanonicalDecl() &&
           "Occurrences should be associated with their canonical decl");
    IsSorted = false;
    Decls.emplace_back(Roles, Offset, D, Relations);
  }
  
  void FileIndexRecord::addMacroOccurence(SymbolRoleSet Roles, unsigned Offset,
                                          const IdentifierInfo *Name,
                                          const MacroInfo *MI) {
    IsSorted = false;
    Decls.emplace_back(Roles, Offset, Name, MI);
  }
  
  void FileIndexRecord::removeHeaderGuardMacros() {
    auto It =
        std::remove_if(Decls.begin(), Decls.end(), [](const DeclOccurrence &D) {
          if (const auto *MI = D.DeclOrMacro.dyn_cast<const MacroInfo *>())
            return MI->isUsedForHeaderGuard();
          return false;
        });
    Decls.erase(It, Decls.end());
  }
  
  void FileIndexRecord::print(llvm::raw_ostream &OS, SourceManager &SM) const {
    OS << "DECLS BEGIN ---\n";
    for (auto &DclInfo : Decls) {
      if (const auto *D = DclInfo.DeclOrMacro.dyn_cast<const Decl *>()) {
        SourceLocation Loc = SM.getFileLoc(D->getLocation());
        PresumedLoc PLoc = SM.getPresumedLoc(Loc);
        OS << llvm::sys::path::filename(PLoc.getFilename()) << ':'
           << PLoc.getLine() << ':' << PLoc.getColumn();
  
        if (const auto *ND = dyn_cast<NamedDecl>(D)) {
          OS << ' ' << ND->getDeclName();
        }
      } else {
        const auto *MI = DclInfo.DeclOrMacro.get<const MacroInfo *>();
        SourceLocation Loc = SM.getFileLoc(MI->getDefinitionLoc());
        PresumedLoc PLoc = SM.getPresumedLoc(Loc);
        OS << llvm::sys::path::filename(PLoc.getFilename()) << ':'
           << PLoc.getLine() << ':' << PLoc.getColumn();
        OS << ' ' << DclInfo.MacroName->getName();
      }
  
      OS << '\n';
    }
    OS << "DECLS END ---\n";
  }
  