| 1 | //===--- Ref.cpp -------------------------------------------------*- C++-*-===// |
|---|
| 2 | // |
|---|
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
|---|
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
|---|
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
|---|
| 6 | // |
|---|
| 7 | //===----------------------------------------------------------------------===// |
|---|
| 8 | |
|---|
| 9 | #include "Ref.h" |
|---|
| 10 | #include "llvm/ADT/STLExtras.h" |
|---|
| 11 | |
|---|
| 12 | namespace clang { |
|---|
| 13 | namespace clangd { |
|---|
| 14 | |
|---|
| 15 | llvm::raw_ostream &operator<<(llvm::raw_ostream &OS, RefKind K) { |
|---|
| 16 | if (K == RefKind::Unknown) |
|---|
| 17 | return OS << "Unknown" ; |
|---|
| 18 | static constexpr std::array<const char *, 4> Messages = {"Decl" , "Def" , "Ref" , |
|---|
| 19 | "Spelled" }; |
|---|
| 20 | bool VisitedOnce = false; |
|---|
| 21 | for (unsigned I = 0; I < Messages.size(); ++I) { |
|---|
| 22 | if (static_cast<uint8_t>(K) & 1u << I) { |
|---|
| 23 | if (VisitedOnce) |
|---|
| 24 | OS << ", " ; |
|---|
| 25 | OS << Messages[I]; |
|---|
| 26 | VisitedOnce = true; |
|---|
| 27 | } |
|---|
| 28 | } |
|---|
| 29 | return OS; |
|---|
| 30 | } |
|---|
| 31 | |
|---|
| 32 | llvm::raw_ostream &operator<<(llvm::raw_ostream &OS, const Ref &R) { |
|---|
| 33 | return OS << R.Location << ":" << R.Kind; |
|---|
| 34 | } |
|---|
| 35 | |
|---|
| 36 | void RefSlab::Builder::insert(const SymbolID &ID, const Ref &S) { |
|---|
| 37 | Entry E = {.Symbol: ID, .Reference: S}; |
|---|
| 38 | E.Reference.Location.FileURI = UniqueStrings.save(S: S.Location.FileURI).data(); |
|---|
| 39 | Entries.insert(V: std::move(E)); |
|---|
| 40 | } |
|---|
| 41 | |
|---|
| 42 | RefSlab RefSlab::Builder::build() && { |
|---|
| 43 | std::vector<std::pair<SymbolID, llvm::ArrayRef<Ref>>> Result; |
|---|
| 44 | // We'll reuse the arena, as it only has unique strings and we need them all. |
|---|
| 45 | // We need to group refs by symbol and form contiguous arrays on the arena. |
|---|
| 46 | std::vector<std::pair<SymbolID, const Ref *>> Flat; |
|---|
| 47 | Flat.reserve(n: Entries.size()); |
|---|
| 48 | for (const Entry &E : Entries) |
|---|
| 49 | Flat.emplace_back(args: E.Symbol, args: &E.Reference); |
|---|
| 50 | // Group by SymbolID. |
|---|
| 51 | llvm::sort(C&: Flat, Comp: llvm::less_first()); |
|---|
| 52 | std::vector<Ref> Refs; |
|---|
| 53 | // Loop over symbols, copying refs for each onto the arena. |
|---|
| 54 | for (auto I = Flat.begin(), End = Flat.end(); I != End;) { |
|---|
| 55 | SymbolID Sym = I->first; |
|---|
| 56 | Refs.clear(); |
|---|
| 57 | do { |
|---|
| 58 | Refs.push_back(x: *I->second); |
|---|
| 59 | ++I; |
|---|
| 60 | } while (I != End && I->first == Sym); |
|---|
| 61 | llvm::sort(C&: Refs); // By file, affects xrefs display order. |
|---|
| 62 | Result.emplace_back(args&: Sym, args: llvm::ArrayRef<Ref>(Refs).copy(A&: Arena)); |
|---|
| 63 | } |
|---|
| 64 | return RefSlab(std::move(Result), std::move(Arena), Entries.size()); |
|---|
| 65 | } |
|---|
| 66 | |
|---|
| 67 | } // namespace clangd |
|---|
| 68 | } // namespace clang |
|---|
| 69 | |
|---|
