blob: 74db4969867cb460937e4b177ac2987811f4a89c [file] [log] [blame]
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001//===--- TokenAnnotator.cpp - Format C++ code -----------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9///
10/// \file
11/// \brief This file implements a token annotator, i.e. creates
12/// \c AnnotatedTokens out of \c FormatTokens with required extra information.
13///
14//===----------------------------------------------------------------------===//
15
16#include "TokenAnnotator.h"
17#include "clang/Basic/SourceManager.h"
Daniel Jasperbf71ba22013-04-08 20:33:42 +000018#include "llvm/Support/Debug.h"
Daniel Jasper32d28ee2013-01-29 21:01:14 +000019
20namespace clang {
21namespace format {
22
Craig Topper14e66492013-07-01 04:03:19 +000023namespace {
24
Daniel Jasper32d28ee2013-01-29 21:01:14 +000025/// \brief A parser that gathers additional information about tokens.
26///
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +000027/// The \c TokenAnnotator tries to match parenthesis and square brakets and
Daniel Jasper32d28ee2013-01-29 21:01:14 +000028/// store a parenthesis levels. It also tries to resolve matching "<" and ">"
29/// into template parameter lists.
30class AnnotatingParser {
31public:
Daniel Jasperd4a03db2013-08-22 15:00:41 +000032 AnnotatingParser(const FormatStyle &Style, AnnotatedLine &Line,
33 IdentifierInfo &Ident_in)
34 : Style(Style), Line(Line), CurrentToken(Line.First),
35 KeywordVirtualFound(false), NameFound(false), AutoFound(false),
36 Ident_in(Ident_in) {
Nico Weber27268772013-06-26 00:30:14 +000037 Contexts.push_back(Context(tok::unknown, 1, /*IsExpression=*/false));
Daniel Jasper32d28ee2013-01-29 21:01:14 +000038 }
39
Nico Weber95e8e462013-02-12 16:17:07 +000040private:
Daniel Jasper32d28ee2013-01-29 21:01:14 +000041 bool parseAngle() {
42 if (CurrentToken == NULL)
43 return false;
Daniel Jasper923ebef2013-03-14 13:45:21 +000044 ScopedContextCreator ContextCreator(*this, tok::less, 10);
Manuel Klimekb3987012013-05-29 14:47:47 +000045 FormatToken *Left = CurrentToken->Previous;
Daniel Jasper4e778092013-02-06 10:05:46 +000046 Contexts.back().IsExpression = false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +000047 while (CurrentToken != NULL) {
48 if (CurrentToken->is(tok::greater)) {
49 Left->MatchingParen = CurrentToken;
50 CurrentToken->MatchingParen = Left;
51 CurrentToken->Type = TT_TemplateCloser;
52 next();
53 return true;
54 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +000055 if (CurrentToken->isOneOf(tok::r_paren, tok::r_square, tok::r_brace,
Daniel Jasper5d823e32013-05-15 13:46:48 +000056 tok::question, tok::colon))
57 return false;
Daniel Jasper0348be02013-06-01 18:56:00 +000058 // If a && or || is found and interpreted as a binary operator, this set
Daniel Jasper15f33f02013-06-03 16:16:41 +000059 // of angles is likely part of something like "a < b && c > d". If the
Daniel Jasper0348be02013-06-01 18:56:00 +000060 // angles are inside an expression, the ||/&& might also be a binary
61 // operator that was misinterpreted because we are parsing template
62 // parameters.
63 // FIXME: This is getting out of hand, write a decent parser.
Manuel Klimekb3987012013-05-29 14:47:47 +000064 if (CurrentToken->Previous->isOneOf(tok::pipepipe, tok::ampamp) &&
Daniel Jasper0348be02013-06-01 18:56:00 +000065 (CurrentToken->Previous->Type == TT_BinaryOperator ||
66 Contexts[Contexts.size() - 2].IsExpression) &&
Manuel Klimekb3987012013-05-29 14:47:47 +000067 Line.First->isNot(tok::kw_template))
Daniel Jasper32d28ee2013-01-29 21:01:14 +000068 return false;
Daniel Jasper9fc56f22013-02-14 15:01:34 +000069 updateParameterCount(Left, CurrentToken);
Daniel Jasper32d28ee2013-01-29 21:01:14 +000070 if (!consumeToken())
71 return false;
72 }
73 return false;
74 }
75
76 bool parseParens(bool LookForDecls = false) {
77 if (CurrentToken == NULL)
78 return false;
Daniel Jasper923ebef2013-03-14 13:45:21 +000079 ScopedContextCreator ContextCreator(*this, tok::l_paren, 1);
Daniel Jasper4e778092013-02-06 10:05:46 +000080
81 // FIXME: This is a bit of a hack. Do better.
82 Contexts.back().ColonIsForRangeExpr =
83 Contexts.size() == 2 && Contexts[0].ColonIsForRangeExpr;
84
Daniel Jasper32d28ee2013-01-29 21:01:14 +000085 bool StartsObjCMethodExpr = false;
Manuel Klimekb3987012013-05-29 14:47:47 +000086 FormatToken *Left = CurrentToken->Previous;
Daniel Jasper32d28ee2013-01-29 21:01:14 +000087 if (CurrentToken->is(tok::caret)) {
88 // ^( starts a block.
89 Left->Type = TT_ObjCBlockLParen;
Manuel Klimekb3987012013-05-29 14:47:47 +000090 } else if (FormatToken *MaybeSel = Left->Previous) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +000091 // @selector( starts a selector.
Manuel Klimekb3987012013-05-29 14:47:47 +000092 if (MaybeSel->isObjCAtKeyword(tok::objc_selector) && MaybeSel->Previous &&
93 MaybeSel->Previous->is(tok::at)) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +000094 StartsObjCMethodExpr = true;
95 }
96 }
97
Daniel Jasperb644dd62013-08-13 09:09:09 +000098 if (Left->Previous && Left->Previous->isOneOf(tok::kw_static_assert,
99 tok::kw_if, tok::kw_while))
Daniel Jasperb7000ca2013-08-01 17:58:23 +0000100 Contexts.back().IsExpression = true;
101
Daniel Jasper4e778092013-02-06 10:05:46 +0000102 if (StartsObjCMethodExpr) {
103 Contexts.back().ColonIsObjCMethodExpr = true;
104 Left->Type = TT_ObjCMethodExpr;
105 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000106
Daniel Jasper431f5912013-05-28 08:33:00 +0000107 bool MightBeFunctionType = CurrentToken->is(tok::star);
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000108 bool HasMultipleLines = false;
109 bool HasMultipleParametersOnALine = false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000110 while (CurrentToken != NULL) {
111 // LookForDecls is set when "if (" has been seen. Check for
112 // 'identifier' '*' 'identifier' followed by not '=' -- this
113 // '*' has to be a binary operator but determineStarAmpUsage() will
114 // categorize it as an unary operator, so set the right type here.
Manuel Klimekb3987012013-05-29 14:47:47 +0000115 if (LookForDecls && CurrentToken->Next) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000116 FormatToken *Prev = CurrentToken->getPreviousNonComment();
Alexander Kornienko2785b9a2013-06-07 16:02:52 +0000117 if (Prev) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000118 FormatToken *PrevPrev = Prev->getPreviousNonComment();
Alexander Kornienko2785b9a2013-06-07 16:02:52 +0000119 FormatToken *Next = CurrentToken->Next;
120 if (PrevPrev && PrevPrev->is(tok::identifier) &&
121 Prev->isOneOf(tok::star, tok::amp, tok::ampamp) &&
122 CurrentToken->is(tok::identifier) && Next->isNot(tok::equal)) {
123 Prev->Type = TT_BinaryOperator;
124 LookForDecls = false;
125 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000126 }
127 }
128
Daniel Jasper53352602013-08-12 12:16:34 +0000129 if (CurrentToken->Previous->Type == TT_PointerOrReference &&
130 CurrentToken->Previous->Previous->isOneOf(tok::l_paren,
131 tok::coloncolon))
132 MightBeFunctionType = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000133 if (CurrentToken->is(tok::r_paren)) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000134 if (MightBeFunctionType && CurrentToken->Next &&
Daniel Jaspere7d3bff2013-07-16 11:37:21 +0000135 (CurrentToken->Next->is(tok::l_paren) ||
136 (CurrentToken->Next->is(tok::l_square) &&
137 !Contexts.back().IsExpression)))
Daniel Jasper431f5912013-05-28 08:33:00 +0000138 Left->Type = TT_FunctionTypeLParen;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000139 Left->MatchingParen = CurrentToken;
140 CurrentToken->MatchingParen = Left;
141
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000142 if (StartsObjCMethodExpr) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000143 CurrentToken->Type = TT_ObjCMethodExpr;
144 if (Contexts.back().FirstObjCSelectorName != NULL) {
145 Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
146 Contexts.back().LongestObjCSelectorName;
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000147 }
148 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000149
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000150 if (!HasMultipleLines)
151 Left->PackingKind = PPK_Inconclusive;
152 else if (HasMultipleParametersOnALine)
153 Left->PackingKind = PPK_BinPacked;
154 else
155 Left->PackingKind = PPK_OnePerLine;
156
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000157 next();
158 return true;
159 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000160 if (CurrentToken->isOneOf(tok::r_square, tok::r_brace))
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000161 return false;
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000162 updateParameterCount(Left, CurrentToken);
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000163 if (CurrentToken->is(tok::comma) && CurrentToken->Next &&
164 !CurrentToken->Next->HasUnescapedNewline &&
165 !CurrentToken->Next->isTrailingComment())
166 HasMultipleParametersOnALine = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000167 if (!consumeToken())
168 return false;
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000169 if (CurrentToken && CurrentToken->HasUnescapedNewline)
170 HasMultipleLines = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000171 }
172 return false;
173 }
174
175 bool parseSquare() {
176 if (!CurrentToken)
177 return false;
178
Alexander Kornienkod71b15b2013-06-17 13:19:53 +0000179 // A '[' could be an index subscript (after an identifier or after
Nico Weber051860e2013-02-10 02:08:05 +0000180 // ')' or ']'), it could be the start of an Objective-C method
181 // expression, or it could the the start of an Objective-C array literal.
Manuel Klimekb3987012013-05-29 14:47:47 +0000182 FormatToken *Left = CurrentToken->Previous;
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000183 FormatToken *Parent = Left->getPreviousNonComment();
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000184 bool StartsObjCMethodExpr =
Daniel Jasper6f21a982013-03-13 07:49:51 +0000185 Contexts.back().CanBeExpression &&
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000186 (!Parent || Parent->isOneOf(tok::colon, tok::l_square, tok::l_paren,
187 tok::kw_return, tok::kw_throw) ||
Daniel Jasperac3223e2013-04-10 09:49:49 +0000188 Parent->isUnaryOperator() || Parent->Type == TT_ObjCForIn ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000189 Parent->Type == TT_CastRParen ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000190 getBinOpPrecedence(Parent->Tok.getKind(), true, true) > prec::Unknown);
Daniel Jasper923ebef2013-03-14 13:45:21 +0000191 ScopedContextCreator ContextCreator(*this, tok::l_square, 10);
Daniel Jasper6f21a982013-03-13 07:49:51 +0000192 Contexts.back().IsExpression = true;
Nico Weber051860e2013-02-10 02:08:05 +0000193 bool StartsObjCArrayLiteral = Parent && Parent->is(tok::at);
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000194
Daniel Jasper4e778092013-02-06 10:05:46 +0000195 if (StartsObjCMethodExpr) {
196 Contexts.back().ColonIsObjCMethodExpr = true;
197 Left->Type = TT_ObjCMethodExpr;
Nico Weber051860e2013-02-10 02:08:05 +0000198 } else if (StartsObjCArrayLiteral) {
199 Left->Type = TT_ObjCArrayLiteral;
Daniel Jasper4e778092013-02-06 10:05:46 +0000200 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000201
202 while (CurrentToken != NULL) {
203 if (CurrentToken->is(tok::r_square)) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000204 if (CurrentToken->Next && CurrentToken->Next->is(tok::l_paren)) {
Nico Webere8a97982013-02-06 06:20:11 +0000205 // An ObjC method call is rarely followed by an open parenthesis.
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000206 // FIXME: Do we incorrectly label ":" with this?
207 StartsObjCMethodExpr = false;
208 Left->Type = TT_Unknown;
209 }
Daniel Jasper01786732013-02-04 07:21:18 +0000210 if (StartsObjCMethodExpr) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000211 CurrentToken->Type = TT_ObjCMethodExpr;
Nico Webere8a97982013-02-06 06:20:11 +0000212 // determineStarAmpUsage() thinks that '*' '[' is allocating an
213 // array of pointers, but if '[' starts a selector then '*' is a
214 // binary operator.
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +0000215 if (Parent != NULL && Parent->Type == TT_PointerOrReference)
Nico Weber4ed7f3e2013-02-06 16:54:35 +0000216 Parent->Type = TT_BinaryOperator;
Nico Weber051860e2013-02-10 02:08:05 +0000217 } else if (StartsObjCArrayLiteral) {
218 CurrentToken->Type = TT_ObjCArrayLiteral;
Daniel Jasper01786732013-02-04 07:21:18 +0000219 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000220 Left->MatchingParen = CurrentToken;
221 CurrentToken->MatchingParen = Left;
Daniel Jasper4e778092013-02-06 10:05:46 +0000222 if (Contexts.back().FirstObjCSelectorName != NULL)
223 Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
224 Contexts.back().LongestObjCSelectorName;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000225 next();
226 return true;
227 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000228 if (CurrentToken->isOneOf(tok::r_paren, tok::r_brace))
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000229 return false;
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000230 updateParameterCount(Left, CurrentToken);
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000231 if (!consumeToken())
232 return false;
233 }
234 return false;
235 }
236
237 bool parseBrace() {
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000238 if (CurrentToken != NULL) {
239 ScopedContextCreator ContextCreator(*this, tok::l_brace, 1);
Manuel Klimekb3987012013-05-29 14:47:47 +0000240 FormatToken *Left = CurrentToken->Previous;
Nico Weberf2ff8122013-05-26 05:39:26 +0000241
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000242 FormatToken *Parent = Left->getPreviousNonComment();
Nico Weberf2ff8122013-05-26 05:39:26 +0000243 bool StartsObjCDictLiteral = Parent && Parent->is(tok::at);
244 if (StartsObjCDictLiteral) {
245 Contexts.back().ColonIsObjCDictLiteral = true;
246 Left->Type = TT_ObjCDictLiteral;
247 }
248
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000249 while (CurrentToken != NULL) {
250 if (CurrentToken->is(tok::r_brace)) {
Nico Weberf2ff8122013-05-26 05:39:26 +0000251 if (StartsObjCDictLiteral)
252 CurrentToken->Type = TT_ObjCDictLiteral;
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000253 Left->MatchingParen = CurrentToken;
254 CurrentToken->MatchingParen = Left;
255 next();
256 return true;
257 }
258 if (CurrentToken->isOneOf(tok::r_paren, tok::r_square))
259 return false;
260 updateParameterCount(Left, CurrentToken);
261 if (!consumeToken())
262 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000263 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000264 }
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000265 // No closing "}" found, this probably starts a definition.
266 Line.StartsDefinition = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000267 return true;
268 }
Daniel Jasperc4615b72013-02-20 12:56:39 +0000269
Manuel Klimekb3987012013-05-29 14:47:47 +0000270 void updateParameterCount(FormatToken *Left, FormatToken *Current) {
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000271 if (Current->is(tok::comma)) {
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000272 ++Left->ParameterCount;
Daniel Jasperd4a03db2013-08-22 15:00:41 +0000273 if (!Left->Role)
274 Left->Role.reset(new CommaSeparatedList(Style));
275 Left->Role->CommaFound(Current);
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000276 } else if (Left->ParameterCount == 0 && Current->isNot(tok::comment)) {
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000277 Left->ParameterCount = 1;
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000278 }
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000279 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000280
281 bool parseConditional() {
282 while (CurrentToken != NULL) {
283 if (CurrentToken->is(tok::colon)) {
284 CurrentToken->Type = TT_ConditionalExpr;
285 next();
286 return true;
287 }
288 if (!consumeToken())
289 return false;
290 }
291 return false;
292 }
293
294 bool parseTemplateDeclaration() {
295 if (CurrentToken != NULL && CurrentToken->is(tok::less)) {
296 CurrentToken->Type = TT_TemplateOpener;
297 next();
298 if (!parseAngle())
299 return false;
Daniel Jasper34511fb2013-02-19 17:14:38 +0000300 if (CurrentToken != NULL)
Manuel Klimekb3987012013-05-29 14:47:47 +0000301 CurrentToken->Previous->ClosesTemplateDeclaration = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000302 return true;
303 }
304 return false;
305 }
306
307 bool consumeToken() {
Manuel Klimekb3987012013-05-29 14:47:47 +0000308 FormatToken *Tok = CurrentToken;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000309 next();
Manuel Klimekb3987012013-05-29 14:47:47 +0000310 switch (Tok->Tok.getKind()) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000311 case tok::plus:
312 case tok::minus:
Manuel Klimekb3987012013-05-29 14:47:47 +0000313 if (Tok->Previous == NULL && Line.MustBeDeclaration)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000314 Tok->Type = TT_ObjCMethodSpecifier;
315 break;
316 case tok::colon:
Manuel Klimekb3987012013-05-29 14:47:47 +0000317 if (Tok->Previous == NULL)
Daniel Jaspercf6d76a2013-03-18 12:50:26 +0000318 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000319 // Colons from ?: are handled in parseConditional().
Manuel Klimekb3987012013-05-29 14:47:47 +0000320 if (Tok->Previous->is(tok::r_paren) && Contexts.size() == 1) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000321 Tok->Type = TT_CtorInitializerColon;
Nico Weberf2ff8122013-05-26 05:39:26 +0000322 } else if (Contexts.back().ColonIsObjCDictLiteral) {
323 Tok->Type = TT_ObjCDictLiteral;
Daniel Jasper4e778092013-02-06 10:05:46 +0000324 } else if (Contexts.back().ColonIsObjCMethodExpr ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000325 Line.First->Type == TT_ObjCMethodSpecifier) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000326 Tok->Type = TT_ObjCMethodExpr;
Manuel Klimekb3987012013-05-29 14:47:47 +0000327 Tok->Previous->Type = TT_ObjCSelectorName;
Alexander Kornienko00895102013-06-05 14:09:10 +0000328 if (Tok->Previous->CodePointCount >
329 Contexts.back().LongestObjCSelectorName) {
330 Contexts.back().LongestObjCSelectorName =
331 Tok->Previous->CodePointCount;
332 }
Daniel Jasper4e778092013-02-06 10:05:46 +0000333 if (Contexts.back().FirstObjCSelectorName == NULL)
Manuel Klimekb3987012013-05-29 14:47:47 +0000334 Contexts.back().FirstObjCSelectorName = Tok->Previous;
Daniel Jasper4e778092013-02-06 10:05:46 +0000335 } else if (Contexts.back().ColonIsForRangeExpr) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000336 Tok->Type = TT_RangeBasedForLoopColon;
Daniel Jasper6cabab42013-02-14 08:42:54 +0000337 } else if (Contexts.size() == 1) {
338 Tok->Type = TT_InheritanceColon;
Daniel Jasper923ebef2013-03-14 13:45:21 +0000339 } else if (Contexts.back().ContextKind == tok::l_paren) {
340 Tok->Type = TT_InlineASMColon;
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000341 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000342 break;
343 case tok::kw_if:
344 case tok::kw_while:
345 if (CurrentToken != NULL && CurrentToken->is(tok::l_paren)) {
346 next();
Nico Weber27268772013-06-26 00:30:14 +0000347 if (!parseParens(/*LookForDecls=*/true))
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000348 return false;
349 }
350 break;
351 case tok::kw_for:
Daniel Jasper4e778092013-02-06 10:05:46 +0000352 Contexts.back().ColonIsForRangeExpr = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000353 next();
354 if (!parseParens())
355 return false;
356 break;
357 case tok::l_paren:
358 if (!parseParens())
359 return false;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000360 if (Line.MustBeDeclaration && NameFound && !Contexts.back().IsExpression)
Daniel Jasper3c08a812013-02-24 18:54:32 +0000361 Line.MightBeFunctionDecl = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000362 break;
363 case tok::l_square:
364 if (!parseSquare())
365 return false;
366 break;
367 case tok::l_brace:
368 if (!parseBrace())
369 return false;
370 break;
371 case tok::less:
Daniel Jasper0236dd02013-07-30 22:37:19 +0000372 if (Tok->Previous && !Tok->Previous->Tok.isLiteral() && parseAngle())
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000373 Tok->Type = TT_TemplateOpener;
374 else {
375 Tok->Type = TT_BinaryOperator;
376 CurrentToken = Tok;
377 next();
378 }
379 break;
380 case tok::r_paren:
381 case tok::r_square:
382 return false;
383 case tok::r_brace:
384 // Lines can start with '}'.
Manuel Klimekb3987012013-05-29 14:47:47 +0000385 if (Tok->Previous != NULL)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000386 return false;
387 break;
388 case tok::greater:
389 Tok->Type = TT_BinaryOperator;
390 break;
391 case tok::kw_operator:
Daniel Jasper174f60f2013-09-02 09:20:39 +0000392 while (CurrentToken &&
393 !CurrentToken->isOneOf(tok::l_paren, tok::semi, tok::r_paren)) {
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000394 if (CurrentToken->isOneOf(tok::star, tok::amp))
Daniel Jasper2b4c9242013-02-11 08:01:18 +0000395 CurrentToken->Type = TT_PointerOrReference;
396 consumeToken();
Daniel Jasper174f60f2013-09-02 09:20:39 +0000397 if (CurrentToken && CurrentToken->Previous->Type == TT_BinaryOperator)
Daniel Jasperc476ea92013-08-28 07:27:35 +0000398 CurrentToken->Previous->Type = TT_OverloadedOperator;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000399 }
Daniel Jasper6ea933c2013-05-10 07:59:58 +0000400 if (CurrentToken) {
Daniel Jasper2b4c9242013-02-11 08:01:18 +0000401 CurrentToken->Type = TT_OverloadedOperatorLParen;
Manuel Klimekb3987012013-05-29 14:47:47 +0000402 if (CurrentToken->Previous->Type == TT_BinaryOperator)
403 CurrentToken->Previous->Type = TT_OverloadedOperator;
Daniel Jasper6ea933c2013-05-10 07:59:58 +0000404 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000405 break;
406 case tok::question:
407 parseConditional();
408 break;
409 case tok::kw_template:
410 parseTemplateDeclaration();
411 break;
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000412 case tok::identifier:
Manuel Klimekb3987012013-05-29 14:47:47 +0000413 if (Line.First->is(tok::kw_for) &&
414 Tok->Tok.getIdentifierInfo() == &Ident_in)
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000415 Tok->Type = TT_ObjCForIn;
416 break;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000417 case tok::comma:
418 if (Contexts.back().FirstStartOfName)
419 Contexts.back().FirstStartOfName->PartOfMultiVariableDeclStmt = true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000420 if (Contexts.back().InCtorInitializer)
421 Tok->Type = TT_CtorInitializerComma;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000422 break;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000423 default:
424 break;
425 }
426 return true;
427 }
428
429 void parseIncludeDirective() {
430 next();
431 if (CurrentToken != NULL && CurrentToken->is(tok::less)) {
432 next();
433 while (CurrentToken != NULL) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000434 if (CurrentToken->isNot(tok::comment) || CurrentToken->Next)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000435 CurrentToken->Type = TT_ImplicitStringLiteral;
436 next();
437 }
438 } else {
439 while (CurrentToken != NULL) {
Daniel Jasper3a204412013-02-23 07:46:38 +0000440 if (CurrentToken->is(tok::string_literal))
441 // Mark these string literals as "implicit" literals, too, so that
442 // they are not split or line-wrapped.
443 CurrentToken->Type = TT_ImplicitStringLiteral;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000444 next();
445 }
446 }
447 }
448
449 void parseWarningOrError() {
450 next();
451 // We still want to format the whitespace left of the first token of the
452 // warning or error.
453 next();
454 while (CurrentToken != NULL) {
455 CurrentToken->Type = TT_ImplicitStringLiteral;
456 next();
457 }
458 }
459
460 void parsePreprocessorDirective() {
461 next();
462 if (CurrentToken == NULL)
463 return;
464 // Hashes in the middle of a line can lead to any strange token
465 // sequence.
Manuel Klimekb3987012013-05-29 14:47:47 +0000466 if (CurrentToken->Tok.getIdentifierInfo() == NULL)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000467 return;
Manuel Klimekb3987012013-05-29 14:47:47 +0000468 switch (CurrentToken->Tok.getIdentifierInfo()->getPPKeywordID()) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000469 case tok::pp_include:
470 case tok::pp_import:
471 parseIncludeDirective();
472 break;
473 case tok::pp_error:
474 case tok::pp_warning:
475 parseWarningOrError();
476 break;
Daniel Jasperaae7bad2013-04-23 13:54:04 +0000477 case tok::pp_if:
478 case tok::pp_elif:
479 parseLine();
480 break;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000481 default:
482 break;
483 }
Daniel Jasper5b7e7b02013-02-05 09:34:14 +0000484 while (CurrentToken != NULL)
485 next();
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000486 }
487
Nico Weber95e8e462013-02-12 16:17:07 +0000488public:
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000489 LineType parseLine() {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000490 if (CurrentToken->is(tok::hash)) {
491 parsePreprocessorDirective();
492 return LT_PreprocessorDirective;
493 }
494 while (CurrentToken != NULL) {
495 if (CurrentToken->is(tok::kw_virtual))
496 KeywordVirtualFound = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000497 if (!consumeToken())
498 return LT_Invalid;
499 }
500 if (KeywordVirtualFound)
501 return LT_VirtualFunctionDecl;
502
Manuel Klimekb3987012013-05-29 14:47:47 +0000503 if (Line.First->Type == TT_ObjCMethodSpecifier) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000504 if (Contexts.back().FirstObjCSelectorName != NULL)
505 Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
506 Contexts.back().LongestObjCSelectorName;
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000507 return LT_ObjCMethodDecl;
508 }
509
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000510 return LT_Other;
511 }
512
Nico Weber95e8e462013-02-12 16:17:07 +0000513private:
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000514 void next() {
Daniel Jasper01786732013-02-04 07:21:18 +0000515 if (CurrentToken != NULL) {
516 determineTokenType(*CurrentToken);
Daniel Jasper4e778092013-02-06 10:05:46 +0000517 CurrentToken->BindingStrength = Contexts.back().BindingStrength;
Daniel Jasper01786732013-02-04 07:21:18 +0000518 }
519
Manuel Klimekb3987012013-05-29 14:47:47 +0000520 if (CurrentToken != NULL)
521 CurrentToken = CurrentToken->Next;
Daniel Jasperd0f349b2013-02-18 12:44:35 +0000522
523 // Reset token type in case we have already looked at it and then recovered
524 // from an error (e.g. failure to find the matching >).
525 if (CurrentToken != NULL)
526 CurrentToken->Type = TT_Unknown;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000527 }
528
Daniel Jasper4e778092013-02-06 10:05:46 +0000529 /// \brief A struct to hold information valid in a specific context, e.g.
530 /// a pair of parenthesis.
531 struct Context {
Daniel Jasper923ebef2013-03-14 13:45:21 +0000532 Context(tok::TokenKind ContextKind, unsigned BindingStrength,
533 bool IsExpression)
534 : ContextKind(ContextKind), BindingStrength(BindingStrength),
535 LongestObjCSelectorName(0), ColonIsForRangeExpr(false),
Nico Weberf2ff8122013-05-26 05:39:26 +0000536 ColonIsObjCDictLiteral(false), ColonIsObjCMethodExpr(false),
537 FirstObjCSelectorName(NULL), FirstStartOfName(NULL),
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000538 IsExpression(IsExpression), CanBeExpression(true),
539 InCtorInitializer(false) {}
Daniel Jasper01786732013-02-04 07:21:18 +0000540
Daniel Jasper923ebef2013-03-14 13:45:21 +0000541 tok::TokenKind ContextKind;
Daniel Jasper4e778092013-02-06 10:05:46 +0000542 unsigned BindingStrength;
543 unsigned LongestObjCSelectorName;
544 bool ColonIsForRangeExpr;
Nico Weberf2ff8122013-05-26 05:39:26 +0000545 bool ColonIsObjCDictLiteral;
Daniel Jasper4e778092013-02-06 10:05:46 +0000546 bool ColonIsObjCMethodExpr;
Manuel Klimekb3987012013-05-29 14:47:47 +0000547 FormatToken *FirstObjCSelectorName;
548 FormatToken *FirstStartOfName;
Daniel Jasper4e778092013-02-06 10:05:46 +0000549 bool IsExpression;
Daniel Jasper6f21a982013-03-13 07:49:51 +0000550 bool CanBeExpression;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000551 bool InCtorInitializer;
Daniel Jasper4e778092013-02-06 10:05:46 +0000552 };
553
554 /// \brief Puts a new \c Context onto the stack \c Contexts for the lifetime
555 /// of each instance.
556 struct ScopedContextCreator {
557 AnnotatingParser &P;
558
Daniel Jasper923ebef2013-03-14 13:45:21 +0000559 ScopedContextCreator(AnnotatingParser &P, tok::TokenKind ContextKind,
560 unsigned Increase)
561 : P(P) {
Daniel Jasper2a409b62013-07-08 14:34:09 +0000562 P.Contexts.push_back(Context(ContextKind,
563 P.Contexts.back().BindingStrength + Increase,
564 P.Contexts.back().IsExpression));
Daniel Jasper4e778092013-02-06 10:05:46 +0000565 }
566
567 ~ScopedContextCreator() { P.Contexts.pop_back(); }
568 };
Daniel Jasper01786732013-02-04 07:21:18 +0000569
Manuel Klimekb3987012013-05-29 14:47:47 +0000570 void determineTokenType(FormatToken &Current) {
571 if (Current.getPrecedence() == prec::Assignment &&
Daniel Jasper53352602013-08-12 12:16:34 +0000572 !Line.First->isOneOf(tok::kw_template, tok::kw_using) &&
Manuel Klimekb3987012013-05-29 14:47:47 +0000573 (!Current.Previous || Current.Previous->isNot(tok::kw_operator))) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000574 Contexts.back().IsExpression = true;
Manuel Klimekb3987012013-05-29 14:47:47 +0000575 for (FormatToken *Previous = Current.Previous;
Nico Weber95e8e462013-02-12 16:17:07 +0000576 Previous && Previous->isNot(tok::comma);
Manuel Klimekb3987012013-05-29 14:47:47 +0000577 Previous = Previous->Previous) {
Daniel Jasper9c65b062013-02-27 11:43:50 +0000578 if (Previous->is(tok::r_square))
579 Previous = Previous->MatchingParen;
Daniel Jasper01786732013-02-04 07:21:18 +0000580 if (Previous->Type == TT_BinaryOperator &&
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000581 Previous->isOneOf(tok::star, tok::amp)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000582 Previous->Type = TT_PointerOrReference;
583 }
Daniel Jasper01786732013-02-04 07:21:18 +0000584 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000585 } else if (Current.isOneOf(tok::kw_return, tok::kw_throw) ||
Nico Weber95e8e462013-02-12 16:17:07 +0000586 (Current.is(tok::l_paren) && !Line.MustBeDeclaration &&
Daniel Jasper378d93d2013-05-13 07:14:40 +0000587 !Line.InPPDirective &&
Manuel Klimek2530fd52013-08-12 03:51:17 +0000588 (!Current.Previous ||
Daniel Jasper53352602013-08-12 12:16:34 +0000589 !Current.Previous->isOneOf(tok::kw_for, tok::kw_catch)))) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000590 Contexts.back().IsExpression = true;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000591 } else if (Current.isOneOf(tok::r_paren, tok::greater, tok::comma)) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000592 for (FormatToken *Previous = Current.Previous;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000593 Previous && Previous->isOneOf(tok::star, tok::amp);
Manuel Klimekb3987012013-05-29 14:47:47 +0000594 Previous = Previous->Previous)
Nico Weber95e8e462013-02-12 16:17:07 +0000595 Previous->Type = TT_PointerOrReference;
Manuel Klimekb3987012013-05-29 14:47:47 +0000596 } else if (Current.Previous &&
597 Current.Previous->Type == TT_CtorInitializerColon) {
Daniel Jasperd0f349b2013-02-18 12:44:35 +0000598 Contexts.back().IsExpression = true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000599 Contexts.back().InCtorInitializer = true;
Daniel Jasper6f21a982013-03-13 07:49:51 +0000600 } else if (Current.is(tok::kw_new)) {
601 Contexts.back().CanBeExpression = false;
Daniel Jasper16a69ef2013-05-03 14:41:24 +0000602 } else if (Current.is(tok::semi)) {
603 // This should be the condition or increment in a for-loop.
604 Contexts.back().IsExpression = true;
Nico Weber95e8e462013-02-12 16:17:07 +0000605 }
Daniel Jasper01786732013-02-04 07:21:18 +0000606
607 if (Current.Type == TT_Unknown) {
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000608 if (isStartOfName(Current)) {
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000609 Contexts.back().FirstStartOfName = &Current;
Daniel Jasper3c08a812013-02-24 18:54:32 +0000610 Current.Type = TT_StartOfName;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000611 NameFound = true;
Daniel Jasper2ca37412013-07-09 14:36:48 +0000612 } else if (Current.is(tok::kw_auto)) {
613 AutoFound = true;
Daniel Jasper3262f4c2013-07-11 14:33:06 +0000614 } else if (Current.is(tok::arrow) && AutoFound &&
615 Line.MustBeDeclaration) {
Daniel Jasper2ca37412013-07-09 14:36:48 +0000616 Current.Type = TT_TrailingReturnArrow;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000617 } else if (Current.isOneOf(tok::star, tok::amp, tok::ampamp)) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000618 Current.Type =
Daniel Jasperd6104f62013-07-05 13:30:40 +0000619 determineStarAmpUsage(Current, Contexts.back().CanBeExpression &&
620 Contexts.back().IsExpression);
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000621 } else if (Current.isOneOf(tok::minus, tok::plus, tok::caret)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000622 Current.Type = determinePlusMinusCaretUsage(Current);
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000623 } else if (Current.isOneOf(tok::minusminus, tok::plusplus)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000624 Current.Type = determineIncrementUsage(Current);
625 } else if (Current.is(tok::exclaim)) {
626 Current.Type = TT_UnaryOperator;
Manuel Klimek31e44f72013-09-04 08:20:47 +0000627 } else if (Current.isBinaryOperator() &&
628 (!Current.Previous ||
629 Current.Previous->isNot(tok::l_square))) {
Daniel Jasper01786732013-02-04 07:21:18 +0000630 Current.Type = TT_BinaryOperator;
631 } else if (Current.is(tok::comment)) {
Alexander Kornienko00895102013-06-05 14:09:10 +0000632 if (Current.TokenText.startswith("//"))
Daniel Jasper01786732013-02-04 07:21:18 +0000633 Current.Type = TT_LineComment;
634 else
635 Current.Type = TT_BlockComment;
Nico Weber37d69312013-02-13 04:13:13 +0000636 } else if (Current.is(tok::r_paren)) {
Daniel Jasperb8b42952013-05-31 16:14:28 +0000637 FormatToken *LeftOfParens = NULL;
638 if (Current.MatchingParen)
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000639 LeftOfParens = Current.MatchingParen->getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000640 bool IsCast = false;
641 bool ParensAreEmpty = Current.Previous == Current.MatchingParen;
642 bool ParensAreType = !Current.Previous ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000643 Current.Previous->Type == TT_PointerOrReference ||
Daniel Jasperb8b42952013-05-31 16:14:28 +0000644 Current.Previous->Type == TT_TemplateCloser ||
645 isSimpleTypeSpecifier(*Current.Previous);
Nico Weber37d69312013-02-13 04:13:13 +0000646 bool ParensCouldEndDecl =
Manuel Klimekb3987012013-05-29 14:47:47 +0000647 Current.Next &&
648 Current.Next->isOneOf(tok::equal, tok::semi, tok::l_brace);
Daniel Jasper6a365aa2013-03-13 17:13:53 +0000649 bool IsSizeOfOrAlignOf =
Daniel Jasperb8b42952013-05-31 16:14:28 +0000650 LeftOfParens &&
651 LeftOfParens->isOneOf(tok::kw_sizeof, tok::kw_alignof);
652 if (ParensAreType && !ParensCouldEndDecl && !IsSizeOfOrAlignOf &&
Daniel Jasper0c368782013-07-15 15:04:42 +0000653 (Contexts.back().IsExpression ||
654 (Current.Next && Current.Next->isBinaryOperator())))
Daniel Jasperb8b42952013-05-31 16:14:28 +0000655 IsCast = true;
Daniel Jasper2a409b62013-07-08 14:34:09 +0000656 if (Current.Next && Current.Next->isNot(tok::string_literal) &&
Daniel Jasperb8b42952013-05-31 16:14:28 +0000657 (Current.Next->Tok.isLiteral() ||
658 Current.Next->isOneOf(tok::kw_sizeof, tok::kw_alignof)))
659 IsCast = true;
660 // If there is an identifier after the (), it is likely a cast, unless
661 // there is also an identifier before the ().
Daniel Jasperff1a2e52013-06-06 08:20:20 +0000662 if (LeftOfParens && (LeftOfParens->Tok.getIdentifierInfo() == NULL ||
663 LeftOfParens->is(tok::kw_return)) &&
Daniel Jasper526df0f2013-07-08 14:58:01 +0000664 LeftOfParens->Type != TT_OverloadedOperator &&
Nico Weber465e8612013-06-25 00:55:57 +0000665 LeftOfParens->Type != TT_TemplateCloser && Current.Next &&
666 Current.Next->is(tok::identifier))
Daniel Jasperb8b42952013-05-31 16:14:28 +0000667 IsCast = true;
668 if (IsCast && !ParensAreEmpty)
Nico Weber37d69312013-02-13 04:13:13 +0000669 Current.Type = TT_CastRParen;
Manuel Klimekb3987012013-05-29 14:47:47 +0000670 } else if (Current.is(tok::at) && Current.Next) {
671 switch (Current.Next->Tok.getObjCKeywordID()) {
Daniel Jasper01786732013-02-04 07:21:18 +0000672 case tok::objc_interface:
673 case tok::objc_implementation:
674 case tok::objc_protocol:
675 Current.Type = TT_ObjCDecl;
676 break;
677 case tok::objc_property:
678 Current.Type = TT_ObjCProperty;
679 break;
680 default:
681 break;
682 }
Daniel Jasper5ad390d2013-05-28 11:30:49 +0000683 } else if (Current.is(tok::period)) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000684 FormatToken *PreviousNoComment = Current.getPreviousNonComment();
Daniel Jasper5ad390d2013-05-28 11:30:49 +0000685 if (PreviousNoComment &&
686 PreviousNoComment->isOneOf(tok::comma, tok::l_brace))
687 Current.Type = TT_DesignatedInitializerPeriod;
Daniel Jasper01786732013-02-04 07:21:18 +0000688 }
689 }
690 }
691
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000692 /// \brief Take a guess at whether \p Tok starts a name of a function or
693 /// variable declaration.
694 ///
695 /// This is a heuristic based on whether \p Tok is an identifier following
696 /// something that is likely a type.
697 bool isStartOfName(const FormatToken &Tok) {
698 if (Tok.isNot(tok::identifier) || Tok.Previous == NULL)
699 return false;
700
701 // Skip "const" as it does not have an influence on whether this is a name.
702 FormatToken *PreviousNotConst = Tok.Previous;
703 while (PreviousNotConst != NULL && PreviousNotConst->is(tok::kw_const))
704 PreviousNotConst = PreviousNotConst->Previous;
705
706 if (PreviousNotConst == NULL)
707 return false;
708
Daniel Jasper2a409b62013-07-08 14:34:09 +0000709 bool IsPPKeyword = PreviousNotConst->is(tok::identifier) &&
710 PreviousNotConst->Previous &&
711 PreviousNotConst->Previous->is(tok::hash);
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000712
Daniel Jasper92495a82013-08-19 10:16:18 +0000713 if (PreviousNotConst->Type == TT_TemplateCloser)
714 return PreviousNotConst && PreviousNotConst->MatchingParen &&
715 PreviousNotConst->MatchingParen->Previous &&
716 PreviousNotConst->MatchingParen->Previous->isNot(tok::kw_template);
717
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000718 return (!IsPPKeyword && PreviousNotConst->is(tok::identifier)) ||
719 PreviousNotConst->Type == TT_PointerOrReference ||
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000720 isSimpleTypeSpecifier(*PreviousNotConst);
721 }
722
Daniel Jasper01786732013-02-04 07:21:18 +0000723 /// \brief Return the type of the given token assuming it is * or &.
Manuel Klimekb3987012013-05-29 14:47:47 +0000724 TokenType determineStarAmpUsage(const FormatToken &Tok, bool IsExpression) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000725 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasper01786732013-02-04 07:21:18 +0000726 if (PrevToken == NULL)
727 return TT_UnaryOperator;
728
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000729 const FormatToken *NextToken = Tok.getNextNonComment();
Daniel Jasper01786732013-02-04 07:21:18 +0000730 if (NextToken == NULL)
731 return TT_Unknown;
732
Daniel Jasper431f5912013-05-28 08:33:00 +0000733 if (PrevToken->is(tok::coloncolon) ||
734 (PrevToken->is(tok::l_paren) && !IsExpression))
Daniel Jasper8a5d7cd2013-03-01 17:13:29 +0000735 return TT_PointerOrReference;
736
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000737 if (PrevToken->isOneOf(tok::l_paren, tok::l_square, tok::l_brace,
Daniel Jasperd3cf17b2013-03-14 10:50:25 +0000738 tok::comma, tok::semi, tok::kw_return, tok::colon,
Daniel Jasperdbef71e2013-05-07 14:17:18 +0000739 tok::equal, tok::kw_delete) ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000740 PrevToken->Type == TT_BinaryOperator ||
Daniel Jasper01786732013-02-04 07:21:18 +0000741 PrevToken->Type == TT_UnaryOperator || PrevToken->Type == TT_CastRParen)
742 return TT_UnaryOperator;
743
Nico Webere8a97982013-02-06 06:20:11 +0000744 if (NextToken->is(tok::l_square))
745 return TT_PointerOrReference;
746
Manuel Klimekb3987012013-05-29 14:47:47 +0000747 if (PrevToken->Tok.isLiteral() ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000748 PrevToken->isOneOf(tok::r_paren, tok::r_square) ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000749 NextToken->Tok.isLiteral() || NextToken->isUnaryOperator())
Daniel Jasper01786732013-02-04 07:21:18 +0000750 return TT_BinaryOperator;
751
Daniel Jasper01786732013-02-04 07:21:18 +0000752 // It is very unlikely that we are going to find a pointer or reference type
753 // definition on the RHS of an assignment.
754 if (IsExpression)
755 return TT_BinaryOperator;
756
757 return TT_PointerOrReference;
758 }
759
Manuel Klimekb3987012013-05-29 14:47:47 +0000760 TokenType determinePlusMinusCaretUsage(const FormatToken &Tok) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000761 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000762 if (PrevToken == NULL || PrevToken->Type == TT_CastRParen)
Daniel Jasper01786732013-02-04 07:21:18 +0000763 return TT_UnaryOperator;
764
765 // Use heuristics to recognize unary operators.
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000766 if (PrevToken->isOneOf(tok::equal, tok::l_paren, tok::comma, tok::l_square,
767 tok::question, tok::colon, tok::kw_return,
768 tok::kw_case, tok::at, tok::l_brace))
Daniel Jasper01786732013-02-04 07:21:18 +0000769 return TT_UnaryOperator;
770
Nico Weberee0feec2013-02-05 16:21:00 +0000771 // There can't be two consecutive binary operators.
Daniel Jasper01786732013-02-04 07:21:18 +0000772 if (PrevToken->Type == TT_BinaryOperator)
773 return TT_UnaryOperator;
774
775 // Fall back to marking the token as binary operator.
776 return TT_BinaryOperator;
777 }
778
779 /// \brief Determine whether ++/-- are pre- or post-increments/-decrements.
Manuel Klimekb3987012013-05-29 14:47:47 +0000780 TokenType determineIncrementUsage(const FormatToken &Tok) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000781 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000782 if (PrevToken == NULL || PrevToken->Type == TT_CastRParen)
Daniel Jasper01786732013-02-04 07:21:18 +0000783 return TT_UnaryOperator;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000784 if (PrevToken->isOneOf(tok::r_paren, tok::r_square, tok::identifier))
Daniel Jasper01786732013-02-04 07:21:18 +0000785 return TT_TrailingUnaryOperator;
786
787 return TT_UnaryOperator;
788 }
Daniel Jasper4e778092013-02-06 10:05:46 +0000789
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000790 // FIXME: This is copy&pasted from Sema. Put it in a common place and remove
791 // duplication.
792 /// \brief Determine whether the token kind starts a simple-type-specifier.
Manuel Klimekb3987012013-05-29 14:47:47 +0000793 bool isSimpleTypeSpecifier(const FormatToken &Tok) const {
794 switch (Tok.Tok.getKind()) {
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000795 case tok::kw_short:
796 case tok::kw_long:
797 case tok::kw___int64:
798 case tok::kw___int128:
799 case tok::kw_signed:
800 case tok::kw_unsigned:
801 case tok::kw_void:
802 case tok::kw_char:
803 case tok::kw_int:
804 case tok::kw_half:
805 case tok::kw_float:
806 case tok::kw_double:
807 case tok::kw_wchar_t:
808 case tok::kw_bool:
809 case tok::kw___underlying_type:
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000810 case tok::annot_typename:
811 case tok::kw_char16_t:
812 case tok::kw_char32_t:
813 case tok::kw_typeof:
814 case tok::kw_decltype:
Alexander Kornienko00895102013-06-05 14:09:10 +0000815 return true;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000816 default:
Alexander Kornienko00895102013-06-05 14:09:10 +0000817 return false;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000818 }
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000819 }
820
Daniel Jasper4e778092013-02-06 10:05:46 +0000821 SmallVector<Context, 8> Contexts;
822
Daniel Jasperd4a03db2013-08-22 15:00:41 +0000823 const FormatStyle &Style;
Daniel Jasper4e778092013-02-06 10:05:46 +0000824 AnnotatedLine &Line;
Manuel Klimekb3987012013-05-29 14:47:47 +0000825 FormatToken *CurrentToken;
Daniel Jasper4e778092013-02-06 10:05:46 +0000826 bool KeywordVirtualFound;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000827 bool NameFound;
Daniel Jasper2ca37412013-07-09 14:36:48 +0000828 bool AutoFound;
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000829 IdentifierInfo &Ident_in;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000830};
831
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000832static int PrecedenceUnaryOperator = prec::PointerToMember + 1;
833static int PrecedenceArrowAndPeriod = prec::PointerToMember + 2;
834
Daniel Jasper29f123b2013-02-08 15:28:42 +0000835/// \brief Parses binary expressions by inserting fake parenthesis based on
836/// operator precedence.
837class ExpressionParser {
838public:
Daniel Jasper9acb8b42013-06-06 09:11:58 +0000839 ExpressionParser(AnnotatedLine &Line) : Current(Line.First) {
840 // Skip leading "}", e.g. in "} else if (...) {".
841 if (Current->is(tok::r_brace))
842 next();
843 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000844
845 /// \brief Parse expressions with the given operatore precedence.
Daniel Jasper237d4c12013-02-23 21:01:55 +0000846 void parse(int Precedence = 0) {
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000847 if (Current == NULL || Precedence > PrecedenceArrowAndPeriod)
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000848 return;
849
Daniel Jasperc01897c2013-05-31 14:56:12 +0000850 // Conditional expressions need to be parsed separately for proper nesting.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000851 if (Precedence == prec::Conditional) {
Daniel Jasperc01897c2013-05-31 14:56:12 +0000852 parseConditionalExpr();
853 return;
854 }
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000855
856 // Parse unary operators, which all have a higher precedence than binary
857 // operators.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000858 if (Precedence == PrecedenceUnaryOperator) {
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000859 parseUnaryOperator();
Daniel Jasper29f123b2013-02-08 15:28:42 +0000860 return;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000861 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000862
Manuel Klimekb3987012013-05-29 14:47:47 +0000863 FormatToken *Start = Current;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000864 FormatToken *LatestOperator = NULL;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000865
Daniel Jasper237d4c12013-02-23 21:01:55 +0000866 while (Current) {
Daniel Jasper29f123b2013-02-08 15:28:42 +0000867 // Consume operators with higher precedence.
Daniel Jasperbf71ba22013-04-08 20:33:42 +0000868 parse(Precedence + 1);
Daniel Jasper29f123b2013-02-08 15:28:42 +0000869
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000870 int CurrentPrecedence = getCurrentPrecedence();
871
872 if (Current && Current->Type == TT_ObjCSelectorName &&
873 Precedence == CurrentPrecedence)
874 Start = Current;
Daniel Jasper237d4c12013-02-23 21:01:55 +0000875
Daniel Jasper29f123b2013-02-08 15:28:42 +0000876 // At the end of the line or when an operator with higher precedence is
877 // found, insert fake parenthesis and return.
Daniel Jasperac3223e2013-04-10 09:49:49 +0000878 if (Current == NULL || Current->closesScope() ||
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000879 (CurrentPrecedence != -1 && CurrentPrecedence < Precedence)) {
880 if (LatestOperator) {
881 if (Precedence == PrecedenceArrowAndPeriod) {
882 LatestOperator->LastInChainOfCalls = true;
883 // Call expressions don't have a binary operator precedence.
884 addFakeParenthesis(Start, prec::Unknown);
885 } else {
886 addFakeParenthesis(Start, prec::Level(Precedence));
887 }
888 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000889 return;
890 }
891
892 // Consume scopes: (), [], <> and {}
Daniel Jasperac3223e2013-04-10 09:49:49 +0000893 if (Current->opensScope()) {
894 while (Current && !Current->closesScope()) {
Daniel Jasper29f123b2013-02-08 15:28:42 +0000895 next();
896 parse();
897 }
898 next();
899 } else {
900 // Operator found.
Daniel Jasper237d4c12013-02-23 21:01:55 +0000901 if (CurrentPrecedence == Precedence)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000902 LatestOperator = Current;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000903
904 next();
905 }
906 }
907 }
908
909private:
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000910 /// \brief Gets the precedence (+1) of the given token for binary operators
911 /// and other tokens that we treat like binary operators.
912 int getCurrentPrecedence() {
913 if (Current) {
914 if (Current->Type == TT_ConditionalExpr)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000915 return prec::Conditional;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000916 else if (Current->is(tok::semi) || Current->Type == TT_InlineASMColon)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000917 return 0;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000918 else if (Current->Type == TT_BinaryOperator || Current->is(tok::comma))
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000919 return Current->getPrecedence();
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000920 else if (Current->Type == TT_ObjCSelectorName)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000921 return prec::Assignment;
922 else if (Current->isOneOf(tok::period, tok::arrow))
923 return PrecedenceArrowAndPeriod;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000924 }
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000925 return -1;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000926 }
927
Daniel Jasperc01897c2013-05-31 14:56:12 +0000928 void addFakeParenthesis(FormatToken *Start, prec::Level Precedence) {
929 Start->FakeLParens.push_back(Precedence);
930 if (Current)
931 ++Current->Previous->FakeRParens;
932 }
933
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000934 /// \brief Parse unary operator expressions and surround them with fake
935 /// parentheses if appropriate.
936 void parseUnaryOperator() {
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000937 if (Current == NULL || Current->Type != TT_UnaryOperator) {
938 parse(PrecedenceArrowAndPeriod);
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000939 return;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000940 }
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000941
942 FormatToken *Start = Current;
943 next();
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000944 parseUnaryOperator();
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000945
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000946 // The actual precedence doesn't matter.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000947 addFakeParenthesis(Start, prec::Unknown);
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000948 }
949
Daniel Jasperc01897c2013-05-31 14:56:12 +0000950 void parseConditionalExpr() {
951 FormatToken *Start = Current;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000952 parse(prec::LogicalOr);
Daniel Jasperc01897c2013-05-31 14:56:12 +0000953 if (!Current || !Current->is(tok::question))
954 return;
955 next();
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000956 parse(prec::LogicalOr);
Daniel Jasperc01897c2013-05-31 14:56:12 +0000957 if (!Current || Current->Type != TT_ConditionalExpr)
958 return;
959 next();
960 parseConditionalExpr();
961 addFakeParenthesis(Start, prec::Conditional);
962 }
963
Daniel Jasper29f123b2013-02-08 15:28:42 +0000964 void next() {
Alexander Kornienkod71b15b2013-06-17 13:19:53 +0000965 if (Current)
966 Current = Current->Next;
967 while (Current && Current->isTrailingComment())
Manuel Klimekb3987012013-05-29 14:47:47 +0000968 Current = Current->Next;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000969 }
970
Manuel Klimekb3987012013-05-29 14:47:47 +0000971 FormatToken *Current;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000972};
973
Craig Topper14e66492013-07-01 04:03:19 +0000974} // end anonymous namespace
975
Daniel Jasper8ff690a2013-02-06 14:22:40 +0000976void TokenAnnotator::annotate(AnnotatedLine &Line) {
Daniel Jasperd4a03db2013-08-22 15:00:41 +0000977 AnnotatingParser Parser(Style, Line, Ident_in);
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000978 Line.Type = Parser.parseLine();
979 if (Line.Type == LT_Invalid)
980 return;
981
Daniel Jasper29f123b2013-02-08 15:28:42 +0000982 ExpressionParser ExprParser(Line);
983 ExprParser.parse();
984
Manuel Klimekb3987012013-05-29 14:47:47 +0000985 if (Line.First->Type == TT_ObjCMethodSpecifier)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000986 Line.Type = LT_ObjCMethodDecl;
Manuel Klimekb3987012013-05-29 14:47:47 +0000987 else if (Line.First->Type == TT_ObjCDecl)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000988 Line.Type = LT_ObjCDecl;
Manuel Klimekb3987012013-05-29 14:47:47 +0000989 else if (Line.First->Type == TT_ObjCProperty)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000990 Line.Type = LT_ObjCProperty;
991
Manuel Klimekb3987012013-05-29 14:47:47 +0000992 Line.First->SpacesRequiredBefore = 1;
993 Line.First->CanBreakBefore = Line.First->MustBreakBefore;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000994}
995
Daniel Jasper8ff690a2013-02-06 14:22:40 +0000996void TokenAnnotator::calculateFormattingInformation(AnnotatedLine &Line) {
Alexander Kornienko00895102013-06-05 14:09:10 +0000997 Line.First->TotalLength = Line.First->CodePointCount;
Manuel Klimekb3987012013-05-29 14:47:47 +0000998 if (!Line.First->Next)
Daniel Jasper8ff690a2013-02-06 14:22:40 +0000999 return;
Manuel Klimekb3987012013-05-29 14:47:47 +00001000 FormatToken *Current = Line.First->Next;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001001 while (Current != NULL) {
Daniel Jasper729a7432013-02-11 12:36:37 +00001002 if (Current->Type == TT_LineComment)
1003 Current->SpacesRequiredBefore = Style.SpacesBeforeTrailingComments;
1004 else
1005 Current->SpacesRequiredBefore =
1006 spaceRequiredBefore(Line, *Current) ? 1 : 0;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001007
Daniel Jasper561211d2013-07-16 20:28:33 +00001008 if (Current->is(tok::comment)) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001009 Current->MustBreakBefore = Current->NewlinesBefore > 0;
1010 } else if (Current->Previous->isTrailingComment() ||
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001011 (Current->is(tok::string_literal) &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001012 Current->Previous->is(tok::string_literal))) {
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001013 Current->MustBreakBefore = true;
Daniel Jasper561211d2013-07-16 20:28:33 +00001014 } else if (Current->Previous->IsUnterminatedLiteral) {
1015 Current->MustBreakBefore = true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001016 } else if (Current->is(tok::lessless) && Current->Next &&
1017 Current->Previous->is(tok::string_literal) &&
1018 Current->Next->is(tok::string_literal)) {
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001019 Current->MustBreakBefore = true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001020 } else if (Current->Previous->ClosesTemplateDeclaration &&
Daniel Jasperbbc87762013-05-29 12:07:31 +00001021 Style.AlwaysBreakTemplateDeclarations) {
1022 Current->MustBreakBefore = true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +00001023 } else if (Current->Type == TT_CtorInitializerComma &&
1024 Style.BreakConstructorInitializersBeforeComma) {
1025 Current->MustBreakBefore = true;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001026 }
1027 Current->CanBreakBefore =
1028 Current->MustBreakBefore || canBreakBefore(Line, *Current);
Daniel Jasper215c57f2013-07-17 15:38:19 +00001029 if (Current->MustBreakBefore ||
Alexander Kornienko4b762a92013-09-02 13:58:14 +00001030 (Current->is(tok::string_literal) && Current->isMultiline()))
Manuel Klimekb3987012013-05-29 14:47:47 +00001031 Current->TotalLength = Current->Previous->TotalLength + Style.ColumnLimit;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001032 else
Daniel Jasper2a409b62013-07-08 14:34:09 +00001033 Current->TotalLength = Current->Previous->TotalLength +
1034 Current->CodePointCount +
1035 Current->SpacesRequiredBefore;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001036 // FIXME: Only calculate this if CanBreakBefore is true once static
1037 // initializers etc. are sorted out.
1038 // FIXME: Move magic numbers to a better place.
1039 Current->SplitPenalty =
1040 20 * Current->BindingStrength + splitPenalty(Line, *Current);
1041
Manuel Klimekb3987012013-05-29 14:47:47 +00001042 Current = Current->Next;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001043 }
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001044
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001045 calculateUnbreakableTailLengths(Line);
Daniel Jasperd4a03db2013-08-22 15:00:41 +00001046 for (Current = Line.First; Current != NULL; Current = Current->Next) {
1047 if (Current->Role)
1048 Current->Role->precomputeFormattingInfos(Current);
1049 }
1050
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001051 DEBUG({
1052 printDebugInfo(Line);
1053 });
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001054}
1055
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001056void TokenAnnotator::calculateUnbreakableTailLengths(AnnotatedLine &Line) {
1057 unsigned UnbreakableTailLength = 0;
Manuel Klimekb3987012013-05-29 14:47:47 +00001058 FormatToken *Current = Line.Last;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001059 while (Current != NULL) {
1060 Current->UnbreakableTailLength = UnbreakableTailLength;
1061 if (Current->CanBreakBefore ||
1062 Current->isOneOf(tok::comment, tok::string_literal)) {
1063 UnbreakableTailLength = 0;
1064 } else {
1065 UnbreakableTailLength +=
Alexander Kornienko00895102013-06-05 14:09:10 +00001066 Current->CodePointCount + Current->SpacesRequiredBefore;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001067 }
Manuel Klimekb3987012013-05-29 14:47:47 +00001068 Current = Current->Previous;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001069 }
1070}
1071
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001072unsigned TokenAnnotator::splitPenalty(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001073 const FormatToken &Tok) {
1074 const FormatToken &Left = *Tok.Previous;
1075 const FormatToken &Right = Tok;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001076
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001077 if (Left.is(tok::semi))
1078 return 0;
1079 if (Left.is(tok::comma))
1080 return 1;
Daniel Jasper011c35d2013-07-12 11:19:37 +00001081 if (Right.is(tok::l_square))
1082 return 150;
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001083
Daniel Jasper6561f6a2013-07-09 07:43:55 +00001084 if (Right.Type == TT_StartOfName || Right.is(tok::kw_operator)) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001085 if (Line.First->is(tok::kw_for) && Right.PartOfMultiVariableDeclStmt)
Daniel Jasper3c08a812013-02-24 18:54:32 +00001086 return 3;
Daniel Jasperc18cff32013-07-11 12:34:23 +00001087 if (Left.Type == TT_StartOfName)
1088 return 20;
Daniel Jasper92495a82013-08-19 10:16:18 +00001089 if (Line.MightBeFunctionDecl && Right.BindingStrength == 1)
Daniel Jasper3c08a812013-02-24 18:54:32 +00001090 // FIXME: Clean up hack of using BindingStrength to find top-level names.
1091 return Style.PenaltyReturnTypeOnItsOwnLine;
Daniel Jasper92495a82013-08-19 10:16:18 +00001092 return 200;
Daniel Jasper3c08a812013-02-24 18:54:32 +00001093 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001094 if (Left.is(tok::equal) && Right.is(tok::l_brace))
1095 return 150;
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001096 if (Left.Type == TT_CastRParen)
1097 return 100;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001098 if (Left.is(tok::coloncolon))
1099 return 500;
Daniel Jasper6b119d62013-04-05 17:22:09 +00001100 if (Left.isOneOf(tok::kw_class, tok::kw_struct))
1101 return 5000;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001102
Daniel Jasper6cabab42013-02-14 08:42:54 +00001103 if (Left.Type == TT_RangeBasedForLoopColon ||
1104 Left.Type == TT_InheritanceColon)
Daniel Jasper84a1a632013-02-26 13:18:08 +00001105 return 2;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001106
Daniel Jasperd3fef0f2013-08-27 14:24:43 +00001107 if (Right.isMemberAccess()) {
1108 if (Left.isOneOf(tok::r_paren, tok::r_square))
Daniel Jasper518ee342013-02-26 13:59:14 +00001109 return 20; // Should be smaller than breaking at a nested comma.
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001110 return 150;
1111 }
1112
Daniel Jasper20a0f8c2013-07-11 21:02:56 +00001113 // Breaking before a trailing 'const' or not-function-like annotation is bad.
Daniel Jasperaa9e7b12013-08-01 13:46:58 +00001114 if (Left.is(tok::r_paren) && Line.Type != LT_ObjCProperty &&
Daniel Jasper20a0f8c2013-07-11 21:02:56 +00001115 (Right.is(tok::kw_const) || (Right.is(tok::identifier) && Right.Next &&
1116 Right.Next->isNot(tok::l_paren))))
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001117 return 150;
1118
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001119 // In for-loops, prefer breaking at ',' and ';'.
Manuel Klimekb3987012013-05-29 14:47:47 +00001120 if (Line.First->is(tok::kw_for) && Left.is(tok::equal))
Daniel Jasper7d812812013-02-21 15:00:29 +00001121 return 4;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001122
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001123 // In Objective-C method expressions, prefer breaking before "param:" over
1124 // breaking after it.
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001125 if (Right.Type == TT_ObjCSelectorName)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001126 return 0;
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001127 if (Left.is(tok::colon) && Left.Type == TT_ObjCMethodExpr)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001128 return 20;
1129
Daniel Jasper1407bee2013-04-11 14:29:13 +00001130 if (Left.is(tok::l_paren) && Line.MightBeFunctionDecl)
1131 return 100;
Daniel Jasperac3223e2013-04-10 09:49:49 +00001132 if (Left.opensScope())
Daniel Jaspere60084d2013-08-13 06:50:04 +00001133 return Left.ParameterCount > 1 ? prec::Comma : 19;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001134
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001135 if (Right.is(tok::lessless)) {
1136 if (Left.is(tok::string_literal)) {
Alexander Kornienko00895102013-06-05 14:09:10 +00001137 StringRef Content = Left.TokenText;
Daniel Jasperbfa1edd2013-03-14 14:00:17 +00001138 Content = Content.drop_back(1).drop_front(1).trim();
1139 if (Content.size() > 1 &&
1140 (Content.back() == ':' || Content.back() == '='))
Daniel Jasper9637dda2013-07-15 14:33:14 +00001141 return 25;
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001142 }
Daniel Jasper0c368782013-07-15 15:04:42 +00001143 return 1; // Breaking at a << is really cheap.
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001144 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001145 if (Left.Type == TT_ConditionalExpr)
Daniel Jasper518ee342013-02-26 13:59:14 +00001146 return prec::Conditional;
Manuel Klimekb3987012013-05-29 14:47:47 +00001147 prec::Level Level = Left.getPrecedence();
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001148
1149 if (Level != prec::Unknown)
1150 return Level;
Daniel Jasper24849712013-03-01 16:48:32 +00001151
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001152 return 3;
1153}
1154
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001155bool TokenAnnotator::spaceRequiredBetween(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001156 const FormatToken &Left,
1157 const FormatToken &Right) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001158 if (Right.is(tok::hashhash))
1159 return Left.is(tok::hash);
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001160 if (Left.isOneOf(tok::hashhash, tok::hash))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001161 return Right.is(tok::hash);
Daniel Jasper7df56bf2013-08-20 12:36:34 +00001162 if (Left.is(tok::l_paren) && Right.is(tok::r_paren))
1163 return Style.SpaceInEmptyParentheses;
1164 if (Left.is(tok::l_paren) || Right.is(tok::r_paren))
Daniel Jasper34f3d052013-08-21 08:39:01 +00001165 return (Right.Type == TT_CastRParen ||
1166 (Left.MatchingParen && Left.MatchingParen->Type == TT_CastRParen))
Daniel Jasper7df56bf2013-08-20 12:36:34 +00001167 ? Style.SpacesInCStyleCastParentheses
1168 : Style.SpacesInParentheses;
1169 if (Right.isOneOf(tok::semi, tok::comma))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001170 return false;
1171 if (Right.is(tok::less) &&
1172 (Left.is(tok::kw_template) ||
1173 (Line.Type == LT_ObjCDecl && Style.ObjCSpaceBeforeProtocolList)))
1174 return true;
1175 if (Left.is(tok::arrow) || Right.is(tok::arrow))
1176 return false;
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001177 if (Left.isOneOf(tok::exclaim, tok::tilde))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001178 return false;
1179 if (Left.is(tok::at) &&
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001180 Right.isOneOf(tok::identifier, tok::string_literal, tok::char_constant,
1181 tok::numeric_constant, tok::l_paren, tok::l_brace,
1182 tok::kw_true, tok::kw_false))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001183 return false;
1184 if (Left.is(tok::coloncolon))
1185 return false;
1186 if (Right.is(tok::coloncolon))
Daniel Jasper081fc172013-08-28 07:07:07 +00001187 return !Left.isOneOf(tok::identifier, tok::greater, tok::l_paren,
1188 tok::r_paren);
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001189 if (Left.is(tok::less) || Right.isOneOf(tok::greater, tok::less))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001190 return false;
Daniel Jasperc47d7f12013-07-01 09:47:25 +00001191 if (Right.is(tok::ellipsis))
1192 return false;
Manuel Klimek31e44f72013-09-04 08:20:47 +00001193 if (Left.is(tok::l_square) && Right.is(tok::amp))
1194 return false;
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001195 if (Right.Type == TT_PointerOrReference)
Manuel Klimekb3987012013-05-29 14:47:47 +00001196 return Left.Tok.isLiteral() ||
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001197 ((Left.Type != TT_PointerOrReference) && Left.isNot(tok::l_paren) &&
1198 !Style.PointerBindsToType);
Daniel Jasper3ff4a2f2013-05-28 15:27:10 +00001199 if (Right.Type == TT_FunctionTypeLParen && Left.isNot(tok::l_paren) &&
Daniel Jasper395228f2013-05-08 14:58:20 +00001200 (Left.Type != TT_PointerOrReference || Style.PointerBindsToType))
1201 return true;
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001202 if (Left.Type == TT_PointerOrReference)
Daniel Jasper3a1847e2013-07-01 09:34:09 +00001203 return Right.Tok.isLiteral() || Right.Type == TT_BlockComment ||
Daniel Jasper9322aae2013-03-20 09:53:18 +00001204 ((Right.Type != TT_PointerOrReference) &&
Daniel Jasper81d2d382013-04-01 17:13:26 +00001205 Right.isNot(tok::l_paren) && Style.PointerBindsToType &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001206 Left.Previous &&
1207 !Left.Previous->isOneOf(tok::l_paren, tok::coloncolon));
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001208 if (Right.is(tok::star) && Left.is(tok::l_paren))
1209 return false;
Nico Weber051860e2013-02-10 02:08:05 +00001210 if (Left.is(tok::l_square))
1211 return Left.Type == TT_ObjCArrayLiteral && Right.isNot(tok::r_square);
1212 if (Right.is(tok::r_square))
1213 return Right.Type == TT_ObjCArrayLiteral;
Daniel Jasperec172262013-08-30 10:36:58 +00001214 if (Right.is(tok::l_square) && Right.Type != TT_ObjCMethodExpr &&
1215 Left.isNot(tok::numeric_constant))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001216 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001217 if (Left.is(tok::colon))
1218 return Left.Type != TT_ObjCMethodExpr;
1219 if (Right.is(tok::colon))
Daniel Jasperab3ce592013-08-01 22:05:00 +00001220 return Right.Type != TT_ObjCMethodExpr && !Left.is(tok::question);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001221 if (Right.is(tok::l_paren)) {
Daniel Jaspere0fa4c52013-07-17 20:25:02 +00001222 if (Left.is(tok::r_paren) && Left.MatchingParen &&
1223 Left.MatchingParen->Previous &&
1224 Left.MatchingParen->Previous->is(tok::kw___attribute))
1225 return true;
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001226 return Line.Type == LT_ObjCDecl ||
Daniel Jasper34f3d052013-08-21 08:39:01 +00001227 Left.isOneOf(tok::kw_return, tok::kw_new, tok::kw_delete,
1228 tok::semi) ||
1229 (Style.SpaceAfterControlStatementKeyword &&
1230 Left.isOneOf(tok::kw_if, tok::kw_for, tok::kw_while, tok::kw_switch,
1231 tok::kw_catch));
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001232 }
Manuel Klimekb3987012013-05-29 14:47:47 +00001233 if (Left.is(tok::at) && Right.Tok.getObjCKeywordID() != tok::objc_not_keyword)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001234 return false;
1235 if (Left.is(tok::l_brace) && Right.is(tok::r_brace))
Daniel Jasper2424eef2013-05-23 10:15:45 +00001236 return false; // No spaces in "{}".
1237 if (Left.is(tok::l_brace) || Right.is(tok::r_brace))
Daniel Jasperb5dc3f42013-07-16 18:22:10 +00001238 return !Style.Cpp11BracedListStyle;
Daniel Jasper1bee0732013-05-23 18:05:18 +00001239 if (Right.Type == TT_UnaryOperator)
1240 return !Left.isOneOf(tok::l_paren, tok::l_square, tok::at) &&
1241 (Left.isNot(tok::colon) || Left.Type != TT_ObjCMethodExpr);
Daniel Jasperce933562013-05-23 21:35:49 +00001242 if (Left.isOneOf(tok::identifier, tok::greater, tok::r_square) &&
Manuel Klimek31e44f72013-09-04 08:20:47 +00001243 Right.is(tok::l_brace) && Right.getNextNonComment() &&
1244 Right.BlockKind != BK_Block)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001245 return false;
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001246 if (Left.is(tok::period) || Right.is(tok::period))
1247 return false;
Nico Weber861576b2013-06-26 00:15:19 +00001248 if (Left.Type == TT_BlockComment && Left.TokenText.endswith("=*/"))
1249 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001250 return true;
1251}
1252
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001253bool TokenAnnotator::spaceRequiredBefore(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001254 const FormatToken &Tok) {
1255 if (Tok.Tok.getIdentifierInfo() && Tok.Previous->Tok.getIdentifierInfo())
Daniel Jasper2b4c9242013-02-11 08:01:18 +00001256 return true; // Never ever merge two identifiers.
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001257 if (Line.Type == LT_ObjCMethodDecl) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001258 if (Tok.Previous->Type == TT_ObjCMethodSpecifier)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001259 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001260 if (Tok.Previous->is(tok::r_paren) && Tok.is(tok::identifier))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001261 // Don't space between ')' and <id>
1262 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001263 }
1264 if (Line.Type == LT_ObjCProperty &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001265 (Tok.is(tok::equal) || Tok.Previous->is(tok::equal)))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001266 return false;
1267
Daniel Jasper2ca37412013-07-09 14:36:48 +00001268 if (Tok.Type == TT_TrailingReturnArrow ||
1269 Tok.Previous->Type == TT_TrailingReturnArrow)
1270 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001271 if (Tok.Previous->is(tok::comma))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001272 return true;
Daniel Jasper9c3c7b32013-02-28 13:40:17 +00001273 if (Tok.is(tok::comma))
1274 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001275 if (Tok.Type == TT_CtorInitializerColon || Tok.Type == TT_ObjCBlockLParen)
1276 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001277 if (Tok.Previous->Tok.is(tok::kw_operator))
Daniel Jasper2b4c9242013-02-11 08:01:18 +00001278 return false;
1279 if (Tok.Type == TT_OverloadedOperatorLParen)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001280 return false;
1281 if (Tok.is(tok::colon))
Manuel Klimekb3987012013-05-29 14:47:47 +00001282 return !Line.First->isOneOf(tok::kw_case, tok::kw_default) &&
Daniel Jasperab3ce592013-08-01 22:05:00 +00001283 Tok.getNextNonComment() != NULL && Tok.Type != TT_ObjCMethodExpr &&
1284 !Tok.Previous->is(tok::question);
Manuel Klimekb3987012013-05-29 14:47:47 +00001285 if (Tok.Previous->Type == TT_UnaryOperator ||
1286 Tok.Previous->Type == TT_CastRParen)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001287 return false;
Manuel Klimekb3987012013-05-29 14:47:47 +00001288 if (Tok.Previous->is(tok::greater) && Tok.is(tok::greater)) {
Daniel Jasper29f123b2013-02-08 15:28:42 +00001289 return Tok.Type == TT_TemplateCloser &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001290 Tok.Previous->Type == TT_TemplateCloser &&
Daniel Jasper29f123b2013-02-08 15:28:42 +00001291 Style.Standard != FormatStyle::LS_Cpp11;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001292 }
Alexander Kornienko54a38bd2013-03-20 16:41:56 +00001293 if (Tok.isOneOf(tok::arrowstar, tok::periodstar) ||
Manuel Klimekb3987012013-05-29 14:47:47 +00001294 Tok.Previous->isOneOf(tok::arrowstar, tok::periodstar))
Daniel Jasper9c3c7b32013-02-28 13:40:17 +00001295 return false;
Daniel Jasper1dc6f742013-08-07 16:29:23 +00001296 if ((Tok.Type == TT_BinaryOperator && !Tok.Previous->is(tok::l_paren)) ||
1297 Tok.Previous->Type == TT_BinaryOperator)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001298 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001299 if (Tok.Previous->Type == TT_TemplateCloser && Tok.is(tok::l_paren))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001300 return false;
Daniel Jaspera4dd9822013-08-28 08:24:04 +00001301 if (Tok.is(tok::less) && Tok.Previous->isNot(tok::l_paren) &&
1302 Line.First->is(tok::hash))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001303 return true;
1304 if (Tok.Type == TT_TrailingUnaryOperator)
1305 return false;
Manuel Klimekb3987012013-05-29 14:47:47 +00001306 return spaceRequiredBetween(Line, *Tok.Previous, Tok);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001307}
1308
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001309bool TokenAnnotator::canBreakBefore(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001310 const FormatToken &Right) {
1311 const FormatToken &Left = *Right.Previous;
Daniel Jasper6561f6a2013-07-09 07:43:55 +00001312 if (Right.Type == TT_StartOfName || Right.is(tok::kw_operator))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001313 return true;
Nico Weberf2ff8122013-05-26 05:39:26 +00001314 if (Right.is(tok::colon) &&
1315 (Right.Type == TT_ObjCDictLiteral || Right.Type == TT_ObjCMethodExpr))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001316 return false;
Nico Weberf2ff8122013-05-26 05:39:26 +00001317 if (Left.is(tok::colon) &&
1318 (Left.Type == TT_ObjCDictLiteral || Left.Type == TT_ObjCMethodExpr))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001319 return true;
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001320 if (Right.Type == TT_ObjCSelectorName)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001321 return true;
Daniel Jasperaa9e7b12013-08-01 13:46:58 +00001322 if (Left.is(tok::r_paren) && Line.Type == LT_ObjCProperty)
1323 return true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001324 if (Left.ClosesTemplateDeclaration)
1325 return true;
Daniel Jasperab3ce592013-08-01 22:05:00 +00001326 if ((Right.Type == TT_ConditionalExpr &&
1327 !(Right.is(tok::colon) && Left.is(tok::question))) ||
1328 Right.is(tok::question))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001329 return true;
Daniel Jasper6cabab42013-02-14 08:42:54 +00001330 if (Right.Type == TT_RangeBasedForLoopColon ||
Daniel Jasperc476ea92013-08-28 07:27:35 +00001331 Right.Type == TT_OverloadedOperatorLParen ||
1332 Right.Type == TT_OverloadedOperator)
Daniel Jasper6cabab42013-02-14 08:42:54 +00001333 return false;
Daniel Jasperc194c952013-05-06 06:45:09 +00001334 if (Left.Type == TT_RangeBasedForLoopColon)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001335 return true;
Daniel Jasper7d812812013-02-21 15:00:29 +00001336 if (Right.Type == TT_RangeBasedForLoopColon)
1337 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001338 if (Left.Type == TT_PointerOrReference || Left.Type == TT_TemplateCloser ||
1339 Left.Type == TT_UnaryOperator || Left.Type == TT_ConditionalExpr ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001340 Left.isOneOf(tok::question, tok::kw_operator))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001341 return false;
1342 if (Left.is(tok::equal) && Line.Type == LT_VirtualFunctionDecl)
1343 return false;
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001344 if (Left.Previous) {
1345 if (Left.is(tok::l_paren) && Right.is(tok::l_paren) &&
1346 Left.Previous->is(tok::kw___attribute))
1347 return false;
Daniel Jasper2ca37412013-07-09 14:36:48 +00001348 if (Left.is(tok::l_paren) && (Left.Previous->Type == TT_BinaryOperator ||
Daniel Jasper5e2169f2013-07-18 14:46:07 +00001349 Left.Previous->Type == TT_CastRParen))
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001350 return false;
1351 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001352
Daniel Jasper65d2c382013-06-06 16:08:57 +00001353 if (Right.isTrailingComment())
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001354 // We rely on MustBreakBefore being set correctly here as we should not
1355 // change the "binding" behavior of a comment.
1356 return false;
1357
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001358 // We only break before r_brace if there was a corresponding break before
1359 // the l_brace, which is tracked by BreakBeforeClosingBrace.
Daniel Jaspere8b10d32013-07-26 16:56:36 +00001360 if (Right.isOneOf(tok::r_brace, tok::r_paren) ||
1361 Right.Type == TT_TemplateCloser)
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001362 return false;
1363
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001364 // Allow breaking after a trailing 'const', e.g. after a method declaration,
1365 // unless it is follow by ';', '{' or '='.
Manuel Klimekb3987012013-05-29 14:47:47 +00001366 if (Left.is(tok::kw_const) && Left.Previous != NULL &&
1367 Left.Previous->is(tok::r_paren))
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001368 return !Right.isOneOf(tok::l_brace, tok::semi, tok::equal);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001369
Daniel Jasper8ef19a22013-03-14 09:50:46 +00001370 if (Right.is(tok::kw___attribute))
1371 return true;
1372
Daniel Jasper3a204412013-02-23 07:46:38 +00001373 if (Left.is(tok::identifier) && Right.is(tok::string_literal))
1374 return true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +00001375
1376 if (Left.Type == TT_CtorInitializerComma &&
1377 Style.BreakConstructorInitializersBeforeComma)
1378 return false;
1379 if (Right.isBinaryOperator() && Style.BreakBeforeBinaryOperators)
1380 return true;
1381 return (Left.isBinaryOperator() && Left.isNot(tok::lessless) &&
1382 !Style.BreakBeforeBinaryOperators) ||
Daniel Jasper6b119d62013-04-05 17:22:09 +00001383 Left.isOneOf(tok::comma, tok::coloncolon, tok::semi, tok::l_brace,
1384 tok::kw_class, tok::kw_struct) ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001385 Right.isOneOf(tok::lessless, tok::arrow, tok::period, tok::colon) ||
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001386 (Left.is(tok::r_paren) &&
Daniel Jaspere033e872013-05-21 09:16:31 +00001387 Right.isOneOf(tok::identifier, tok::kw_const, tok::kw___attribute)) ||
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001388 (Left.is(tok::l_paren) && !Right.is(tok::r_paren)) ||
Daniel Jasper011c35d2013-07-12 11:19:37 +00001389 Right.is(tok::l_square);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001390}
1391
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001392void TokenAnnotator::printDebugInfo(const AnnotatedLine &Line) {
1393 llvm::errs() << "AnnotatedTokens:\n";
Manuel Klimekb3987012013-05-29 14:47:47 +00001394 const FormatToken *Tok = Line.First;
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001395 while (Tok) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001396 llvm::errs() << " M=" << Tok->MustBreakBefore
Daniel Jasperc7bd68f2013-07-10 14:02:49 +00001397 << " C=" << Tok->CanBreakBefore << " T=" << Tok->Type
1398 << " S=" << Tok->SpacesRequiredBefore
1399 << " P=" << Tok->SplitPenalty << " Name=" << Tok->Tok.getName()
1400 << " PPK=" << Tok->PackingKind << " FakeLParens=";
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001401 for (unsigned i = 0, e = Tok->FakeLParens.size(); i != e; ++i)
1402 llvm::errs() << Tok->FakeLParens[i] << "/";
1403 llvm::errs() << " FakeRParens=" << Tok->FakeRParens << "\n";
Manuel Klimekb3987012013-05-29 14:47:47 +00001404 Tok = Tok->Next;
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001405 }
1406 llvm::errs() << "----\n";
1407}
1408
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001409} // namespace format
1410} // namespace clang