aboutsummaryrefslogtreecommitdiff
path: root/clang-tools-extra/clangd/unittests/Annotations.cpp
blob: 32b916d54d8709ab9f29057ed0312fb5f33e920d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
//===--- Annotations.cpp - Annotated source code for unit tests --*- 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 "Annotations.h"
#include "SourceCode.h"

namespace clang {
namespace clangd {

Position Annotations::point(llvm::StringRef Name) const {
  return pointWithPayload(Name).first;
}

std::pair<Position, llvm::StringRef>
Annotations::pointWithPayload(llvm::StringRef Name) const {
  auto [BasePoint, Payload] = Base::pointWithPayload(Name);
  return {offsetToPosition(code(), BasePoint), Payload};
}

std::vector<Position> Annotations::points(llvm::StringRef Name) const {
  auto BasePoints = Base::points(Name);

  std::vector<Position> Ps;
  Ps.reserve(BasePoints.size());
  for (const auto Point : BasePoints)
    Ps.push_back(offsetToPosition(code(), Point));

  return Ps;
}

std::vector<std::pair<Position, llvm::StringRef>>
Annotations::pointsWithPayload(llvm::StringRef Name) const {
  auto BasePoints = Base::pointsWithPayload(Name);

  std::vector<std::pair<Position, llvm::StringRef>> Ps;
  Ps.reserve(BasePoints.size());
  for (const auto &[Point, Payload] : BasePoints)
    Ps.push_back({offsetToPosition(code(), Point), Payload});

  return Ps;
}

static clangd::Range toLSPRange(llvm::StringRef Code,
                                llvm::Annotations::Range R) {
  clangd::Range LSPRange;
  LSPRange.start = offsetToPosition(Code, R.Begin);
  LSPRange.end = offsetToPosition(Code, R.End);
  return LSPRange;
}

Range Annotations::range(llvm::StringRef Name) const {
  return rangeWithPayload(Name).first;
}

std::pair<clangd::Range, llvm::StringRef>
Annotations::rangeWithPayload(llvm::StringRef Name) const {
  auto [BaseRange, Payload] = Base::rangeWithPayload(Name);
  return {toLSPRange(code(), BaseRange), Payload};
}

std::vector<Range> Annotations::ranges(llvm::StringRef Name) const {
  auto OffsetRanges = Base::ranges(Name);

  std::vector<clangd::Range> Rs;
  Rs.reserve(OffsetRanges.size());
  for (const auto &R : OffsetRanges)
    Rs.push_back(toLSPRange(code(), R));

  return Rs;
}

std::vector<std::pair<clangd::Range, llvm::StringRef>>
Annotations::rangesWithPayload(llvm::StringRef Name) const {
  auto OffsetRanges = Base::rangesWithPayload(Name);

  std::vector<std::pair<clangd::Range, llvm::StringRef>> Rs;
  Rs.reserve(OffsetRanges.size());
  for (const auto &[R, Payload] : OffsetRanges)
    Rs.push_back({toLSPRange(code(), R), Payload});

  return Rs;
}

} // namespace clangd
} // namespace clang