annotate clang-tools-extra/clangd/SemanticHighlighting.cpp @ 221:79ff65ed7e25

LLVM12 Original
author Shinji KONO <kono@ie.u-ryukyu.ac.jp>
date Tue, 15 Jun 2021 19:15:29 +0900
parents 0572611fdcc8
children 5f17cb93ff66
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
150
anatofuz
parents:
diff changeset
1 //===--- SemanticHighlighting.cpp - ------------------------- ---*- C++ -*-===//
anatofuz
parents:
diff changeset
2 //
anatofuz
parents:
diff changeset
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
anatofuz
parents:
diff changeset
4 // See https://llvm.org/LICENSE.txt for license information.
anatofuz
parents:
diff changeset
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
anatofuz
parents:
diff changeset
6 //
anatofuz
parents:
diff changeset
7 //===----------------------------------------------------------------------===//
anatofuz
parents:
diff changeset
8
anatofuz
parents:
diff changeset
9 #include "SemanticHighlighting.h"
anatofuz
parents:
diff changeset
10 #include "FindTarget.h"
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
11 #include "HeuristicResolver.h"
150
anatofuz
parents:
diff changeset
12 #include "ParsedAST.h"
anatofuz
parents:
diff changeset
13 #include "Protocol.h"
anatofuz
parents:
diff changeset
14 #include "SourceCode.h"
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
15 #include "support/Logger.h"
150
anatofuz
parents:
diff changeset
16 #include "clang/AST/ASTContext.h"
anatofuz
parents:
diff changeset
17 #include "clang/AST/Decl.h"
anatofuz
parents:
diff changeset
18 #include "clang/AST/DeclCXX.h"
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
19 #include "clang/AST/DeclObjC.h"
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
20 #include "clang/AST/DeclTemplate.h"
150
anatofuz
parents:
diff changeset
21 #include "clang/AST/DeclarationName.h"
anatofuz
parents:
diff changeset
22 #include "clang/AST/ExprCXX.h"
anatofuz
parents:
diff changeset
23 #include "clang/AST/RecursiveASTVisitor.h"
anatofuz
parents:
diff changeset
24 #include "clang/AST/Type.h"
anatofuz
parents:
diff changeset
25 #include "clang/AST/TypeLoc.h"
anatofuz
parents:
diff changeset
26 #include "clang/Basic/LangOptions.h"
anatofuz
parents:
diff changeset
27 #include "clang/Basic/SourceLocation.h"
anatofuz
parents:
diff changeset
28 #include "clang/Basic/SourceManager.h"
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
29 #include "clang/Tooling/Syntax/Tokens.h"
150
anatofuz
parents:
diff changeset
30 #include "llvm/ADT/None.h"
anatofuz
parents:
diff changeset
31 #include "llvm/ADT/Optional.h"
anatofuz
parents:
diff changeset
32 #include "llvm/ADT/STLExtras.h"
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
33 #include "llvm/Support/Base64.h"
150
anatofuz
parents:
diff changeset
34 #include "llvm/Support/Casting.h"
anatofuz
parents:
diff changeset
35 #include <algorithm>
anatofuz
parents:
diff changeset
36
anatofuz
parents:
diff changeset
37 namespace clang {
anatofuz
parents:
diff changeset
38 namespace clangd {
anatofuz
parents:
diff changeset
39 namespace {
anatofuz
parents:
diff changeset
40
anatofuz
parents:
diff changeset
41 /// Some names are not written in the source code and cannot be highlighted,
anatofuz
parents:
diff changeset
42 /// e.g. anonymous classes. This function detects those cases.
anatofuz
parents:
diff changeset
43 bool canHighlightName(DeclarationName Name) {
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
44 switch (Name.getNameKind()) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
45 case DeclarationName::Identifier: {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
46 auto *II = Name.getAsIdentifierInfo();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
47 return II && !II->getName().empty();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
48 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
49 case DeclarationName::CXXConstructorName:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
50 case DeclarationName::CXXDestructorName:
150
anatofuz
parents:
diff changeset
51 return true;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
52 case DeclarationName::ObjCZeroArgSelector:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
53 case DeclarationName::ObjCOneArgSelector:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
54 case DeclarationName::ObjCMultiArgSelector:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
55 // Multi-arg selectors need special handling, and we handle 0/1 arg
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
56 // selectors there too.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
57 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
58 case DeclarationName::CXXConversionFunctionName:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
59 case DeclarationName::CXXOperatorName:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
60 case DeclarationName::CXXDeductionGuideName:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
61 case DeclarationName::CXXLiteralOperatorName:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
62 case DeclarationName::CXXUsingDirective:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
63 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
64 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
65 llvm_unreachable("invalid name kind");
150
anatofuz
parents:
diff changeset
66 }
anatofuz
parents:
diff changeset
67
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
68 llvm::Optional<HighlightingKind> kindForType(const Type *TP,
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
69 const HeuristicResolver *Resolver);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
70 llvm::Optional<HighlightingKind>
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
71 kindForDecl(const NamedDecl *D, const HeuristicResolver *Resolver) {
150
anatofuz
parents:
diff changeset
72 if (auto *USD = dyn_cast<UsingShadowDecl>(D)) {
anatofuz
parents:
diff changeset
73 if (auto *Target = USD->getTargetDecl())
anatofuz
parents:
diff changeset
74 D = Target;
anatofuz
parents:
diff changeset
75 }
anatofuz
parents:
diff changeset
76 if (auto *TD = dyn_cast<TemplateDecl>(D)) {
anatofuz
parents:
diff changeset
77 if (auto *Templated = TD->getTemplatedDecl())
anatofuz
parents:
diff changeset
78 D = Templated;
anatofuz
parents:
diff changeset
79 }
anatofuz
parents:
diff changeset
80 if (auto *TD = dyn_cast<TypedefNameDecl>(D)) {
anatofuz
parents:
diff changeset
81 // We try to highlight typedefs as their underlying type.
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
82 if (auto K =
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
83 kindForType(TD->getUnderlyingType().getTypePtrOrNull(), Resolver))
150
anatofuz
parents:
diff changeset
84 return K;
anatofuz
parents:
diff changeset
85 // And fallback to a generic kind if this fails.
anatofuz
parents:
diff changeset
86 return HighlightingKind::Typedef;
anatofuz
parents:
diff changeset
87 }
anatofuz
parents:
diff changeset
88 // We highlight class decls, constructor decls and destructor decls as
anatofuz
parents:
diff changeset
89 // `Class` type. The destructor decls are handled in `VisitTagTypeLoc` (we
anatofuz
parents:
diff changeset
90 // will visit a TypeLoc where the underlying Type is a CXXRecordDecl).
anatofuz
parents:
diff changeset
91 if (auto *RD = llvm::dyn_cast<RecordDecl>(D)) {
anatofuz
parents:
diff changeset
92 // We don't want to highlight lambdas like classes.
anatofuz
parents:
diff changeset
93 if (RD->isLambda())
anatofuz
parents:
diff changeset
94 return llvm::None;
anatofuz
parents:
diff changeset
95 return HighlightingKind::Class;
anatofuz
parents:
diff changeset
96 }
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
97 if (isa<ClassTemplateDecl, RecordDecl, CXXConstructorDecl, ObjCInterfaceDecl,
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
98 ObjCImplementationDecl>(D))
150
anatofuz
parents:
diff changeset
99 return HighlightingKind::Class;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
100 if (isa<ObjCProtocolDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
101 return HighlightingKind::Interface;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
102 if (isa<ObjCCategoryDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
103 return HighlightingKind::Namespace;
150
anatofuz
parents:
diff changeset
104 if (auto *MD = dyn_cast<CXXMethodDecl>(D))
anatofuz
parents:
diff changeset
105 return MD->isStatic() ? HighlightingKind::StaticMethod
anatofuz
parents:
diff changeset
106 : HighlightingKind::Method;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
107 if (auto *OMD = dyn_cast<ObjCMethodDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
108 return OMD->isClassMethod() ? HighlightingKind::StaticMethod
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
109 : HighlightingKind::Method;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
110 if (isa<FieldDecl, ObjCPropertyDecl>(D))
150
anatofuz
parents:
diff changeset
111 return HighlightingKind::Field;
anatofuz
parents:
diff changeset
112 if (isa<EnumDecl>(D))
anatofuz
parents:
diff changeset
113 return HighlightingKind::Enum;
anatofuz
parents:
diff changeset
114 if (isa<EnumConstantDecl>(D))
anatofuz
parents:
diff changeset
115 return HighlightingKind::EnumConstant;
anatofuz
parents:
diff changeset
116 if (isa<ParmVarDecl>(D))
anatofuz
parents:
diff changeset
117 return HighlightingKind::Parameter;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
118 if (auto *VD = dyn_cast<VarDecl>(D)) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
119 if (isa<ImplicitParamDecl>(VD)) // e.g. ObjC Self
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
120 return llvm::None;
150
anatofuz
parents:
diff changeset
121 return VD->isStaticDataMember()
anatofuz
parents:
diff changeset
122 ? HighlightingKind::StaticField
anatofuz
parents:
diff changeset
123 : VD->isLocalVarDecl() ? HighlightingKind::LocalVariable
anatofuz
parents:
diff changeset
124 : HighlightingKind::Variable;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
125 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
126 if (const auto *BD = dyn_cast<BindingDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
127 return BD->getDeclContext()->isFunctionOrMethod()
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
128 ? HighlightingKind::LocalVariable
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
129 : HighlightingKind::Variable;
150
anatofuz
parents:
diff changeset
130 if (isa<FunctionDecl>(D))
anatofuz
parents:
diff changeset
131 return HighlightingKind::Function;
anatofuz
parents:
diff changeset
132 if (isa<NamespaceDecl>(D) || isa<NamespaceAliasDecl>(D) ||
anatofuz
parents:
diff changeset
133 isa<UsingDirectiveDecl>(D))
anatofuz
parents:
diff changeset
134 return HighlightingKind::Namespace;
anatofuz
parents:
diff changeset
135 if (isa<TemplateTemplateParmDecl>(D) || isa<TemplateTypeParmDecl>(D) ||
anatofuz
parents:
diff changeset
136 isa<NonTypeTemplateParmDecl>(D))
anatofuz
parents:
diff changeset
137 return HighlightingKind::TemplateParameter;
anatofuz
parents:
diff changeset
138 if (isa<ConceptDecl>(D))
anatofuz
parents:
diff changeset
139 return HighlightingKind::Concept;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
140 if (const auto *UUVD = dyn_cast<UnresolvedUsingValueDecl>(D)) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
141 auto Targets = Resolver->resolveUsingValueDecl(UUVD);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
142 if (!Targets.empty()) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
143 return kindForDecl(Targets[0], Resolver);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
144 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
145 return HighlightingKind::Unknown;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
146 }
150
anatofuz
parents:
diff changeset
147 return llvm::None;
anatofuz
parents:
diff changeset
148 }
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
149 llvm::Optional<HighlightingKind>
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
150 kindForType(const Type *TP, const HeuristicResolver *Resolver) {
150
anatofuz
parents:
diff changeset
151 if (!TP)
anatofuz
parents:
diff changeset
152 return llvm::None;
anatofuz
parents:
diff changeset
153 if (TP->isBuiltinType()) // Builtins are special, they do not have decls.
anatofuz
parents:
diff changeset
154 return HighlightingKind::Primitive;
anatofuz
parents:
diff changeset
155 if (auto *TD = dyn_cast<TemplateTypeParmType>(TP))
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
156 return kindForDecl(TD->getDecl(), Resolver);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
157 if (isa<ObjCObjectPointerType>(TP))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
158 return HighlightingKind::Class;
150
anatofuz
parents:
diff changeset
159 if (auto *TD = TP->getAsTagDecl())
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
160 return kindForDecl(TD, Resolver);
150
anatofuz
parents:
diff changeset
161 return llvm::None;
anatofuz
parents:
diff changeset
162 }
anatofuz
parents:
diff changeset
163
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
164 // Whether T is const in a loose sense - is a variable with this type readonly?
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
165 bool isConst(QualType T) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
166 if (T.isNull() || T->isDependentType())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
167 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
168 T = T.getNonReferenceType();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
169 if (T.isConstQualified())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
170 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
171 if (const auto *AT = T->getAsArrayTypeUnsafe())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
172 return isConst(AT->getElementType());
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
173 if (isConst(T->getPointeeType()))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
174 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
175 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
176 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
177
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
178 // Whether D is const in a loose sense (should it be highlighted as such?)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
179 // FIXME: This is separate from whether *a particular usage* can mutate D.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
180 // We may want V in V.size() to be readonly even if V is mutable.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
181 bool isConst(const Decl *D) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
182 if (llvm::isa<EnumConstantDecl>(D) || llvm::isa<NonTypeTemplateParmDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
183 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
184 if (llvm::isa<FieldDecl>(D) || llvm::isa<VarDecl>(D) ||
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
185 llvm::isa<MSPropertyDecl>(D) || llvm::isa<BindingDecl>(D)) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
186 if (isConst(llvm::cast<ValueDecl>(D)->getType()))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
187 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
188 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
189 if (const auto *OCPD = llvm::dyn_cast<ObjCPropertyDecl>(D)) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
190 if (OCPD->isReadOnly())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
191 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
192 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
193 if (const auto *MPD = llvm::dyn_cast<MSPropertyDecl>(D)) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
194 if (!MPD->hasSetter())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
195 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
196 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
197 if (const auto *CMD = llvm::dyn_cast<CXXMethodDecl>(D)) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
198 if (CMD->isConst())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
199 return true;
150
anatofuz
parents:
diff changeset
200 }
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
201 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
202 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
203
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
204 // "Static" means many things in C++, only some get the "static" modifier.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
205 //
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
206 // Meanings that do:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
207 // - Members associated with the class rather than the instance.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
208 // This is what 'static' most often means across languages.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
209 // - static local variables
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
210 // These are similarly "detached from their context" by the static keyword.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
211 // In practice, these are rarely used inside classes, reducing confusion.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
212 //
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
213 // Meanings that don't:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
214 // - Namespace-scoped variables, which have static storage class.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
215 // This is implicit, so the keyword "static" isn't so strongly associated.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
216 // If we want a modifier for these, "global scope" is probably the concept.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
217 // - Namespace-scoped variables/functions explicitly marked "static".
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
218 // There the keyword changes *linkage* , which is a totally different concept.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
219 // If we want to model this, "file scope" would be a nice modifier.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
220 //
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
221 // This is confusing, and maybe we should use another name, but because "static"
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
222 // is a standard LSP modifier, having one with that name has advantages.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
223 bool isStatic(const Decl *D) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
224 if (const auto *CMD = llvm::dyn_cast<CXXMethodDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
225 return CMD->isStatic();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
226 if (const VarDecl *VD = llvm::dyn_cast<VarDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
227 return VD->isStaticDataMember() || VD->isStaticLocal();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
228 if (const auto *OPD = llvm::dyn_cast<ObjCPropertyDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
229 return OPD->isClassProperty();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
230 if (const auto *OMD = llvm::dyn_cast<ObjCMethodDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
231 return OMD->isClassMethod();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
232 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
233 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
234
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
235 bool isAbstract(const Decl *D) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
236 if (const auto *CMD = llvm::dyn_cast<CXXMethodDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
237 return CMD->isPure();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
238 if (const auto *CRD = llvm::dyn_cast<CXXRecordDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
239 return CRD->hasDefinition() && CRD->isAbstract();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
240 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
241 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
242
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
243 bool isDependent(const Decl *D) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
244 if (isa<UnresolvedUsingValueDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
245 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
246 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
247 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
248
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
249 /// Returns true if `Decl` is considered to be from a default/system library.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
250 /// This currently checks the systemness of the file by include type, although
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
251 /// different heuristics may be used in the future (e.g. sysroot paths).
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
252 bool isDefaultLibrary(const Decl *D) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
253 SourceLocation Loc = D->getLocation();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
254 if (!Loc.isValid())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
255 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
256 return D->getASTContext().getSourceManager().isInSystemHeader(Loc);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
257 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
258
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
259 bool isDefaultLibrary(const Type *T) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
260 if (!T)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
261 return false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
262 const Type *Underlying = T->getPointeeOrArrayElementType();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
263 if (Underlying->isBuiltinType())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
264 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
265 if (auto *TD = dyn_cast<TemplateTypeParmType>(Underlying))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
266 return isDefaultLibrary(TD->getDecl());
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
267 if (auto *TD = Underlying->getAsTagDecl())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
268 return isDefaultLibrary(TD);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
269 return false;
150
anatofuz
parents:
diff changeset
270 }
anatofuz
parents:
diff changeset
271
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
272 // For a macro usage `DUMP(foo)`, we want:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
273 // - DUMP --> "macro"
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
274 // - foo --> "variable".
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
275 SourceLocation getHighlightableSpellingToken(SourceLocation L,
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
276 const SourceManager &SM) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
277 if (L.isFileID())
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
278 return SM.isWrittenInMainFile(L) ? L : SourceLocation{};
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
279 // Tokens expanded from the macro body contribute no highlightings.
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
280 if (!SM.isMacroArgExpansion(L))
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
281 return {};
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
282 // Tokens expanded from macro args are potentially highlightable.
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
283 return getHighlightableSpellingToken(SM.getImmediateSpellingLoc(L), SM);
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
284 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
285
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
286 unsigned evaluateHighlightPriority(const HighlightingToken &Tok) {
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
287 enum HighlightPriority { Dependent = 0, Resolved = 1 };
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
288 return (Tok.Modifiers & (1 << uint32_t(HighlightingModifier::DependentName)))
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
289 ? Dependent
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
290 : Resolved;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
291 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
292
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
293 // Sometimes we get multiple tokens at the same location:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
294 //
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
295 // - findExplicitReferences() returns a heuristic result for a dependent name
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
296 // (e.g. Method) and CollectExtraHighlighting returning a fallback dependent
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
297 // highlighting (e.g. Unknown+Dependent).
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
298 // - macro arguments are expanded multiple times and have different roles
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
299 // - broken code recovery produces several AST nodes at the same location
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
300 //
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
301 // We should either resolve these to a single token, or drop them all.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
302 // Our heuristics are:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
303 //
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
304 // - token kinds that come with "dependent-name" modifiers are less reliable
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
305 // (these tend to be vague, like Type or Unknown)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
306 // - if we have multiple equally reliable kinds, drop token rather than guess
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
307 // - take the union of modifiers from all tokens
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
308 //
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
309 // In particular, heuristically resolved dependent names get their heuristic
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
310 // kind, plus the dependent modifier.
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
311 llvm::Optional<HighlightingToken>
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
312 resolveConflict(ArrayRef<HighlightingToken> Tokens) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
313 if (Tokens.size() == 1)
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
314 return Tokens[0];
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
315
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
316 if (Tokens.size() != 2)
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
317 return llvm::None;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
318
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
319 unsigned Priority1 = evaluateHighlightPriority(Tokens[0]);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
320 unsigned Priority2 = evaluateHighlightPriority(Tokens[1]);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
321 if (Priority1 == Priority2 && Tokens[0].Kind != Tokens[1].Kind)
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
322 return llvm::None;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
323 auto Result = Priority1 > Priority2 ? Tokens[0] : Tokens[1];
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
324 Result.Modifiers = Tokens[0].Modifiers | Tokens[1].Modifiers;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
325 return Result;
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
326 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
327
150
anatofuz
parents:
diff changeset
328 /// Consumes source locations and maps them to text ranges for highlightings.
anatofuz
parents:
diff changeset
329 class HighlightingsBuilder {
anatofuz
parents:
diff changeset
330 public:
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
331 HighlightingsBuilder(const ParsedAST &AST)
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
332 : TB(AST.getTokens()), SourceMgr(AST.getSourceManager()),
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
333 LangOpts(AST.getLangOpts()) {}
150
anatofuz
parents:
diff changeset
334
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
335 HighlightingToken &addToken(SourceLocation Loc, HighlightingKind Kind) {
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
336 Loc = getHighlightableSpellingToken(Loc, SourceMgr);
150
anatofuz
parents:
diff changeset
337 if (Loc.isInvalid())
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
338 return InvalidHighlightingToken;
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
339 const auto *Tok = TB.spelledTokenAt(Loc);
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
340 assert(Tok);
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
341 return addToken(
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
342 halfOpenToRange(SourceMgr,
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
343 Tok->range(SourceMgr).toCharRange(SourceMgr)),
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
344 Kind);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
345 }
150
anatofuz
parents:
diff changeset
346
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
347 HighlightingToken &addToken(Range R, HighlightingKind Kind) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
348 HighlightingToken HT;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
349 HT.R = std::move(R);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
350 HT.Kind = Kind;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
351 Tokens.push_back(std::move(HT));
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
352 return Tokens.back();
150
anatofuz
parents:
diff changeset
353 }
anatofuz
parents:
diff changeset
354
anatofuz
parents:
diff changeset
355 std::vector<HighlightingToken> collect(ParsedAST &AST) && {
anatofuz
parents:
diff changeset
356 // Initializer lists can give duplicates of tokens, therefore all tokens
anatofuz
parents:
diff changeset
357 // must be deduplicated.
anatofuz
parents:
diff changeset
358 llvm::sort(Tokens);
anatofuz
parents:
diff changeset
359 auto Last = std::unique(Tokens.begin(), Tokens.end());
anatofuz
parents:
diff changeset
360 Tokens.erase(Last, Tokens.end());
anatofuz
parents:
diff changeset
361
anatofuz
parents:
diff changeset
362 // Macros can give tokens that have the same source range but conflicting
anatofuz
parents:
diff changeset
363 // kinds. In this case all tokens sharing this source range should be
anatofuz
parents:
diff changeset
364 // removed.
anatofuz
parents:
diff changeset
365 std::vector<HighlightingToken> NonConflicting;
anatofuz
parents:
diff changeset
366 NonConflicting.reserve(Tokens.size());
anatofuz
parents:
diff changeset
367 for (ArrayRef<HighlightingToken> TokRef = Tokens; !TokRef.empty();) {
anatofuz
parents:
diff changeset
368 ArrayRef<HighlightingToken> Conflicting =
anatofuz
parents:
diff changeset
369 TokRef.take_while([&](const HighlightingToken &T) {
anatofuz
parents:
diff changeset
370 // TokRef is guaranteed at least one element here because otherwise
anatofuz
parents:
diff changeset
371 // this predicate would never fire.
anatofuz
parents:
diff changeset
372 return T.R == TokRef.front().R;
anatofuz
parents:
diff changeset
373 });
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
374 if (auto Resolved = resolveConflict(Conflicting))
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
375 NonConflicting.push_back(*Resolved);
150
anatofuz
parents:
diff changeset
376 // TokRef[Conflicting.size()] is the next token with a different range (or
anatofuz
parents:
diff changeset
377 // the end of the Tokens).
anatofuz
parents:
diff changeset
378 TokRef = TokRef.drop_front(Conflicting.size());
anatofuz
parents:
diff changeset
379 }
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
380 const auto &SM = AST.getSourceManager();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
381 StringRef MainCode = SM.getBufferOrFake(SM.getMainFileID()).getBuffer();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
382
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
383 // Merge token stream with "inactive line" markers.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
384 std::vector<HighlightingToken> WithInactiveLines;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
385 auto SortedSkippedRanges = AST.getMacros().SkippedRanges;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
386 llvm::sort(SortedSkippedRanges);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
387 auto It = NonConflicting.begin();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
388 for (const Range &R : SortedSkippedRanges) {
150
anatofuz
parents:
diff changeset
389 // Create one token for each line in the skipped range, so it works
anatofuz
parents:
diff changeset
390 // with line-based diffing.
anatofuz
parents:
diff changeset
391 assert(R.start.line <= R.end.line);
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
392 for (int Line = R.start.line; Line <= R.end.line; ++Line) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
393 // If the end of the inactive range is at the beginning
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
394 // of a line, that line is not inactive.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
395 if (Line == R.end.line && R.end.character == 0)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
396 continue;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
397 // Copy tokens before the inactive line
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
398 for (; It != NonConflicting.end() && It->R.start.line < Line; ++It)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
399 WithInactiveLines.push_back(std::move(*It));
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
400 // Add a token for the inactive line itself.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
401 auto StartOfLine = positionToOffset(MainCode, Position{Line, 0});
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
402 if (StartOfLine) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
403 StringRef LineText =
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
404 MainCode.drop_front(*StartOfLine).take_until([](char C) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
405 return C == '\n';
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
406 });
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
407 HighlightingToken HT;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
408 WithInactiveLines.emplace_back();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
409 WithInactiveLines.back().Kind = HighlightingKind::InactiveCode;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
410 WithInactiveLines.back().R.start.line = Line;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
411 WithInactiveLines.back().R.end.line = Line;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
412 WithInactiveLines.back().R.end.character =
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
413 static_cast<int>(lspLength(LineText));
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
414 } else {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
415 elog("Failed to convert position to offset: {0}",
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
416 StartOfLine.takeError());
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
417 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
418
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
419 // Skip any other tokens on the inactive line. e.g.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
420 // `#ifndef Foo` is considered as part of an inactive region when Foo is
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
421 // defined, and there is a Foo macro token.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
422 // FIXME: we should reduce the scope of the inactive region to not
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
423 // include the directive itself.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
424 while (It != NonConflicting.end() && It->R.start.line == Line)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
425 ++It;
150
anatofuz
parents:
diff changeset
426 }
anatofuz
parents:
diff changeset
427 }
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
428 // Copy tokens after the last inactive line
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
429 for (; It != NonConflicting.end(); ++It)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
430 WithInactiveLines.push_back(std::move(*It));
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
431 return WithInactiveLines;
150
anatofuz
parents:
diff changeset
432 }
anatofuz
parents:
diff changeset
433
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
434 const HeuristicResolver *getResolver() const { return Resolver; }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
435
150
anatofuz
parents:
diff changeset
436 private:
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
437 const syntax::TokenBuffer &TB;
150
anatofuz
parents:
diff changeset
438 const SourceManager &SourceMgr;
anatofuz
parents:
diff changeset
439 const LangOptions &LangOpts;
anatofuz
parents:
diff changeset
440 std::vector<HighlightingToken> Tokens;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
441 const HeuristicResolver *Resolver;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
442 // returned from addToken(InvalidLoc)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
443 HighlightingToken InvalidHighlightingToken;
150
anatofuz
parents:
diff changeset
444 };
anatofuz
parents:
diff changeset
445
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
446 llvm::Optional<HighlightingModifier> scopeModifier(const NamedDecl *D) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
447 const DeclContext *DC = D->getDeclContext();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
448 // Injected "Foo" within the class "Foo" has file scope, not class scope.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
449 if (auto *R = dyn_cast_or_null<RecordDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
450 if (R->isInjectedClassName())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
451 DC = DC->getParent();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
452 // Lambda captures are considered function scope, not class scope.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
453 if (llvm::isa<FieldDecl>(D))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
454 if (const auto *RD = llvm::dyn_cast<RecordDecl>(DC))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
455 if (RD->isLambda())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
456 return HighlightingModifier::FunctionScope;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
457 // Walk up the DeclContext hierarchy until we find something interesting.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
458 for (; !DC->isFileContext(); DC = DC->getParent()) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
459 if (DC->isFunctionOrMethod())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
460 return HighlightingModifier::FunctionScope;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
461 if (DC->isRecord())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
462 return HighlightingModifier::ClassScope;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
463 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
464 // Some template parameters (e.g. those for variable templates) don't have
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
465 // meaningful DeclContexts. That doesn't mean they're global!
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
466 if (DC->isTranslationUnit() && D->isTemplateParameter())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
467 return llvm::None;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
468 // ExternalLinkage threshold could be tweaked, e.g. module-visible as global.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
469 if (D->getLinkageInternal() < ExternalLinkage)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
470 return HighlightingModifier::FileScope;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
471 return HighlightingModifier::GlobalScope;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
472 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
473
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
474 llvm::Optional<HighlightingModifier> scopeModifier(const Type *T) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
475 if (!T)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
476 return llvm::None;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
477 if (T->isBuiltinType())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
478 return HighlightingModifier::GlobalScope;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
479 if (auto *TD = dyn_cast<TemplateTypeParmType>(T))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
480 return scopeModifier(TD->getDecl());
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
481 if (auto *TD = T->getAsTagDecl())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
482 return scopeModifier(TD);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
483 return llvm::None;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
484 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
485
150
anatofuz
parents:
diff changeset
486 /// Produces highlightings, which are not captured by findExplicitReferences,
anatofuz
parents:
diff changeset
487 /// e.g. highlights dependent names and 'auto' as the underlying type.
anatofuz
parents:
diff changeset
488 class CollectExtraHighlightings
anatofuz
parents:
diff changeset
489 : public RecursiveASTVisitor<CollectExtraHighlightings> {
anatofuz
parents:
diff changeset
490 public:
anatofuz
parents:
diff changeset
491 CollectExtraHighlightings(HighlightingsBuilder &H) : H(H) {}
anatofuz
parents:
diff changeset
492
anatofuz
parents:
diff changeset
493 bool VisitDecltypeTypeLoc(DecltypeTypeLoc L) {
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
494 if (auto K = kindForType(L.getTypePtr(), H.getResolver())) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
495 auto &Tok = H.addToken(L.getBeginLoc(), *K)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
496 .addModifier(HighlightingModifier::Deduced);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
497 if (auto Mod = scopeModifier(L.getTypePtr()))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
498 Tok.addModifier(*Mod);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
499 if (isDefaultLibrary(L.getTypePtr()))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
500 Tok.addModifier(HighlightingModifier::DefaultLibrary);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
501 }
150
anatofuz
parents:
diff changeset
502 return true;
anatofuz
parents:
diff changeset
503 }
anatofuz
parents:
diff changeset
504
anatofuz
parents:
diff changeset
505 bool VisitDeclaratorDecl(DeclaratorDecl *D) {
anatofuz
parents:
diff changeset
506 auto *AT = D->getType()->getContainedAutoType();
anatofuz
parents:
diff changeset
507 if (!AT)
anatofuz
parents:
diff changeset
508 return true;
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
509 if (auto K = kindForType(AT->getDeducedType().getTypePtrOrNull(),
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
510 H.getResolver())) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
511 auto &Tok = H.addToken(D->getTypeSpecStartLoc(), *K)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
512 .addModifier(HighlightingModifier::Deduced);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
513 const Type *Deduced = AT->getDeducedType().getTypePtrOrNull();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
514 if (auto Mod = scopeModifier(Deduced))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
515 Tok.addModifier(*Mod);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
516 if (isDefaultLibrary(Deduced))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
517 Tok.addModifier(HighlightingModifier::DefaultLibrary);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
518 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
519 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
520 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
521
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
522 // We handle objective-C selectors specially, because one reference can
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
523 // cover several non-contiguous tokens.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
524 void highlightObjCSelector(const ArrayRef<SourceLocation> &Locs, bool Decl,
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
525 bool Class, bool DefaultLibrary) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
526 HighlightingKind Kind =
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
527 Class ? HighlightingKind::StaticMethod : HighlightingKind::Method;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
528 for (SourceLocation Part : Locs) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
529 auto &Tok =
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
530 H.addToken(Part, Kind).addModifier(HighlightingModifier::ClassScope);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
531 if (Decl)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
532 Tok.addModifier(HighlightingModifier::Declaration);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
533 if (Class)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
534 Tok.addModifier(HighlightingModifier::Static);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
535 if (DefaultLibrary)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
536 Tok.addModifier(HighlightingModifier::DefaultLibrary);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
537 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
538 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
539
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
540 bool VisitObjCMethodDecl(ObjCMethodDecl *OMD) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
541 llvm::SmallVector<SourceLocation> Locs;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
542 OMD->getSelectorLocs(Locs);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
543 highlightObjCSelector(Locs, /*Decl=*/true, OMD->isClassMethod(),
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
544 isDefaultLibrary(OMD));
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
545 return true;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
546 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
547
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
548 bool VisitObjCMessageExpr(ObjCMessageExpr *OME) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
549 llvm::SmallVector<SourceLocation> Locs;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
550 OME->getSelectorLocs(Locs);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
551 bool DefaultLibrary = false;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
552 if (ObjCMethodDecl *OMD = OME->getMethodDecl())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
553 DefaultLibrary = isDefaultLibrary(OMD);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
554 highlightObjCSelector(Locs, /*Decl=*/false, OME->isClassMessage(),
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
555 DefaultLibrary);
150
anatofuz
parents:
diff changeset
556 return true;
anatofuz
parents:
diff changeset
557 }
anatofuz
parents:
diff changeset
558
anatofuz
parents:
diff changeset
559 bool VisitOverloadExpr(OverloadExpr *E) {
anatofuz
parents:
diff changeset
560 if (!E->decls().empty())
anatofuz
parents:
diff changeset
561 return true; // handled by findExplicitReferences.
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
562 auto &Tok = H.addToken(E->getNameLoc(), HighlightingKind::Unknown)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
563 .addModifier(HighlightingModifier::DependentName);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
564 if (llvm::isa<UnresolvedMemberExpr>(E))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
565 Tok.addModifier(HighlightingModifier::ClassScope);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
566 // other case is UnresolvedLookupExpr, scope is unknown.
150
anatofuz
parents:
diff changeset
567 return true;
anatofuz
parents:
diff changeset
568 }
anatofuz
parents:
diff changeset
569
anatofuz
parents:
diff changeset
570 bool VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
571 H.addToken(E->getMemberNameInfo().getLoc(), HighlightingKind::Unknown)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
572 .addModifier(HighlightingModifier::DependentName)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
573 .addModifier(HighlightingModifier::ClassScope);
150
anatofuz
parents:
diff changeset
574 return true;
anatofuz
parents:
diff changeset
575 }
anatofuz
parents:
diff changeset
576
anatofuz
parents:
diff changeset
577 bool VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
578 H.addToken(E->getNameInfo().getLoc(), HighlightingKind::Unknown)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
579 .addModifier(HighlightingModifier::DependentName)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
580 .addModifier(HighlightingModifier::ClassScope);
150
anatofuz
parents:
diff changeset
581 return true;
anatofuz
parents:
diff changeset
582 }
anatofuz
parents:
diff changeset
583
anatofuz
parents:
diff changeset
584 bool VisitDependentNameTypeLoc(DependentNameTypeLoc L) {
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
585 H.addToken(L.getNameLoc(), HighlightingKind::Type)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
586 .addModifier(HighlightingModifier::DependentName)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
587 .addModifier(HighlightingModifier::ClassScope);
150
anatofuz
parents:
diff changeset
588 return true;
anatofuz
parents:
diff changeset
589 }
anatofuz
parents:
diff changeset
590
anatofuz
parents:
diff changeset
591 bool VisitDependentTemplateSpecializationTypeLoc(
anatofuz
parents:
diff changeset
592 DependentTemplateSpecializationTypeLoc L) {
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
593 H.addToken(L.getTemplateNameLoc(), HighlightingKind::Type)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
594 .addModifier(HighlightingModifier::DependentName)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
595 .addModifier(HighlightingModifier::ClassScope);
150
anatofuz
parents:
diff changeset
596 return true;
anatofuz
parents:
diff changeset
597 }
anatofuz
parents:
diff changeset
598
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
599 bool TraverseTemplateArgumentLoc(TemplateArgumentLoc L) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
600 // Handle template template arguments only (other arguments are handled by
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
601 // their Expr, TypeLoc etc values).
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
602 if (L.getArgument().getKind() != TemplateArgument::Template &&
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
603 L.getArgument().getKind() != TemplateArgument::TemplateExpansion)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
604 return RecursiveASTVisitor::TraverseTemplateArgumentLoc(L);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
605
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
606 TemplateName N = L.getArgument().getAsTemplateOrTemplatePattern();
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
607 switch (N.getKind()) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
608 case TemplateName::OverloadedTemplate:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
609 // Template template params must always be class templates.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
610 // Don't bother to try to work out the scope here.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
611 H.addToken(L.getTemplateNameLoc(), HighlightingKind::Class);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
612 break;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
613 case TemplateName::DependentTemplate:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
614 case TemplateName::AssumedTemplate:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
615 H.addToken(L.getTemplateNameLoc(), HighlightingKind::Class)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
616 .addModifier(HighlightingModifier::DependentName);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
617 break;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
618 case TemplateName::Template:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
619 case TemplateName::QualifiedTemplate:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
620 case TemplateName::SubstTemplateTemplateParm:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
621 case TemplateName::SubstTemplateTemplateParmPack:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
622 // Names that could be resolved to a TemplateDecl are handled elsewhere.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
623 break;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
624 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
625 return RecursiveASTVisitor::TraverseTemplateArgumentLoc(L);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
626 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
627
150
anatofuz
parents:
diff changeset
628 // findExplicitReferences will walk nested-name-specifiers and
anatofuz
parents:
diff changeset
629 // find anything that can be resolved to a Decl. However, non-leaf
anatofuz
parents:
diff changeset
630 // components of nested-name-specifiers which are dependent names
anatofuz
parents:
diff changeset
631 // (kind "Identifier") cannot be resolved to a decl, so we visit
anatofuz
parents:
diff changeset
632 // them here.
anatofuz
parents:
diff changeset
633 bool TraverseNestedNameSpecifierLoc(NestedNameSpecifierLoc Q) {
anatofuz
parents:
diff changeset
634 if (NestedNameSpecifier *NNS = Q.getNestedNameSpecifier()) {
anatofuz
parents:
diff changeset
635 if (NNS->getKind() == NestedNameSpecifier::Identifier)
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
636 H.addToken(Q.getLocalBeginLoc(), HighlightingKind::Type)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
637 .addModifier(HighlightingModifier::DependentName)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
638 .addModifier(HighlightingModifier::ClassScope);
150
anatofuz
parents:
diff changeset
639 }
anatofuz
parents:
diff changeset
640 return RecursiveASTVisitor::TraverseNestedNameSpecifierLoc(Q);
anatofuz
parents:
diff changeset
641 }
anatofuz
parents:
diff changeset
642
anatofuz
parents:
diff changeset
643 private:
anatofuz
parents:
diff changeset
644 HighlightingsBuilder &H;
anatofuz
parents:
diff changeset
645 };
anatofuz
parents:
diff changeset
646 } // namespace
anatofuz
parents:
diff changeset
647
anatofuz
parents:
diff changeset
648 std::vector<HighlightingToken> getSemanticHighlightings(ParsedAST &AST) {
anatofuz
parents:
diff changeset
649 auto &C = AST.getASTContext();
anatofuz
parents:
diff changeset
650 // Add highlightings for AST nodes.
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
651 HighlightingsBuilder Builder(AST);
150
anatofuz
parents:
diff changeset
652 // Highlight 'decltype' and 'auto' as their underlying types.
anatofuz
parents:
diff changeset
653 CollectExtraHighlightings(Builder).TraverseAST(C);
anatofuz
parents:
diff changeset
654 // Highlight all decls and references coming from the AST.
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
655 findExplicitReferences(
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
656 C,
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
657 [&](ReferenceLoc R) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
658 for (const NamedDecl *Decl : R.Targets) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
659 if (!canHighlightName(Decl->getDeclName()))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
660 continue;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
661 auto Kind = kindForDecl(Decl, AST.getHeuristicResolver());
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
662 if (!Kind)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
663 continue;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
664 auto &Tok = Builder.addToken(R.NameLoc, *Kind);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
665
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
666 // The attribute tests don't want to look at the template.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
667 if (auto *TD = dyn_cast<TemplateDecl>(Decl)) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
668 if (auto *Templated = TD->getTemplatedDecl())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
669 Decl = Templated;
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
670 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
671 if (auto Mod = scopeModifier(Decl))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
672 Tok.addModifier(*Mod);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
673 if (isConst(Decl))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
674 Tok.addModifier(HighlightingModifier::Readonly);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
675 if (isStatic(Decl))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
676 Tok.addModifier(HighlightingModifier::Static);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
677 if (isAbstract(Decl))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
678 Tok.addModifier(HighlightingModifier::Abstract);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
679 if (isDependent(Decl))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
680 Tok.addModifier(HighlightingModifier::DependentName);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
681 if (isDefaultLibrary(Decl))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
682 Tok.addModifier(HighlightingModifier::DefaultLibrary);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
683 if (Decl->isDeprecated())
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
684 Tok.addModifier(HighlightingModifier::Deprecated);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
685 // Do not treat an UnresolvedUsingValueDecl as a declaration.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
686 // It's more common to think of it as a reference to the
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
687 // underlying declaration.
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
688 if (R.IsDecl && !isa<UnresolvedUsingValueDecl>(Decl))
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
689 Tok.addModifier(HighlightingModifier::Declaration);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
690 }
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
691 },
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
692 AST.getHeuristicResolver());
150
anatofuz
parents:
diff changeset
693 // Add highlightings for macro references.
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
694 auto AddMacro = [&](const MacroOccurrence &M) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
695 auto &T = Builder.addToken(M.Rng, HighlightingKind::Macro);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
696 T.addModifier(HighlightingModifier::GlobalScope);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
697 if (M.IsDefinition)
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
698 T.addModifier(HighlightingModifier::Declaration);
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
699 };
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
700 for (const auto &SIDToRefs : AST.getMacros().MacroRefs)
150
anatofuz
parents:
diff changeset
701 for (const auto &M : SIDToRefs.second)
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
702 AddMacro(M);
150
anatofuz
parents:
diff changeset
703 for (const auto &M : AST.getMacros().UnknownMacros)
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
704 AddMacro(M);
150
anatofuz
parents:
diff changeset
705
anatofuz
parents:
diff changeset
706 return std::move(Builder).collect(AST);
anatofuz
parents:
diff changeset
707 }
anatofuz
parents:
diff changeset
708
anatofuz
parents:
diff changeset
709 llvm::raw_ostream &operator<<(llvm::raw_ostream &OS, HighlightingKind K) {
anatofuz
parents:
diff changeset
710 switch (K) {
anatofuz
parents:
diff changeset
711 case HighlightingKind::Variable:
anatofuz
parents:
diff changeset
712 return OS << "Variable";
anatofuz
parents:
diff changeset
713 case HighlightingKind::LocalVariable:
anatofuz
parents:
diff changeset
714 return OS << "LocalVariable";
anatofuz
parents:
diff changeset
715 case HighlightingKind::Parameter:
anatofuz
parents:
diff changeset
716 return OS << "Parameter";
anatofuz
parents:
diff changeset
717 case HighlightingKind::Function:
anatofuz
parents:
diff changeset
718 return OS << "Function";
anatofuz
parents:
diff changeset
719 case HighlightingKind::Method:
anatofuz
parents:
diff changeset
720 return OS << "Method";
anatofuz
parents:
diff changeset
721 case HighlightingKind::StaticMethod:
anatofuz
parents:
diff changeset
722 return OS << "StaticMethod";
anatofuz
parents:
diff changeset
723 case HighlightingKind::Field:
anatofuz
parents:
diff changeset
724 return OS << "Field";
anatofuz
parents:
diff changeset
725 case HighlightingKind::StaticField:
anatofuz
parents:
diff changeset
726 return OS << "StaticField";
anatofuz
parents:
diff changeset
727 case HighlightingKind::Class:
anatofuz
parents:
diff changeset
728 return OS << "Class";
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
729 case HighlightingKind::Interface:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
730 return OS << "Interface";
150
anatofuz
parents:
diff changeset
731 case HighlightingKind::Enum:
anatofuz
parents:
diff changeset
732 return OS << "Enum";
anatofuz
parents:
diff changeset
733 case HighlightingKind::EnumConstant:
anatofuz
parents:
diff changeset
734 return OS << "EnumConstant";
anatofuz
parents:
diff changeset
735 case HighlightingKind::Typedef:
anatofuz
parents:
diff changeset
736 return OS << "Typedef";
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
737 case HighlightingKind::Type:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
738 return OS << "Type";
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
739 case HighlightingKind::Unknown:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
740 return OS << "Unknown";
150
anatofuz
parents:
diff changeset
741 case HighlightingKind::Namespace:
anatofuz
parents:
diff changeset
742 return OS << "Namespace";
anatofuz
parents:
diff changeset
743 case HighlightingKind::TemplateParameter:
anatofuz
parents:
diff changeset
744 return OS << "TemplateParameter";
anatofuz
parents:
diff changeset
745 case HighlightingKind::Concept:
anatofuz
parents:
diff changeset
746 return OS << "Concept";
anatofuz
parents:
diff changeset
747 case HighlightingKind::Primitive:
anatofuz
parents:
diff changeset
748 return OS << "Primitive";
anatofuz
parents:
diff changeset
749 case HighlightingKind::Macro:
anatofuz
parents:
diff changeset
750 return OS << "Macro";
anatofuz
parents:
diff changeset
751 case HighlightingKind::InactiveCode:
anatofuz
parents:
diff changeset
752 return OS << "InactiveCode";
anatofuz
parents:
diff changeset
753 }
anatofuz
parents:
diff changeset
754 llvm_unreachable("invalid HighlightingKind");
anatofuz
parents:
diff changeset
755 }
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
756 llvm::raw_ostream &operator<<(llvm::raw_ostream &OS, HighlightingModifier K) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
757 switch (K) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
758 case HighlightingModifier::Declaration:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
759 return OS << "decl"; // abbrevation for common case
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
760 default:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
761 return OS << toSemanticTokenModifier(K);
150
anatofuz
parents:
diff changeset
762 }
anatofuz
parents:
diff changeset
763 }
anatofuz
parents:
diff changeset
764
anatofuz
parents:
diff changeset
765 bool operator==(const HighlightingToken &L, const HighlightingToken &R) {
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
766 return std::tie(L.R, L.Kind, L.Modifiers) ==
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
767 std::tie(R.R, R.Kind, R.Modifiers);
150
anatofuz
parents:
diff changeset
768 }
anatofuz
parents:
diff changeset
769 bool operator<(const HighlightingToken &L, const HighlightingToken &R) {
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
770 return std::tie(L.R, L.Kind, R.Modifiers) <
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
771 std::tie(R.R, R.Kind, R.Modifiers);
150
anatofuz
parents:
diff changeset
772 }
anatofuz
parents:
diff changeset
773
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
774 std::vector<SemanticToken>
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
775 toSemanticTokens(llvm::ArrayRef<HighlightingToken> Tokens) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
776 assert(std::is_sorted(Tokens.begin(), Tokens.end()));
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
777 std::vector<SemanticToken> Result;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
778 const HighlightingToken *Last = nullptr;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
779 for (const HighlightingToken &Tok : Tokens) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
780 Result.emplace_back();
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
781 SemanticToken &Out = Result.back();
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
782 // deltaStart/deltaLine are relative if possible.
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
783 if (Last) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
784 assert(Tok.R.start.line >= Last->R.start.line);
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
785 Out.deltaLine = Tok.R.start.line - Last->R.start.line;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
786 if (Out.deltaLine == 0) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
787 assert(Tok.R.start.character >= Last->R.start.character);
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
788 Out.deltaStart = Tok.R.start.character - Last->R.start.character;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
789 } else {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
790 Out.deltaStart = Tok.R.start.character;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
791 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
792 } else {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
793 Out.deltaLine = Tok.R.start.line;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
794 Out.deltaStart = Tok.R.start.character;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
795 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
796 assert(Tok.R.end.line == Tok.R.start.line);
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
797 Out.length = Tok.R.end.character - Tok.R.start.character;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
798 Out.tokenType = static_cast<unsigned>(Tok.Kind);
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
799 Out.tokenModifiers = Tok.Modifiers;
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
800
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
801 Last = &Tok;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
802 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
803 return Result;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
804 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
805 llvm::StringRef toSemanticTokenType(HighlightingKind Kind) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
806 switch (Kind) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
807 case HighlightingKind::Variable:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
808 case HighlightingKind::LocalVariable:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
809 case HighlightingKind::StaticField:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
810 return "variable";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
811 case HighlightingKind::Parameter:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
812 return "parameter";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
813 case HighlightingKind::Function:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
814 return "function";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
815 case HighlightingKind::Method:
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
816 return "method";
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
817 case HighlightingKind::StaticMethod:
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
818 // FIXME: better method with static modifier?
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
819 return "function";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
820 case HighlightingKind::Field:
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
821 return "property";
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
822 case HighlightingKind::Class:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
823 return "class";
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
824 case HighlightingKind::Interface:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
825 return "interface";
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
826 case HighlightingKind::Enum:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
827 return "enum";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
828 case HighlightingKind::EnumConstant:
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
829 return "enumMember";
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
830 case HighlightingKind::Typedef:
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
831 case HighlightingKind::Type:
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
832 return "type";
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
833 case HighlightingKind::Unknown:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
834 return "unknown"; // nonstandard
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
835 case HighlightingKind::Namespace:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
836 return "namespace";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
837 case HighlightingKind::TemplateParameter:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
838 return "typeParameter";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
839 case HighlightingKind::Concept:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
840 return "concept"; // nonstandard
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
841 case HighlightingKind::Primitive:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
842 return "type";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
843 case HighlightingKind::Macro:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
844 return "macro";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
845 case HighlightingKind::InactiveCode:
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
846 return "comment";
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
847 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
848 llvm_unreachable("unhandled HighlightingKind");
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
849 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
850
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
851 llvm::StringRef toSemanticTokenModifier(HighlightingModifier Modifier) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
852 switch (Modifier) {
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
853 case HighlightingModifier::Declaration:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
854 return "declaration";
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
855 case HighlightingModifier::Deprecated:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
856 return "deprecated";
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
857 case HighlightingModifier::Readonly:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
858 return "readonly";
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
859 case HighlightingModifier::Static:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
860 return "static";
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
861 case HighlightingModifier::Deduced:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
862 return "deduced"; // nonstandard
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
863 case HighlightingModifier::Abstract:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
864 return "abstract";
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
865 case HighlightingModifier::DependentName:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
866 return "dependentName"; // nonstandard
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
867 case HighlightingModifier::DefaultLibrary:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
868 return "defaultLibrary";
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
869 case HighlightingModifier::FunctionScope:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
870 return "functionScope"; // nonstandard
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
871 case HighlightingModifier::ClassScope:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
872 return "classScope"; // nonstandard
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
873 case HighlightingModifier::FileScope:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
874 return "fileScope"; // nonstandard
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
875 case HighlightingModifier::GlobalScope:
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
876 return "globalScope"; // nonstandard
150
anatofuz
parents:
diff changeset
877 }
221
79ff65ed7e25 LLVM12 Original
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 173
diff changeset
878 llvm_unreachable("unhandled HighlightingModifier");
150
anatofuz
parents:
diff changeset
879 }
anatofuz
parents:
diff changeset
880
173
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
881 std::vector<SemanticTokensEdit>
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
882 diffTokens(llvm::ArrayRef<SemanticToken> Old,
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
883 llvm::ArrayRef<SemanticToken> New) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
884 // For now, just replace everything from the first-last modification.
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
885 // FIXME: use a real diff instead, this is bad with include-insertion.
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
886
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
887 unsigned Offset = 0;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
888 while (!Old.empty() && !New.empty() && Old.front() == New.front()) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
889 ++Offset;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
890 Old = Old.drop_front();
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
891 New = New.drop_front();
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
892 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
893 while (!Old.empty() && !New.empty() && Old.back() == New.back()) {
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
894 Old = Old.drop_back();
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
895 New = New.drop_back();
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
896 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
897
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
898 if (Old.empty() && New.empty())
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
899 return {};
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
900 SemanticTokensEdit Edit;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
901 Edit.startToken = Offset;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
902 Edit.deleteTokens = Old.size();
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
903 Edit.tokens = New;
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
904 return {std::move(Edit)};
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
905 }
0572611fdcc8 reorgnization done
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 150
diff changeset
906
150
anatofuz
parents:
diff changeset
907 } // namespace clangd
anatofuz
parents:
diff changeset
908 } // namespace clang