blob: e3984428bd3c94e60b5aa48095e45b9108f26143 [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,
Daniel Jasper363193b2013-10-20 18:15:30 +000099 tok::kw_if, tok::kw_while)) {
100 // static_assert, if and while usually contain expressions.
Daniel Jasperb7000ca2013-08-01 17:58:23 +0000101 Contexts.back().IsExpression = true;
Daniel Jasper363193b2013-10-20 18:15:30 +0000102 } else if (Left->Previous && Left->Previous->is(tok::r_square) &&
103 Left->Previous->MatchingParen &&
104 Left->Previous->MatchingParen->Type == TT_LambdaLSquare) {
105 // This is a parameter list of a lambda expression.
106 Contexts.back().IsExpression = false;
107 }
Daniel Jasperb7000ca2013-08-01 17:58:23 +0000108
Daniel Jasper4e778092013-02-06 10:05:46 +0000109 if (StartsObjCMethodExpr) {
110 Contexts.back().ColonIsObjCMethodExpr = true;
111 Left->Type = TT_ObjCMethodExpr;
112 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000113
Daniel Jasper431f5912013-05-28 08:33:00 +0000114 bool MightBeFunctionType = CurrentToken->is(tok::star);
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000115 bool HasMultipleLines = false;
116 bool HasMultipleParametersOnALine = false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000117 while (CurrentToken != NULL) {
118 // LookForDecls is set when "if (" has been seen. Check for
119 // 'identifier' '*' 'identifier' followed by not '=' -- this
120 // '*' has to be a binary operator but determineStarAmpUsage() will
121 // categorize it as an unary operator, so set the right type here.
Manuel Klimekb3987012013-05-29 14:47:47 +0000122 if (LookForDecls && CurrentToken->Next) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000123 FormatToken *Prev = CurrentToken->getPreviousNonComment();
Alexander Kornienko2785b9a2013-06-07 16:02:52 +0000124 if (Prev) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000125 FormatToken *PrevPrev = Prev->getPreviousNonComment();
Alexander Kornienko2785b9a2013-06-07 16:02:52 +0000126 FormatToken *Next = CurrentToken->Next;
127 if (PrevPrev && PrevPrev->is(tok::identifier) &&
128 Prev->isOneOf(tok::star, tok::amp, tok::ampamp) &&
129 CurrentToken->is(tok::identifier) && Next->isNot(tok::equal)) {
130 Prev->Type = TT_BinaryOperator;
131 LookForDecls = false;
132 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000133 }
134 }
135
Daniel Jasper53352602013-08-12 12:16:34 +0000136 if (CurrentToken->Previous->Type == TT_PointerOrReference &&
137 CurrentToken->Previous->Previous->isOneOf(tok::l_paren,
138 tok::coloncolon))
139 MightBeFunctionType = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000140 if (CurrentToken->is(tok::r_paren)) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000141 if (MightBeFunctionType && CurrentToken->Next &&
Daniel Jaspere7d3bff2013-07-16 11:37:21 +0000142 (CurrentToken->Next->is(tok::l_paren) ||
143 (CurrentToken->Next->is(tok::l_square) &&
144 !Contexts.back().IsExpression)))
Daniel Jasper431f5912013-05-28 08:33:00 +0000145 Left->Type = TT_FunctionTypeLParen;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000146 Left->MatchingParen = CurrentToken;
147 CurrentToken->MatchingParen = Left;
148
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000149 if (StartsObjCMethodExpr) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000150 CurrentToken->Type = TT_ObjCMethodExpr;
151 if (Contexts.back().FirstObjCSelectorName != NULL) {
152 Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
153 Contexts.back().LongestObjCSelectorName;
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000154 }
155 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000156
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000157 if (!HasMultipleLines)
158 Left->PackingKind = PPK_Inconclusive;
159 else if (HasMultipleParametersOnALine)
160 Left->PackingKind = PPK_BinPacked;
161 else
162 Left->PackingKind = PPK_OnePerLine;
163
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000164 next();
165 return true;
166 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000167 if (CurrentToken->isOneOf(tok::r_square, tok::r_brace))
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000168 return false;
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000169 updateParameterCount(Left, CurrentToken);
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000170 if (CurrentToken->is(tok::comma) && CurrentToken->Next &&
171 !CurrentToken->Next->HasUnescapedNewline &&
172 !CurrentToken->Next->isTrailingComment())
173 HasMultipleParametersOnALine = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000174 if (!consumeToken())
175 return false;
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000176 if (CurrentToken && CurrentToken->HasUnescapedNewline)
177 HasMultipleLines = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000178 }
179 return false;
180 }
181
182 bool parseSquare() {
183 if (!CurrentToken)
184 return false;
185
Alexander Kornienkod71b15b2013-06-17 13:19:53 +0000186 // A '[' could be an index subscript (after an identifier or after
Nico Weber051860e2013-02-10 02:08:05 +0000187 // ')' or ']'), it could be the start of an Objective-C method
188 // expression, or it could the the start of an Objective-C array literal.
Manuel Klimekb3987012013-05-29 14:47:47 +0000189 FormatToken *Left = CurrentToken->Previous;
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000190 FormatToken *Parent = Left->getPreviousNonComment();
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000191 bool StartsObjCMethodExpr =
Daniel Jasper567dcf92013-09-05 09:29:45 +0000192 Contexts.back().CanBeExpression && Left->Type != TT_LambdaLSquare &&
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000193 (!Parent || Parent->isOneOf(tok::colon, tok::l_square, tok::l_paren,
194 tok::kw_return, tok::kw_throw) ||
Daniel Jasperac3223e2013-04-10 09:49:49 +0000195 Parent->isUnaryOperator() || Parent->Type == TT_ObjCForIn ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000196 Parent->Type == TT_CastRParen ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000197 getBinOpPrecedence(Parent->Tok.getKind(), true, true) > prec::Unknown);
Daniel Jasper923ebef2013-03-14 13:45:21 +0000198 ScopedContextCreator ContextCreator(*this, tok::l_square, 10);
Daniel Jasper6f21a982013-03-13 07:49:51 +0000199 Contexts.back().IsExpression = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000200
Daniel Jasper4e778092013-02-06 10:05:46 +0000201 if (StartsObjCMethodExpr) {
202 Contexts.back().ColonIsObjCMethodExpr = true;
203 Left->Type = TT_ObjCMethodExpr;
Daniel Jaspera07aa662013-10-22 15:30:28 +0000204 } else if (Parent && Parent->is(tok::at)) {
205 Left->Type = TT_ArrayInitializerLSquare;
206 } else if (Left->Type == TT_Unknown) {
207 Left->Type = TT_ArraySubscriptLSquare;
Daniel Jasper4e778092013-02-06 10:05:46 +0000208 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000209
210 while (CurrentToken != NULL) {
211 if (CurrentToken->is(tok::r_square)) {
Daniel Jasperac2c9742013-09-05 10:04:31 +0000212 if (CurrentToken->Next && CurrentToken->Next->is(tok::l_paren) &&
213 Left->Type == TT_ObjCMethodExpr) {
Nico Webere8a97982013-02-06 06:20:11 +0000214 // An ObjC method call is rarely followed by an open parenthesis.
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000215 // FIXME: Do we incorrectly label ":" with this?
216 StartsObjCMethodExpr = false;
217 Left->Type = TT_Unknown;
218 }
Daniel Jasper01786732013-02-04 07:21:18 +0000219 if (StartsObjCMethodExpr) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000220 CurrentToken->Type = TT_ObjCMethodExpr;
Nico Webere8a97982013-02-06 06:20:11 +0000221 // determineStarAmpUsage() thinks that '*' '[' is allocating an
222 // array of pointers, but if '[' starts a selector then '*' is a
223 // binary operator.
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +0000224 if (Parent != NULL && Parent->Type == TT_PointerOrReference)
Nico Weber4ed7f3e2013-02-06 16:54:35 +0000225 Parent->Type = TT_BinaryOperator;
Daniel Jasper01786732013-02-04 07:21:18 +0000226 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000227 Left->MatchingParen = CurrentToken;
228 CurrentToken->MatchingParen = Left;
Daniel Jasper4e778092013-02-06 10:05:46 +0000229 if (Contexts.back().FirstObjCSelectorName != NULL)
230 Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
231 Contexts.back().LongestObjCSelectorName;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000232 next();
233 return true;
234 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000235 if (CurrentToken->isOneOf(tok::r_paren, tok::r_brace))
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000236 return false;
Daniel Jaspera07aa662013-10-22 15:30:28 +0000237 if (CurrentToken->is(tok::comma) &&
238 Left->Type == TT_ArraySubscriptLSquare)
239 Left->Type = TT_ArrayInitializerLSquare;
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000240 updateParameterCount(Left, CurrentToken);
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000241 if (!consumeToken())
242 return false;
243 }
244 return false;
245 }
246
247 bool parseBrace() {
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000248 if (CurrentToken != NULL) {
249 ScopedContextCreator ContextCreator(*this, tok::l_brace, 1);
Manuel Klimekb3987012013-05-29 14:47:47 +0000250 FormatToken *Left = CurrentToken->Previous;
Nico Weberf2ff8122013-05-26 05:39:26 +0000251
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000252 FormatToken *Parent = Left->getPreviousNonComment();
Nico Weberf2ff8122013-05-26 05:39:26 +0000253 bool StartsObjCDictLiteral = Parent && Parent->is(tok::at);
254 if (StartsObjCDictLiteral) {
255 Contexts.back().ColonIsObjCDictLiteral = true;
256 Left->Type = TT_ObjCDictLiteral;
257 }
258
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000259 while (CurrentToken != NULL) {
260 if (CurrentToken->is(tok::r_brace)) {
Nico Weberf2ff8122013-05-26 05:39:26 +0000261 if (StartsObjCDictLiteral)
262 CurrentToken->Type = TT_ObjCDictLiteral;
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000263 Left->MatchingParen = CurrentToken;
264 CurrentToken->MatchingParen = Left;
265 next();
266 return true;
267 }
268 if (CurrentToken->isOneOf(tok::r_paren, tok::r_square))
269 return false;
270 updateParameterCount(Left, CurrentToken);
271 if (!consumeToken())
272 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000273 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000274 }
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000275 // No closing "}" found, this probably starts a definition.
276 Line.StartsDefinition = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000277 return true;
278 }
Daniel Jasperc4615b72013-02-20 12:56:39 +0000279
Manuel Klimekb3987012013-05-29 14:47:47 +0000280 void updateParameterCount(FormatToken *Left, FormatToken *Current) {
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000281 if (Current->is(tok::comma)) {
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000282 ++Left->ParameterCount;
Daniel Jasperd4a03db2013-08-22 15:00:41 +0000283 if (!Left->Role)
284 Left->Role.reset(new CommaSeparatedList(Style));
285 Left->Role->CommaFound(Current);
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000286 } else if (Left->ParameterCount == 0 && Current->isNot(tok::comment)) {
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000287 Left->ParameterCount = 1;
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000288 }
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000289 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000290
291 bool parseConditional() {
292 while (CurrentToken != NULL) {
293 if (CurrentToken->is(tok::colon)) {
294 CurrentToken->Type = TT_ConditionalExpr;
295 next();
296 return true;
297 }
298 if (!consumeToken())
299 return false;
300 }
301 return false;
302 }
303
304 bool parseTemplateDeclaration() {
305 if (CurrentToken != NULL && CurrentToken->is(tok::less)) {
306 CurrentToken->Type = TT_TemplateOpener;
307 next();
308 if (!parseAngle())
309 return false;
Daniel Jasper34511fb2013-02-19 17:14:38 +0000310 if (CurrentToken != NULL)
Manuel Klimekb3987012013-05-29 14:47:47 +0000311 CurrentToken->Previous->ClosesTemplateDeclaration = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000312 return true;
313 }
314 return false;
315 }
316
317 bool consumeToken() {
Manuel Klimekb3987012013-05-29 14:47:47 +0000318 FormatToken *Tok = CurrentToken;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000319 next();
Manuel Klimekb3987012013-05-29 14:47:47 +0000320 switch (Tok->Tok.getKind()) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000321 case tok::plus:
322 case tok::minus:
Manuel Klimekb3987012013-05-29 14:47:47 +0000323 if (Tok->Previous == NULL && Line.MustBeDeclaration)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000324 Tok->Type = TT_ObjCMethodSpecifier;
325 break;
326 case tok::colon:
Manuel Klimekb3987012013-05-29 14:47:47 +0000327 if (Tok->Previous == NULL)
Daniel Jaspercf6d76a2013-03-18 12:50:26 +0000328 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000329 // Colons from ?: are handled in parseConditional().
Manuel Klimekb3987012013-05-29 14:47:47 +0000330 if (Tok->Previous->is(tok::r_paren) && Contexts.size() == 1) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000331 Tok->Type = TT_CtorInitializerColon;
Nico Weberf2ff8122013-05-26 05:39:26 +0000332 } else if (Contexts.back().ColonIsObjCDictLiteral) {
333 Tok->Type = TT_ObjCDictLiteral;
Daniel Jasper4e778092013-02-06 10:05:46 +0000334 } else if (Contexts.back().ColonIsObjCMethodExpr ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000335 Line.First->Type == TT_ObjCMethodSpecifier) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000336 Tok->Type = TT_ObjCMethodExpr;
Manuel Klimekb3987012013-05-29 14:47:47 +0000337 Tok->Previous->Type = TT_ObjCSelectorName;
Alexander Kornienko83a7dcd2013-09-10 09:38:25 +0000338 if (Tok->Previous->ColumnWidth >
Alexander Kornienko00895102013-06-05 14:09:10 +0000339 Contexts.back().LongestObjCSelectorName) {
Alexander Kornienko01fe9f92013-10-10 13:36:20 +0000340 Contexts.back().LongestObjCSelectorName = Tok->Previous->ColumnWidth;
Alexander Kornienko00895102013-06-05 14:09:10 +0000341 }
Daniel Jasper4e778092013-02-06 10:05:46 +0000342 if (Contexts.back().FirstObjCSelectorName == NULL)
Manuel Klimekb3987012013-05-29 14:47:47 +0000343 Contexts.back().FirstObjCSelectorName = Tok->Previous;
Daniel Jasper4e778092013-02-06 10:05:46 +0000344 } else if (Contexts.back().ColonIsForRangeExpr) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000345 Tok->Type = TT_RangeBasedForLoopColon;
Alexander Kornienko01fe9f92013-10-10 13:36:20 +0000346 } else if (CurrentToken != NULL &&
347 CurrentToken->is(tok::numeric_constant)) {
348 Tok->Type = TT_BitFieldColon;
Daniel Jaspercea014b2013-10-08 16:24:07 +0000349 } else if (Contexts.size() == 1 && Line.First->isNot(tok::kw_enum)) {
Daniel Jasper6cabab42013-02-14 08:42:54 +0000350 Tok->Type = TT_InheritanceColon;
Daniel Jasper923ebef2013-03-14 13:45:21 +0000351 } else if (Contexts.back().ContextKind == tok::l_paren) {
352 Tok->Type = TT_InlineASMColon;
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000353 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000354 break;
355 case tok::kw_if:
356 case tok::kw_while:
357 if (CurrentToken != NULL && CurrentToken->is(tok::l_paren)) {
358 next();
Nico Weber27268772013-06-26 00:30:14 +0000359 if (!parseParens(/*LookForDecls=*/true))
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000360 return false;
361 }
362 break;
363 case tok::kw_for:
Daniel Jasper4e778092013-02-06 10:05:46 +0000364 Contexts.back().ColonIsForRangeExpr = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000365 next();
366 if (!parseParens())
367 return false;
368 break;
369 case tok::l_paren:
370 if (!parseParens())
371 return false;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000372 if (Line.MustBeDeclaration && NameFound && !Contexts.back().IsExpression)
Daniel Jasper3c08a812013-02-24 18:54:32 +0000373 Line.MightBeFunctionDecl = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000374 break;
375 case tok::l_square:
376 if (!parseSquare())
377 return false;
378 break;
379 case tok::l_brace:
380 if (!parseBrace())
381 return false;
382 break;
383 case tok::less:
Daniel Jasper0236dd02013-07-30 22:37:19 +0000384 if (Tok->Previous && !Tok->Previous->Tok.isLiteral() && parseAngle())
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000385 Tok->Type = TT_TemplateOpener;
386 else {
387 Tok->Type = TT_BinaryOperator;
388 CurrentToken = Tok;
389 next();
390 }
391 break;
392 case tok::r_paren:
393 case tok::r_square:
394 return false;
395 case tok::r_brace:
396 // Lines can start with '}'.
Manuel Klimekb3987012013-05-29 14:47:47 +0000397 if (Tok->Previous != NULL)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000398 return false;
399 break;
400 case tok::greater:
401 Tok->Type = TT_BinaryOperator;
402 break;
403 case tok::kw_operator:
Daniel Jasper174f60f2013-09-02 09:20:39 +0000404 while (CurrentToken &&
405 !CurrentToken->isOneOf(tok::l_paren, tok::semi, tok::r_paren)) {
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000406 if (CurrentToken->isOneOf(tok::star, tok::amp))
Daniel Jasper2b4c9242013-02-11 08:01:18 +0000407 CurrentToken->Type = TT_PointerOrReference;
408 consumeToken();
Daniel Jasper174f60f2013-09-02 09:20:39 +0000409 if (CurrentToken && CurrentToken->Previous->Type == TT_BinaryOperator)
Daniel Jasperc476ea92013-08-28 07:27:35 +0000410 CurrentToken->Previous->Type = TT_OverloadedOperator;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000411 }
Daniel Jasper6ea933c2013-05-10 07:59:58 +0000412 if (CurrentToken) {
Daniel Jasper2b4c9242013-02-11 08:01:18 +0000413 CurrentToken->Type = TT_OverloadedOperatorLParen;
Manuel Klimekb3987012013-05-29 14:47:47 +0000414 if (CurrentToken->Previous->Type == TT_BinaryOperator)
415 CurrentToken->Previous->Type = TT_OverloadedOperator;
Daniel Jasper6ea933c2013-05-10 07:59:58 +0000416 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000417 break;
418 case tok::question:
419 parseConditional();
420 break;
421 case tok::kw_template:
422 parseTemplateDeclaration();
423 break;
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000424 case tok::identifier:
Manuel Klimekb3987012013-05-29 14:47:47 +0000425 if (Line.First->is(tok::kw_for) &&
426 Tok->Tok.getIdentifierInfo() == &Ident_in)
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000427 Tok->Type = TT_ObjCForIn;
428 break;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000429 case tok::comma:
430 if (Contexts.back().FirstStartOfName)
431 Contexts.back().FirstStartOfName->PartOfMultiVariableDeclStmt = true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000432 if (Contexts.back().InCtorInitializer)
433 Tok->Type = TT_CtorInitializerComma;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000434 break;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000435 default:
436 break;
437 }
438 return true;
439 }
440
441 void parseIncludeDirective() {
442 next();
443 if (CurrentToken != NULL && CurrentToken->is(tok::less)) {
444 next();
445 while (CurrentToken != NULL) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000446 if (CurrentToken->isNot(tok::comment) || CurrentToken->Next)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000447 CurrentToken->Type = TT_ImplicitStringLiteral;
448 next();
449 }
450 } else {
451 while (CurrentToken != NULL) {
Daniel Jasper3a204412013-02-23 07:46:38 +0000452 if (CurrentToken->is(tok::string_literal))
453 // Mark these string literals as "implicit" literals, too, so that
454 // they are not split or line-wrapped.
455 CurrentToken->Type = TT_ImplicitStringLiteral;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000456 next();
457 }
458 }
459 }
460
461 void parseWarningOrError() {
462 next();
463 // We still want to format the whitespace left of the first token of the
464 // warning or error.
465 next();
466 while (CurrentToken != NULL) {
467 CurrentToken->Type = TT_ImplicitStringLiteral;
468 next();
469 }
470 }
471
472 void parsePreprocessorDirective() {
473 next();
474 if (CurrentToken == NULL)
475 return;
Alexander Kornienkob18c2582013-10-11 21:43:05 +0000476 if (CurrentToken->Tok.is(tok::numeric_constant)) {
477 CurrentToken->SpacesRequiredBefore = 1;
478 return;
479 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000480 // Hashes in the middle of a line can lead to any strange token
481 // sequence.
Manuel Klimekb3987012013-05-29 14:47:47 +0000482 if (CurrentToken->Tok.getIdentifierInfo() == NULL)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000483 return;
Manuel Klimekb3987012013-05-29 14:47:47 +0000484 switch (CurrentToken->Tok.getIdentifierInfo()->getPPKeywordID()) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000485 case tok::pp_include:
486 case tok::pp_import:
487 parseIncludeDirective();
488 break;
489 case tok::pp_error:
490 case tok::pp_warning:
491 parseWarningOrError();
492 break;
Daniel Jasperaae7bad2013-04-23 13:54:04 +0000493 case tok::pp_if:
494 case tok::pp_elif:
495 parseLine();
496 break;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000497 default:
498 break;
499 }
Daniel Jasper5b7e7b02013-02-05 09:34:14 +0000500 while (CurrentToken != NULL)
501 next();
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000502 }
503
Nico Weber95e8e462013-02-12 16:17:07 +0000504public:
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000505 LineType parseLine() {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000506 if (CurrentToken->is(tok::hash)) {
507 parsePreprocessorDirective();
508 return LT_PreprocessorDirective;
509 }
510 while (CurrentToken != NULL) {
511 if (CurrentToken->is(tok::kw_virtual))
512 KeywordVirtualFound = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000513 if (!consumeToken())
514 return LT_Invalid;
515 }
516 if (KeywordVirtualFound)
517 return LT_VirtualFunctionDecl;
518
Manuel Klimekb3987012013-05-29 14:47:47 +0000519 if (Line.First->Type == TT_ObjCMethodSpecifier) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000520 if (Contexts.back().FirstObjCSelectorName != NULL)
521 Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
522 Contexts.back().LongestObjCSelectorName;
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000523 return LT_ObjCMethodDecl;
524 }
525
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000526 return LT_Other;
527 }
528
Nico Weber95e8e462013-02-12 16:17:07 +0000529private:
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000530 void next() {
Daniel Jasper01786732013-02-04 07:21:18 +0000531 if (CurrentToken != NULL) {
532 determineTokenType(*CurrentToken);
Daniel Jasper4e778092013-02-06 10:05:46 +0000533 CurrentToken->BindingStrength = Contexts.back().BindingStrength;
Daniel Jasper01786732013-02-04 07:21:18 +0000534 }
535
Manuel Klimekb3987012013-05-29 14:47:47 +0000536 if (CurrentToken != NULL)
537 CurrentToken = CurrentToken->Next;
Daniel Jasperd0f349b2013-02-18 12:44:35 +0000538
Manuel Klimekae76f7f2013-10-11 21:25:45 +0000539 if (CurrentToken != NULL) {
540 // Reset token type in case we have already looked at it and then
541 // recovered from an error (e.g. failure to find the matching >).
542 if (CurrentToken->Type != TT_LambdaLSquare &&
543 CurrentToken->Type != TT_ImplicitStringLiteral)
544 CurrentToken->Type = TT_Unknown;
545 if (CurrentToken->Role)
546 CurrentToken->Role.reset(NULL);
547 CurrentToken->FakeLParens.clear();
548 CurrentToken->FakeRParens = 0;
549 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000550 }
551
Daniel Jasper4e778092013-02-06 10:05:46 +0000552 /// \brief A struct to hold information valid in a specific context, e.g.
553 /// a pair of parenthesis.
554 struct Context {
Daniel Jasper923ebef2013-03-14 13:45:21 +0000555 Context(tok::TokenKind ContextKind, unsigned BindingStrength,
556 bool IsExpression)
557 : ContextKind(ContextKind), BindingStrength(BindingStrength),
558 LongestObjCSelectorName(0), ColonIsForRangeExpr(false),
Nico Weberf2ff8122013-05-26 05:39:26 +0000559 ColonIsObjCDictLiteral(false), ColonIsObjCMethodExpr(false),
560 FirstObjCSelectorName(NULL), FirstStartOfName(NULL),
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000561 IsExpression(IsExpression), CanBeExpression(true),
562 InCtorInitializer(false) {}
Daniel Jasper01786732013-02-04 07:21:18 +0000563
Daniel Jasper923ebef2013-03-14 13:45:21 +0000564 tok::TokenKind ContextKind;
Daniel Jasper4e778092013-02-06 10:05:46 +0000565 unsigned BindingStrength;
566 unsigned LongestObjCSelectorName;
567 bool ColonIsForRangeExpr;
Nico Weberf2ff8122013-05-26 05:39:26 +0000568 bool ColonIsObjCDictLiteral;
Daniel Jasper4e778092013-02-06 10:05:46 +0000569 bool ColonIsObjCMethodExpr;
Manuel Klimekb3987012013-05-29 14:47:47 +0000570 FormatToken *FirstObjCSelectorName;
571 FormatToken *FirstStartOfName;
Daniel Jasper4e778092013-02-06 10:05:46 +0000572 bool IsExpression;
Daniel Jasper6f21a982013-03-13 07:49:51 +0000573 bool CanBeExpression;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000574 bool InCtorInitializer;
Daniel Jasper4e778092013-02-06 10:05:46 +0000575 };
576
577 /// \brief Puts a new \c Context onto the stack \c Contexts for the lifetime
578 /// of each instance.
579 struct ScopedContextCreator {
580 AnnotatingParser &P;
581
Daniel Jasper923ebef2013-03-14 13:45:21 +0000582 ScopedContextCreator(AnnotatingParser &P, tok::TokenKind ContextKind,
583 unsigned Increase)
584 : P(P) {
Daniel Jasper2a409b62013-07-08 14:34:09 +0000585 P.Contexts.push_back(Context(ContextKind,
586 P.Contexts.back().BindingStrength + Increase,
587 P.Contexts.back().IsExpression));
Daniel Jasper4e778092013-02-06 10:05:46 +0000588 }
589
590 ~ScopedContextCreator() { P.Contexts.pop_back(); }
591 };
Daniel Jasper01786732013-02-04 07:21:18 +0000592
Manuel Klimekb3987012013-05-29 14:47:47 +0000593 void determineTokenType(FormatToken &Current) {
594 if (Current.getPrecedence() == prec::Assignment &&
Daniel Jasper53352602013-08-12 12:16:34 +0000595 !Line.First->isOneOf(tok::kw_template, tok::kw_using) &&
Manuel Klimekb3987012013-05-29 14:47:47 +0000596 (!Current.Previous || Current.Previous->isNot(tok::kw_operator))) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000597 Contexts.back().IsExpression = true;
Manuel Klimekb3987012013-05-29 14:47:47 +0000598 for (FormatToken *Previous = Current.Previous;
Daniel Jasper7e274002013-09-11 20:37:10 +0000599 Previous && !Previous->isOneOf(tok::comma, tok::semi);
Manuel Klimekb3987012013-05-29 14:47:47 +0000600 Previous = Previous->Previous) {
Daniel Jasper9c65b062013-02-27 11:43:50 +0000601 if (Previous->is(tok::r_square))
602 Previous = Previous->MatchingParen;
Daniel Jasper01786732013-02-04 07:21:18 +0000603 if (Previous->Type == TT_BinaryOperator &&
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000604 Previous->isOneOf(tok::star, tok::amp)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000605 Previous->Type = TT_PointerOrReference;
606 }
Daniel Jasper01786732013-02-04 07:21:18 +0000607 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000608 } else if (Current.isOneOf(tok::kw_return, tok::kw_throw) ||
Nico Weber95e8e462013-02-12 16:17:07 +0000609 (Current.is(tok::l_paren) && !Line.MustBeDeclaration &&
Daniel Jasper378d93d2013-05-13 07:14:40 +0000610 !Line.InPPDirective &&
Manuel Klimek2530fd52013-08-12 03:51:17 +0000611 (!Current.Previous ||
Daniel Jasper53352602013-08-12 12:16:34 +0000612 !Current.Previous->isOneOf(tok::kw_for, tok::kw_catch)))) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000613 Contexts.back().IsExpression = true;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000614 } else if (Current.isOneOf(tok::r_paren, tok::greater, tok::comma)) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000615 for (FormatToken *Previous = Current.Previous;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000616 Previous && Previous->isOneOf(tok::star, tok::amp);
Manuel Klimekb3987012013-05-29 14:47:47 +0000617 Previous = Previous->Previous)
Nico Weber95e8e462013-02-12 16:17:07 +0000618 Previous->Type = TT_PointerOrReference;
Manuel Klimekb3987012013-05-29 14:47:47 +0000619 } else if (Current.Previous &&
620 Current.Previous->Type == TT_CtorInitializerColon) {
Daniel Jasperd0f349b2013-02-18 12:44:35 +0000621 Contexts.back().IsExpression = true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000622 Contexts.back().InCtorInitializer = true;
Daniel Jasper6f21a982013-03-13 07:49:51 +0000623 } else if (Current.is(tok::kw_new)) {
624 Contexts.back().CanBeExpression = false;
Daniel Jasper16a69ef2013-05-03 14:41:24 +0000625 } else if (Current.is(tok::semi)) {
626 // This should be the condition or increment in a for-loop.
627 Contexts.back().IsExpression = true;
Nico Weber95e8e462013-02-12 16:17:07 +0000628 }
Daniel Jasper01786732013-02-04 07:21:18 +0000629
630 if (Current.Type == TT_Unknown) {
Daniel Jasper6b3ff8c2013-09-27 08:29:16 +0000631 // Line.MightBeFunctionDecl can only be true after the parentheses of a
632 // function declaration have been found. In this case, 'Current' is a
633 // trailing token of this declaration and thus cannot be a name.
634 if (isStartOfName(Current) && !Line.MightBeFunctionDecl) {
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000635 Contexts.back().FirstStartOfName = &Current;
Daniel Jasper3c08a812013-02-24 18:54:32 +0000636 Current.Type = TT_StartOfName;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000637 NameFound = true;
Daniel Jasper2ca37412013-07-09 14:36:48 +0000638 } else if (Current.is(tok::kw_auto)) {
639 AutoFound = true;
Daniel Jasper3262f4c2013-07-11 14:33:06 +0000640 } else if (Current.is(tok::arrow) && AutoFound &&
641 Line.MustBeDeclaration) {
Daniel Jasper2ca37412013-07-09 14:36:48 +0000642 Current.Type = TT_TrailingReturnArrow;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000643 } else if (Current.isOneOf(tok::star, tok::amp, tok::ampamp)) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000644 Current.Type =
Daniel Jasperd6104f62013-07-05 13:30:40 +0000645 determineStarAmpUsage(Current, Contexts.back().CanBeExpression &&
646 Contexts.back().IsExpression);
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000647 } else if (Current.isOneOf(tok::minus, tok::plus, tok::caret)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000648 Current.Type = determinePlusMinusCaretUsage(Current);
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000649 } else if (Current.isOneOf(tok::minusminus, tok::plusplus)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000650 Current.Type = determineIncrementUsage(Current);
651 } else if (Current.is(tok::exclaim)) {
652 Current.Type = TT_UnaryOperator;
Manuel Klimek31e44f72013-09-04 08:20:47 +0000653 } else if (Current.isBinaryOperator() &&
654 (!Current.Previous ||
655 Current.Previous->isNot(tok::l_square))) {
Daniel Jasper01786732013-02-04 07:21:18 +0000656 Current.Type = TT_BinaryOperator;
657 } else if (Current.is(tok::comment)) {
Alexander Kornienko00895102013-06-05 14:09:10 +0000658 if (Current.TokenText.startswith("//"))
Daniel Jasper01786732013-02-04 07:21:18 +0000659 Current.Type = TT_LineComment;
660 else
661 Current.Type = TT_BlockComment;
Nico Weber37d69312013-02-13 04:13:13 +0000662 } else if (Current.is(tok::r_paren)) {
Daniel Jasperb8b42952013-05-31 16:14:28 +0000663 FormatToken *LeftOfParens = NULL;
664 if (Current.MatchingParen)
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000665 LeftOfParens = Current.MatchingParen->getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000666 bool IsCast = false;
667 bool ParensAreEmpty = Current.Previous == Current.MatchingParen;
668 bool ParensAreType = !Current.Previous ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000669 Current.Previous->Type == TT_PointerOrReference ||
Daniel Jasperb8b42952013-05-31 16:14:28 +0000670 Current.Previous->Type == TT_TemplateCloser ||
671 isSimpleTypeSpecifier(*Current.Previous);
Nico Weber37d69312013-02-13 04:13:13 +0000672 bool ParensCouldEndDecl =
Manuel Klimekb3987012013-05-29 14:47:47 +0000673 Current.Next &&
674 Current.Next->isOneOf(tok::equal, tok::semi, tok::l_brace);
Daniel Jasper6a365aa2013-03-13 17:13:53 +0000675 bool IsSizeOfOrAlignOf =
Daniel Jasperb8b42952013-05-31 16:14:28 +0000676 LeftOfParens &&
677 LeftOfParens->isOneOf(tok::kw_sizeof, tok::kw_alignof);
678 if (ParensAreType && !ParensCouldEndDecl && !IsSizeOfOrAlignOf &&
Daniel Jasper0c368782013-07-15 15:04:42 +0000679 (Contexts.back().IsExpression ||
680 (Current.Next && Current.Next->isBinaryOperator())))
Daniel Jasperb8b42952013-05-31 16:14:28 +0000681 IsCast = true;
Daniel Jasper2a409b62013-07-08 14:34:09 +0000682 if (Current.Next && Current.Next->isNot(tok::string_literal) &&
Daniel Jasperb8b42952013-05-31 16:14:28 +0000683 (Current.Next->Tok.isLiteral() ||
684 Current.Next->isOneOf(tok::kw_sizeof, tok::kw_alignof)))
685 IsCast = true;
686 // If there is an identifier after the (), it is likely a cast, unless
687 // there is also an identifier before the ().
Daniel Jasperff1a2e52013-06-06 08:20:20 +0000688 if (LeftOfParens && (LeftOfParens->Tok.getIdentifierInfo() == NULL ||
689 LeftOfParens->is(tok::kw_return)) &&
Daniel Jasper526df0f2013-07-08 14:58:01 +0000690 LeftOfParens->Type != TT_OverloadedOperator &&
Nico Weber465e8612013-06-25 00:55:57 +0000691 LeftOfParens->Type != TT_TemplateCloser && Current.Next &&
692 Current.Next->is(tok::identifier))
Daniel Jasperb8b42952013-05-31 16:14:28 +0000693 IsCast = true;
694 if (IsCast && !ParensAreEmpty)
Nico Weber37d69312013-02-13 04:13:13 +0000695 Current.Type = TT_CastRParen;
Manuel Klimekb3987012013-05-29 14:47:47 +0000696 } else if (Current.is(tok::at) && Current.Next) {
697 switch (Current.Next->Tok.getObjCKeywordID()) {
Daniel Jasper01786732013-02-04 07:21:18 +0000698 case tok::objc_interface:
699 case tok::objc_implementation:
700 case tok::objc_protocol:
701 Current.Type = TT_ObjCDecl;
702 break;
703 case tok::objc_property:
704 Current.Type = TT_ObjCProperty;
705 break;
706 default:
707 break;
708 }
Daniel Jasper5ad390d2013-05-28 11:30:49 +0000709 } else if (Current.is(tok::period)) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000710 FormatToken *PreviousNoComment = Current.getPreviousNonComment();
Daniel Jasper5ad390d2013-05-28 11:30:49 +0000711 if (PreviousNoComment &&
712 PreviousNoComment->isOneOf(tok::comma, tok::l_brace))
713 Current.Type = TT_DesignatedInitializerPeriod;
Daniel Jasper01786732013-02-04 07:21:18 +0000714 }
715 }
716 }
717
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000718 /// \brief Take a guess at whether \p Tok starts a name of a function or
719 /// variable declaration.
720 ///
721 /// This is a heuristic based on whether \p Tok is an identifier following
722 /// something that is likely a type.
723 bool isStartOfName(const FormatToken &Tok) {
724 if (Tok.isNot(tok::identifier) || Tok.Previous == NULL)
725 return false;
726
727 // Skip "const" as it does not have an influence on whether this is a name.
728 FormatToken *PreviousNotConst = Tok.Previous;
729 while (PreviousNotConst != NULL && PreviousNotConst->is(tok::kw_const))
730 PreviousNotConst = PreviousNotConst->Previous;
731
732 if (PreviousNotConst == NULL)
733 return false;
734
Daniel Jasper2a409b62013-07-08 14:34:09 +0000735 bool IsPPKeyword = PreviousNotConst->is(tok::identifier) &&
736 PreviousNotConst->Previous &&
737 PreviousNotConst->Previous->is(tok::hash);
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000738
Daniel Jasper92495a82013-08-19 10:16:18 +0000739 if (PreviousNotConst->Type == TT_TemplateCloser)
740 return PreviousNotConst && PreviousNotConst->MatchingParen &&
741 PreviousNotConst->MatchingParen->Previous &&
742 PreviousNotConst->MatchingParen->Previous->isNot(tok::kw_template);
743
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000744 return (!IsPPKeyword && PreviousNotConst->is(tok::identifier)) ||
745 PreviousNotConst->Type == TT_PointerOrReference ||
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000746 isSimpleTypeSpecifier(*PreviousNotConst);
747 }
748
Daniel Jasper01786732013-02-04 07:21:18 +0000749 /// \brief Return the type of the given token assuming it is * or &.
Manuel Klimekb3987012013-05-29 14:47:47 +0000750 TokenType determineStarAmpUsage(const FormatToken &Tok, bool IsExpression) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000751 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasper01786732013-02-04 07:21:18 +0000752 if (PrevToken == NULL)
753 return TT_UnaryOperator;
754
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000755 const FormatToken *NextToken = Tok.getNextNonComment();
Daniel Jasper01786732013-02-04 07:21:18 +0000756 if (NextToken == NULL)
757 return TT_Unknown;
758
Daniel Jasper431f5912013-05-28 08:33:00 +0000759 if (PrevToken->is(tok::coloncolon) ||
760 (PrevToken->is(tok::l_paren) && !IsExpression))
Daniel Jasper8a5d7cd2013-03-01 17:13:29 +0000761 return TT_PointerOrReference;
762
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000763 if (PrevToken->isOneOf(tok::l_paren, tok::l_square, tok::l_brace,
Daniel Jasperd3cf17b2013-03-14 10:50:25 +0000764 tok::comma, tok::semi, tok::kw_return, tok::colon,
Daniel Jasper65da8e92013-09-21 17:31:51 +0000765 tok::equal, tok::kw_delete, tok::kw_sizeof) ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000766 PrevToken->Type == TT_BinaryOperator ||
Daniel Jasper01786732013-02-04 07:21:18 +0000767 PrevToken->Type == TT_UnaryOperator || PrevToken->Type == TT_CastRParen)
768 return TT_UnaryOperator;
769
Nico Webere8a97982013-02-06 06:20:11 +0000770 if (NextToken->is(tok::l_square))
771 return TT_PointerOrReference;
772
Daniel Jasperdb8afe42013-09-10 10:26:38 +0000773 if (PrevToken->is(tok::r_paren) && PrevToken->MatchingParen &&
774 PrevToken->MatchingParen->Previous &&
775 PrevToken->MatchingParen->Previous->is(tok::kw_typeof))
776 return TT_PointerOrReference;
777
Manuel Klimekb3987012013-05-29 14:47:47 +0000778 if (PrevToken->Tok.isLiteral() ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000779 PrevToken->isOneOf(tok::r_paren, tok::r_square) ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000780 NextToken->Tok.isLiteral() || NextToken->isUnaryOperator())
Daniel Jasper01786732013-02-04 07:21:18 +0000781 return TT_BinaryOperator;
782
Daniel Jasper01786732013-02-04 07:21:18 +0000783 // It is very unlikely that we are going to find a pointer or reference type
784 // definition on the RHS of an assignment.
785 if (IsExpression)
786 return TT_BinaryOperator;
787
788 return TT_PointerOrReference;
789 }
790
Manuel Klimekb3987012013-05-29 14:47:47 +0000791 TokenType determinePlusMinusCaretUsage(const FormatToken &Tok) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000792 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000793 if (PrevToken == NULL || PrevToken->Type == TT_CastRParen)
Daniel Jasper01786732013-02-04 07:21:18 +0000794 return TT_UnaryOperator;
795
796 // Use heuristics to recognize unary operators.
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000797 if (PrevToken->isOneOf(tok::equal, tok::l_paren, tok::comma, tok::l_square,
798 tok::question, tok::colon, tok::kw_return,
799 tok::kw_case, tok::at, tok::l_brace))
Daniel Jasper01786732013-02-04 07:21:18 +0000800 return TT_UnaryOperator;
801
Nico Weberee0feec2013-02-05 16:21:00 +0000802 // There can't be two consecutive binary operators.
Daniel Jasper01786732013-02-04 07:21:18 +0000803 if (PrevToken->Type == TT_BinaryOperator)
804 return TT_UnaryOperator;
805
806 // Fall back to marking the token as binary operator.
807 return TT_BinaryOperator;
808 }
809
810 /// \brief Determine whether ++/-- are pre- or post-increments/-decrements.
Manuel Klimekb3987012013-05-29 14:47:47 +0000811 TokenType determineIncrementUsage(const FormatToken &Tok) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000812 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000813 if (PrevToken == NULL || PrevToken->Type == TT_CastRParen)
Daniel Jasper01786732013-02-04 07:21:18 +0000814 return TT_UnaryOperator;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000815 if (PrevToken->isOneOf(tok::r_paren, tok::r_square, tok::identifier))
Daniel Jasper01786732013-02-04 07:21:18 +0000816 return TT_TrailingUnaryOperator;
817
818 return TT_UnaryOperator;
819 }
Daniel Jasper4e778092013-02-06 10:05:46 +0000820
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000821 // FIXME: This is copy&pasted from Sema. Put it in a common place and remove
822 // duplication.
823 /// \brief Determine whether the token kind starts a simple-type-specifier.
Manuel Klimekb3987012013-05-29 14:47:47 +0000824 bool isSimpleTypeSpecifier(const FormatToken &Tok) const {
825 switch (Tok.Tok.getKind()) {
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000826 case tok::kw_short:
827 case tok::kw_long:
828 case tok::kw___int64:
829 case tok::kw___int128:
830 case tok::kw_signed:
831 case tok::kw_unsigned:
832 case tok::kw_void:
833 case tok::kw_char:
834 case tok::kw_int:
835 case tok::kw_half:
836 case tok::kw_float:
837 case tok::kw_double:
838 case tok::kw_wchar_t:
839 case tok::kw_bool:
840 case tok::kw___underlying_type:
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000841 case tok::annot_typename:
842 case tok::kw_char16_t:
843 case tok::kw_char32_t:
844 case tok::kw_typeof:
845 case tok::kw_decltype:
Alexander Kornienko00895102013-06-05 14:09:10 +0000846 return true;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000847 default:
Alexander Kornienko00895102013-06-05 14:09:10 +0000848 return false;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000849 }
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000850 }
851
Daniel Jasper4e778092013-02-06 10:05:46 +0000852 SmallVector<Context, 8> Contexts;
853
Daniel Jasperd4a03db2013-08-22 15:00:41 +0000854 const FormatStyle &Style;
Daniel Jasper4e778092013-02-06 10:05:46 +0000855 AnnotatedLine &Line;
Manuel Klimekb3987012013-05-29 14:47:47 +0000856 FormatToken *CurrentToken;
Daniel Jasper4e778092013-02-06 10:05:46 +0000857 bool KeywordVirtualFound;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000858 bool NameFound;
Daniel Jasper2ca37412013-07-09 14:36:48 +0000859 bool AutoFound;
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000860 IdentifierInfo &Ident_in;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000861};
862
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000863static int PrecedenceUnaryOperator = prec::PointerToMember + 1;
864static int PrecedenceArrowAndPeriod = prec::PointerToMember + 2;
865
Daniel Jasper29f123b2013-02-08 15:28:42 +0000866/// \brief Parses binary expressions by inserting fake parenthesis based on
867/// operator precedence.
868class ExpressionParser {
869public:
Daniel Jasper9acb8b42013-06-06 09:11:58 +0000870 ExpressionParser(AnnotatedLine &Line) : Current(Line.First) {
871 // Skip leading "}", e.g. in "} else if (...) {".
872 if (Current->is(tok::r_brace))
873 next();
874 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000875
876 /// \brief Parse expressions with the given operatore precedence.
Daniel Jasper237d4c12013-02-23 21:01:55 +0000877 void parse(int Precedence = 0) {
Daniel Jasperf78bf4a2013-09-30 08:29:03 +0000878 // Skip 'return' as it is not part of a binary expression.
879 while (Current && Current->is(tok::kw_return))
880 next();
881
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000882 if (Current == NULL || Precedence > PrecedenceArrowAndPeriod)
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000883 return;
884
Daniel Jasperc01897c2013-05-31 14:56:12 +0000885 // Conditional expressions need to be parsed separately for proper nesting.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000886 if (Precedence == prec::Conditional) {
Daniel Jasperc01897c2013-05-31 14:56:12 +0000887 parseConditionalExpr();
888 return;
889 }
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000890
891 // Parse unary operators, which all have a higher precedence than binary
892 // operators.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000893 if (Precedence == PrecedenceUnaryOperator) {
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000894 parseUnaryOperator();
Daniel Jasper29f123b2013-02-08 15:28:42 +0000895 return;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000896 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000897
Manuel Klimekb3987012013-05-29 14:47:47 +0000898 FormatToken *Start = Current;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000899 FormatToken *LatestOperator = NULL;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000900
Daniel Jasper237d4c12013-02-23 21:01:55 +0000901 while (Current) {
Daniel Jasper29f123b2013-02-08 15:28:42 +0000902 // Consume operators with higher precedence.
Daniel Jasperbf71ba22013-04-08 20:33:42 +0000903 parse(Precedence + 1);
Daniel Jasper29f123b2013-02-08 15:28:42 +0000904
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000905 int CurrentPrecedence = getCurrentPrecedence();
906
907 if (Current && Current->Type == TT_ObjCSelectorName &&
908 Precedence == CurrentPrecedence)
909 Start = Current;
Daniel Jasper237d4c12013-02-23 21:01:55 +0000910
Daniel Jasper29f123b2013-02-08 15:28:42 +0000911 // At the end of the line or when an operator with higher precedence is
912 // found, insert fake parenthesis and return.
Daniel Jasperac3223e2013-04-10 09:49:49 +0000913 if (Current == NULL || Current->closesScope() ||
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000914 (CurrentPrecedence != -1 && CurrentPrecedence < Precedence)) {
915 if (LatestOperator) {
916 if (Precedence == PrecedenceArrowAndPeriod) {
917 LatestOperator->LastInChainOfCalls = true;
918 // Call expressions don't have a binary operator precedence.
919 addFakeParenthesis(Start, prec::Unknown);
920 } else {
921 addFakeParenthesis(Start, prec::Level(Precedence));
922 }
923 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000924 return;
925 }
926
927 // Consume scopes: (), [], <> and {}
Daniel Jasperac3223e2013-04-10 09:49:49 +0000928 if (Current->opensScope()) {
929 while (Current && !Current->closesScope()) {
Daniel Jasper29f123b2013-02-08 15:28:42 +0000930 next();
931 parse();
932 }
933 next();
934 } else {
935 // Operator found.
Daniel Jasper237d4c12013-02-23 21:01:55 +0000936 if (CurrentPrecedence == Precedence)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000937 LatestOperator = Current;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000938
939 next();
940 }
941 }
942 }
943
944private:
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000945 /// \brief Gets the precedence (+1) of the given token for binary operators
946 /// and other tokens that we treat like binary operators.
947 int getCurrentPrecedence() {
948 if (Current) {
949 if (Current->Type == TT_ConditionalExpr)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000950 return prec::Conditional;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000951 else if (Current->is(tok::semi) || Current->Type == TT_InlineASMColon)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000952 return 0;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000953 else if (Current->Type == TT_BinaryOperator || Current->is(tok::comma))
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000954 return Current->getPrecedence();
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000955 else if (Current->Type == TT_ObjCSelectorName)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000956 return prec::Assignment;
957 else if (Current->isOneOf(tok::period, tok::arrow))
958 return PrecedenceArrowAndPeriod;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000959 }
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000960 return -1;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000961 }
962
Daniel Jasperc01897c2013-05-31 14:56:12 +0000963 void addFakeParenthesis(FormatToken *Start, prec::Level Precedence) {
964 Start->FakeLParens.push_back(Precedence);
Daniel Jasperdb4813a2013-09-06 08:08:14 +0000965 if (Precedence > prec::Unknown)
966 Start->StartsBinaryExpression = true;
967 if (Current) {
Daniel Jasperc01897c2013-05-31 14:56:12 +0000968 ++Current->Previous->FakeRParens;
Daniel Jasperdb4813a2013-09-06 08:08:14 +0000969 if (Precedence > prec::Unknown)
970 Current->Previous->EndsBinaryExpression = true;
971 }
Daniel Jasperc01897c2013-05-31 14:56:12 +0000972 }
973
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000974 /// \brief Parse unary operator expressions and surround them with fake
975 /// parentheses if appropriate.
976 void parseUnaryOperator() {
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000977 if (Current == NULL || Current->Type != TT_UnaryOperator) {
978 parse(PrecedenceArrowAndPeriod);
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000979 return;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000980 }
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000981
982 FormatToken *Start = Current;
983 next();
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000984 parseUnaryOperator();
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000985
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000986 // The actual precedence doesn't matter.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000987 addFakeParenthesis(Start, prec::Unknown);
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000988 }
989
Daniel Jasperc01897c2013-05-31 14:56:12 +0000990 void parseConditionalExpr() {
991 FormatToken *Start = Current;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000992 parse(prec::LogicalOr);
Daniel Jasperc01897c2013-05-31 14:56:12 +0000993 if (!Current || !Current->is(tok::question))
994 return;
995 next();
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000996 parse(prec::LogicalOr);
Daniel Jasperc01897c2013-05-31 14:56:12 +0000997 if (!Current || Current->Type != TT_ConditionalExpr)
998 return;
999 next();
1000 parseConditionalExpr();
1001 addFakeParenthesis(Start, prec::Conditional);
1002 }
1003
Daniel Jasper29f123b2013-02-08 15:28:42 +00001004 void next() {
Alexander Kornienkod71b15b2013-06-17 13:19:53 +00001005 if (Current)
1006 Current = Current->Next;
1007 while (Current && Current->isTrailingComment())
Manuel Klimekb3987012013-05-29 14:47:47 +00001008 Current = Current->Next;
Daniel Jasper29f123b2013-02-08 15:28:42 +00001009 }
1010
Manuel Klimekb3987012013-05-29 14:47:47 +00001011 FormatToken *Current;
Daniel Jasper29f123b2013-02-08 15:28:42 +00001012};
1013
Craig Topper14e66492013-07-01 04:03:19 +00001014} // end anonymous namespace
1015
Daniel Jasperb77d7412013-09-06 07:54:20 +00001016void
1017TokenAnnotator::setCommentLineLevels(SmallVectorImpl<AnnotatedLine *> &Lines) {
1018 if (Lines.empty())
1019 return;
1020
1021 const AnnotatedLine *NextNonCommentLine = NULL;
1022 for (unsigned i = Lines.size() - 1; i > 0; --i) {
1023 if (NextNonCommentLine && Lines[i]->First->is(tok::comment) &&
1024 !Lines[i]->First->Next)
1025 Lines[i]->Level = NextNonCommentLine->Level;
1026 else
1027 NextNonCommentLine =
1028 Lines[i]->First->isNot(tok::r_brace) ? Lines[i] : NULL;
1029 }
1030}
1031
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001032void TokenAnnotator::annotate(AnnotatedLine &Line) {
Daniel Jasperb77d7412013-09-06 07:54:20 +00001033 setCommentLineLevels(Line.Children);
1034 for (SmallVectorImpl<AnnotatedLine *>::iterator I = Line.Children.begin(),
1035 E = Line.Children.end();
Daniel Jasper567dcf92013-09-05 09:29:45 +00001036 I != E; ++I) {
1037 annotate(**I);
1038 }
Daniel Jasperd4a03db2013-08-22 15:00:41 +00001039 AnnotatingParser Parser(Style, Line, Ident_in);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001040 Line.Type = Parser.parseLine();
1041 if (Line.Type == LT_Invalid)
1042 return;
1043
Daniel Jasper29f123b2013-02-08 15:28:42 +00001044 ExpressionParser ExprParser(Line);
1045 ExprParser.parse();
1046
Manuel Klimekb3987012013-05-29 14:47:47 +00001047 if (Line.First->Type == TT_ObjCMethodSpecifier)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001048 Line.Type = LT_ObjCMethodDecl;
Manuel Klimekb3987012013-05-29 14:47:47 +00001049 else if (Line.First->Type == TT_ObjCDecl)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001050 Line.Type = LT_ObjCDecl;
Manuel Klimekb3987012013-05-29 14:47:47 +00001051 else if (Line.First->Type == TT_ObjCProperty)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001052 Line.Type = LT_ObjCProperty;
1053
Manuel Klimekb3987012013-05-29 14:47:47 +00001054 Line.First->SpacesRequiredBefore = 1;
1055 Line.First->CanBreakBefore = Line.First->MustBreakBefore;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001056}
1057
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001058void TokenAnnotator::calculateFormattingInformation(AnnotatedLine &Line) {
Alexander Kornienko83a7dcd2013-09-10 09:38:25 +00001059 Line.First->TotalLength =
1060 Line.First->IsMultiline ? Style.ColumnLimit : Line.First->ColumnWidth;
Manuel Klimekb3987012013-05-29 14:47:47 +00001061 if (!Line.First->Next)
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001062 return;
Manuel Klimekb3987012013-05-29 14:47:47 +00001063 FormatToken *Current = Line.First->Next;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001064 while (Current != NULL) {
Daniel Jasper729a7432013-02-11 12:36:37 +00001065 if (Current->Type == TT_LineComment)
1066 Current->SpacesRequiredBefore = Style.SpacesBeforeTrailingComments;
Alexander Kornienkob18c2582013-10-11 21:43:05 +00001067 else if (Current->SpacesRequiredBefore == 0 &&
1068 spaceRequiredBefore(Line, *Current))
1069 Current->SpacesRequiredBefore = 1;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001070
Daniel Jasperebaa1712013-09-17 09:52:48 +00001071 Current->MustBreakBefore =
1072 Current->MustBreakBefore || mustBreakBefore(Line, *Current);
1073
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001074 Current->CanBreakBefore =
1075 Current->MustBreakBefore || canBreakBefore(Line, *Current);
Daniel Jasper567dcf92013-09-05 09:29:45 +00001076 if (Current->MustBreakBefore || !Current->Children.empty() ||
Alexander Kornienko83a7dcd2013-09-10 09:38:25 +00001077 Current->IsMultiline)
Manuel Klimekb3987012013-05-29 14:47:47 +00001078 Current->TotalLength = Current->Previous->TotalLength + Style.ColumnLimit;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001079 else
Daniel Jasper2a409b62013-07-08 14:34:09 +00001080 Current->TotalLength = Current->Previous->TotalLength +
Alexander Kornienko83a7dcd2013-09-10 09:38:25 +00001081 Current->ColumnWidth +
Daniel Jasper2a409b62013-07-08 14:34:09 +00001082 Current->SpacesRequiredBefore;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001083 // FIXME: Only calculate this if CanBreakBefore is true once static
1084 // initializers etc. are sorted out.
1085 // FIXME: Move magic numbers to a better place.
1086 Current->SplitPenalty =
1087 20 * Current->BindingStrength + splitPenalty(Line, *Current);
1088
Manuel Klimekb3987012013-05-29 14:47:47 +00001089 Current = Current->Next;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001090 }
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001091
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001092 calculateUnbreakableTailLengths(Line);
Daniel Jasperd4a03db2013-08-22 15:00:41 +00001093 for (Current = Line.First; Current != NULL; Current = Current->Next) {
1094 if (Current->Role)
1095 Current->Role->precomputeFormattingInfos(Current);
1096 }
1097
Daniel Jasper567dcf92013-09-05 09:29:45 +00001098 DEBUG({ printDebugInfo(Line); });
1099
Daniel Jasperb77d7412013-09-06 07:54:20 +00001100 for (SmallVectorImpl<AnnotatedLine *>::iterator I = Line.Children.begin(),
1101 E = Line.Children.end();
Daniel Jasper567dcf92013-09-05 09:29:45 +00001102 I != E; ++I) {
1103 calculateFormattingInformation(**I);
1104 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001105}
1106
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001107void TokenAnnotator::calculateUnbreakableTailLengths(AnnotatedLine &Line) {
1108 unsigned UnbreakableTailLength = 0;
Manuel Klimekb3987012013-05-29 14:47:47 +00001109 FormatToken *Current = Line.Last;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001110 while (Current != NULL) {
1111 Current->UnbreakableTailLength = UnbreakableTailLength;
1112 if (Current->CanBreakBefore ||
1113 Current->isOneOf(tok::comment, tok::string_literal)) {
1114 UnbreakableTailLength = 0;
1115 } else {
1116 UnbreakableTailLength +=
Alexander Kornienko83a7dcd2013-09-10 09:38:25 +00001117 Current->ColumnWidth + Current->SpacesRequiredBefore;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001118 }
Manuel Klimekb3987012013-05-29 14:47:47 +00001119 Current = Current->Previous;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001120 }
1121}
1122
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001123unsigned TokenAnnotator::splitPenalty(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001124 const FormatToken &Tok) {
1125 const FormatToken &Left = *Tok.Previous;
1126 const FormatToken &Right = Tok;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001127
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001128 if (Left.is(tok::semi))
1129 return 0;
1130 if (Left.is(tok::comma))
1131 return 1;
Daniel Jasper011c35d2013-07-12 11:19:37 +00001132 if (Right.is(tok::l_square))
1133 return 150;
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001134
Daniel Jasper6561f6a2013-07-09 07:43:55 +00001135 if (Right.Type == TT_StartOfName || Right.is(tok::kw_operator)) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001136 if (Line.First->is(tok::kw_for) && Right.PartOfMultiVariableDeclStmt)
Daniel Jasper3c08a812013-02-24 18:54:32 +00001137 return 3;
Daniel Jasperc18cff32013-07-11 12:34:23 +00001138 if (Left.Type == TT_StartOfName)
1139 return 20;
Daniel Jasper92495a82013-08-19 10:16:18 +00001140 if (Line.MightBeFunctionDecl && Right.BindingStrength == 1)
Daniel Jasper3c08a812013-02-24 18:54:32 +00001141 // FIXME: Clean up hack of using BindingStrength to find top-level names.
1142 return Style.PenaltyReturnTypeOnItsOwnLine;
Daniel Jasper92495a82013-08-19 10:16:18 +00001143 return 200;
Daniel Jasper3c08a812013-02-24 18:54:32 +00001144 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001145 if (Left.is(tok::equal) && Right.is(tok::l_brace))
1146 return 150;
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001147 if (Left.Type == TT_CastRParen)
1148 return 100;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001149 if (Left.is(tok::coloncolon))
1150 return 500;
Daniel Jasper6b119d62013-04-05 17:22:09 +00001151 if (Left.isOneOf(tok::kw_class, tok::kw_struct))
1152 return 5000;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001153
Daniel Jasper6cabab42013-02-14 08:42:54 +00001154 if (Left.Type == TT_RangeBasedForLoopColon ||
1155 Left.Type == TT_InheritanceColon)
Daniel Jasper84a1a632013-02-26 13:18:08 +00001156 return 2;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001157
Daniel Jasperd3fef0f2013-08-27 14:24:43 +00001158 if (Right.isMemberAccess()) {
Daniel Jasper2f0a0202013-09-06 08:54:24 +00001159 if (Left.isOneOf(tok::r_paren, tok::r_square) && Left.MatchingParen &&
1160 Left.MatchingParen->ParameterCount > 0)
Daniel Jasper518ee342013-02-26 13:59:14 +00001161 return 20; // Should be smaller than breaking at a nested comma.
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001162 return 150;
1163 }
1164
Daniel Jasper20a0f8c2013-07-11 21:02:56 +00001165 // Breaking before a trailing 'const' or not-function-like annotation is bad.
Daniel Jasperaa9e7b12013-08-01 13:46:58 +00001166 if (Left.is(tok::r_paren) && Line.Type != LT_ObjCProperty &&
Daniel Jasper20a0f8c2013-07-11 21:02:56 +00001167 (Right.is(tok::kw_const) || (Right.is(tok::identifier) && Right.Next &&
1168 Right.Next->isNot(tok::l_paren))))
Daniel Jasper53eb05a2013-10-18 16:34:40 +00001169 return 100;
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001170
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001171 // In for-loops, prefer breaking at ',' and ';'.
Manuel Klimekb3987012013-05-29 14:47:47 +00001172 if (Line.First->is(tok::kw_for) && Left.is(tok::equal))
Daniel Jasper7d812812013-02-21 15:00:29 +00001173 return 4;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001174
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001175 // In Objective-C method expressions, prefer breaking before "param:" over
1176 // breaking after it.
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001177 if (Right.Type == TT_ObjCSelectorName)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001178 return 0;
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001179 if (Left.is(tok::colon) && Left.Type == TT_ObjCMethodExpr)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001180 return 20;
1181
Daniel Jasper1407bee2013-04-11 14:29:13 +00001182 if (Left.is(tok::l_paren) && Line.MightBeFunctionDecl)
1183 return 100;
Daniel Jasperac3223e2013-04-10 09:49:49 +00001184 if (Left.opensScope())
Daniel Jaspere60084d2013-08-13 06:50:04 +00001185 return Left.ParameterCount > 1 ? prec::Comma : 19;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001186
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001187 if (Right.is(tok::lessless)) {
1188 if (Left.is(tok::string_literal)) {
Alexander Kornienko00895102013-06-05 14:09:10 +00001189 StringRef Content = Left.TokenText;
Daniel Jasper20376982013-09-29 12:02:57 +00001190 if (Content.startswith("\""))
1191 Content = Content.drop_front(1);
1192 if (Content.endswith("\""))
1193 Content = Content.drop_back(1);
1194 Content = Content.trim();
Daniel Jasperbfa1edd2013-03-14 14:00:17 +00001195 if (Content.size() > 1 &&
1196 (Content.back() == ':' || Content.back() == '='))
Daniel Jasper9637dda2013-07-15 14:33:14 +00001197 return 25;
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001198 }
Daniel Jasper0c368782013-07-15 15:04:42 +00001199 return 1; // Breaking at a << is really cheap.
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001200 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001201 if (Left.Type == TT_ConditionalExpr)
Daniel Jasper518ee342013-02-26 13:59:14 +00001202 return prec::Conditional;
Manuel Klimekb3987012013-05-29 14:47:47 +00001203 prec::Level Level = Left.getPrecedence();
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001204
1205 if (Level != prec::Unknown)
1206 return Level;
Daniel Jasper24849712013-03-01 16:48:32 +00001207
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001208 return 3;
1209}
1210
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001211bool TokenAnnotator::spaceRequiredBetween(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001212 const FormatToken &Left,
1213 const FormatToken &Right) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001214 if (Right.is(tok::hashhash))
1215 return Left.is(tok::hash);
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001216 if (Left.isOneOf(tok::hashhash, tok::hash))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001217 return Right.is(tok::hash);
Daniel Jasper7df56bf2013-08-20 12:36:34 +00001218 if (Left.is(tok::l_paren) && Right.is(tok::r_paren))
1219 return Style.SpaceInEmptyParentheses;
1220 if (Left.is(tok::l_paren) || Right.is(tok::r_paren))
Daniel Jasper34f3d052013-08-21 08:39:01 +00001221 return (Right.Type == TT_CastRParen ||
1222 (Left.MatchingParen && Left.MatchingParen->Type == TT_CastRParen))
Daniel Jasper7df56bf2013-08-20 12:36:34 +00001223 ? Style.SpacesInCStyleCastParentheses
1224 : Style.SpacesInParentheses;
1225 if (Right.isOneOf(tok::semi, tok::comma))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001226 return false;
1227 if (Right.is(tok::less) &&
1228 (Left.is(tok::kw_template) ||
1229 (Line.Type == LT_ObjCDecl && Style.ObjCSpaceBeforeProtocolList)))
1230 return true;
1231 if (Left.is(tok::arrow) || Right.is(tok::arrow))
1232 return false;
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001233 if (Left.isOneOf(tok::exclaim, tok::tilde))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001234 return false;
1235 if (Left.is(tok::at) &&
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001236 Right.isOneOf(tok::identifier, tok::string_literal, tok::char_constant,
1237 tok::numeric_constant, tok::l_paren, tok::l_brace,
1238 tok::kw_true, tok::kw_false))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001239 return false;
1240 if (Left.is(tok::coloncolon))
1241 return false;
1242 if (Right.is(tok::coloncolon))
Daniel Jasper78a4e612013-10-12 05:16:06 +00001243 return (Left.is(tok::less) && Style.Standard == FormatStyle::LS_Cpp03) ||
1244 !Left.isOneOf(tok::identifier, tok::greater, tok::l_paren,
1245 tok::r_paren, tok::less);
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001246 if (Left.is(tok::less) || Right.isOneOf(tok::greater, tok::less))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001247 return false;
Daniel Jasperc47d7f12013-07-01 09:47:25 +00001248 if (Right.is(tok::ellipsis))
Daniel Jasperb3c887d2013-10-20 16:56:16 +00001249 return Left.Tok.isLiteral();
Manuel Klimek31e44f72013-09-04 08:20:47 +00001250 if (Left.is(tok::l_square) && Right.is(tok::amp))
1251 return false;
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001252 if (Right.Type == TT_PointerOrReference)
Manuel Klimekb3987012013-05-29 14:47:47 +00001253 return Left.Tok.isLiteral() ||
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001254 ((Left.Type != TT_PointerOrReference) && Left.isNot(tok::l_paren) &&
1255 !Style.PointerBindsToType);
Daniel Jasper3ff4a2f2013-05-28 15:27:10 +00001256 if (Right.Type == TT_FunctionTypeLParen && Left.isNot(tok::l_paren) &&
Daniel Jasper395228f2013-05-08 14:58:20 +00001257 (Left.Type != TT_PointerOrReference || Style.PointerBindsToType))
1258 return true;
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001259 if (Left.Type == TT_PointerOrReference)
Daniel Jasper3a1847e2013-07-01 09:34:09 +00001260 return Right.Tok.isLiteral() || Right.Type == TT_BlockComment ||
Daniel Jasper9322aae2013-03-20 09:53:18 +00001261 ((Right.Type != TT_PointerOrReference) &&
Daniel Jasper81d2d382013-04-01 17:13:26 +00001262 Right.isNot(tok::l_paren) && Style.PointerBindsToType &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001263 Left.Previous &&
1264 !Left.Previous->isOneOf(tok::l_paren, tok::coloncolon));
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001265 if (Right.is(tok::star) && Left.is(tok::l_paren))
1266 return false;
Nico Weber051860e2013-02-10 02:08:05 +00001267 if (Left.is(tok::l_square))
Daniel Jaspera07aa662013-10-22 15:30:28 +00001268 return Left.Type == TT_ArrayInitializerLSquare &&
1269 Right.isNot(tok::r_square);
Nico Weber051860e2013-02-10 02:08:05 +00001270 if (Right.is(tok::r_square))
Daniel Jaspera07aa662013-10-22 15:30:28 +00001271 return Right.MatchingParen &&
1272 Right.MatchingParen->Type == TT_ArrayInitializerLSquare;
Daniel Jasperec172262013-08-30 10:36:58 +00001273 if (Right.is(tok::l_square) && Right.Type != TT_ObjCMethodExpr &&
Daniel Jasper567dcf92013-09-05 09:29:45 +00001274 Right.Type != TT_LambdaLSquare && Left.isNot(tok::numeric_constant))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001275 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001276 if (Left.is(tok::colon))
1277 return Left.Type != TT_ObjCMethodExpr;
1278 if (Right.is(tok::colon))
Daniel Jasperab3ce592013-08-01 22:05:00 +00001279 return Right.Type != TT_ObjCMethodExpr && !Left.is(tok::question);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001280 if (Right.is(tok::l_paren)) {
Daniel Jaspere0fa4c52013-07-17 20:25:02 +00001281 if (Left.is(tok::r_paren) && Left.MatchingParen &&
1282 Left.MatchingParen->Previous &&
1283 Left.MatchingParen->Previous->is(tok::kw___attribute))
1284 return true;
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001285 return Line.Type == LT_ObjCDecl ||
Daniel Jasper34f3d052013-08-21 08:39:01 +00001286 Left.isOneOf(tok::kw_return, tok::kw_new, tok::kw_delete,
1287 tok::semi) ||
1288 (Style.SpaceAfterControlStatementKeyword &&
1289 Left.isOneOf(tok::kw_if, tok::kw_for, tok::kw_while, tok::kw_switch,
1290 tok::kw_catch));
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001291 }
Manuel Klimekb3987012013-05-29 14:47:47 +00001292 if (Left.is(tok::at) && Right.Tok.getObjCKeywordID() != tok::objc_not_keyword)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001293 return false;
1294 if (Left.is(tok::l_brace) && Right.is(tok::r_brace))
Daniel Jasper567dcf92013-09-05 09:29:45 +00001295 return !Left.Children.empty(); // No spaces in "{}".
Daniel Jasper2424eef2013-05-23 10:15:45 +00001296 if (Left.is(tok::l_brace) || Right.is(tok::r_brace))
Daniel Jasperb5dc3f42013-07-16 18:22:10 +00001297 return !Style.Cpp11BracedListStyle;
Daniel Jasper1bee0732013-05-23 18:05:18 +00001298 if (Right.Type == TT_UnaryOperator)
1299 return !Left.isOneOf(tok::l_paren, tok::l_square, tok::at) &&
1300 (Left.isNot(tok::colon) || Left.Type != TT_ObjCMethodExpr);
Daniel Jasperce933562013-05-23 21:35:49 +00001301 if (Left.isOneOf(tok::identifier, tok::greater, tok::r_square) &&
Manuel Klimek31e44f72013-09-04 08:20:47 +00001302 Right.is(tok::l_brace) && Right.getNextNonComment() &&
1303 Right.BlockKind != BK_Block)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001304 return false;
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001305 if (Left.is(tok::period) || Right.is(tok::period))
1306 return false;
Nico Weber861576b2013-06-26 00:15:19 +00001307 if (Left.Type == TT_BlockComment && Left.TokenText.endswith("=*/"))
1308 return false;
Alexander Kornienkodaa07e92013-09-10 13:41:43 +00001309 if (Right.is(tok::hash) && Left.is(tok::identifier) && Left.TokenText == "L")
1310 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001311 return true;
1312}
1313
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001314bool TokenAnnotator::spaceRequiredBefore(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001315 const FormatToken &Tok) {
1316 if (Tok.Tok.getIdentifierInfo() && Tok.Previous->Tok.getIdentifierInfo())
Daniel Jasper2b4c9242013-02-11 08:01:18 +00001317 return true; // Never ever merge two identifiers.
Daniel Jasper1d82b1a2013-10-11 19:45:02 +00001318 if (Tok.Previous->Type == TT_ImplicitStringLiteral)
1319 return Tok.WhitespaceRange.getBegin() != Tok.WhitespaceRange.getEnd();
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001320 if (Line.Type == LT_ObjCMethodDecl) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001321 if (Tok.Previous->Type == TT_ObjCMethodSpecifier)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001322 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001323 if (Tok.Previous->is(tok::r_paren) && Tok.is(tok::identifier))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001324 // Don't space between ')' and <id>
1325 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001326 }
1327 if (Line.Type == LT_ObjCProperty &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001328 (Tok.is(tok::equal) || Tok.Previous->is(tok::equal)))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001329 return false;
1330
Daniel Jasper2ca37412013-07-09 14:36:48 +00001331 if (Tok.Type == TT_TrailingReturnArrow ||
1332 Tok.Previous->Type == TT_TrailingReturnArrow)
1333 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001334 if (Tok.Previous->is(tok::comma))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001335 return true;
Daniel Jasper9c3c7b32013-02-28 13:40:17 +00001336 if (Tok.is(tok::comma))
1337 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001338 if (Tok.Type == TT_CtorInitializerColon || Tok.Type == TT_ObjCBlockLParen)
1339 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001340 if (Tok.Previous->Tok.is(tok::kw_operator))
Daniel Jasper2b4c9242013-02-11 08:01:18 +00001341 return false;
1342 if (Tok.Type == TT_OverloadedOperatorLParen)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001343 return false;
1344 if (Tok.is(tok::colon))
Manuel Klimekb3987012013-05-29 14:47:47 +00001345 return !Line.First->isOneOf(tok::kw_case, tok::kw_default) &&
Daniel Jasperab3ce592013-08-01 22:05:00 +00001346 Tok.getNextNonComment() != NULL && Tok.Type != TT_ObjCMethodExpr &&
1347 !Tok.Previous->is(tok::question);
Manuel Klimekb3987012013-05-29 14:47:47 +00001348 if (Tok.Previous->Type == TT_UnaryOperator ||
1349 Tok.Previous->Type == TT_CastRParen)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001350 return false;
Manuel Klimekb3987012013-05-29 14:47:47 +00001351 if (Tok.Previous->is(tok::greater) && Tok.is(tok::greater)) {
Daniel Jasper29f123b2013-02-08 15:28:42 +00001352 return Tok.Type == TT_TemplateCloser &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001353 Tok.Previous->Type == TT_TemplateCloser &&
Daniel Jasper29f123b2013-02-08 15:28:42 +00001354 Style.Standard != FormatStyle::LS_Cpp11;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001355 }
Alexander Kornienko54a38bd2013-03-20 16:41:56 +00001356 if (Tok.isOneOf(tok::arrowstar, tok::periodstar) ||
Manuel Klimekb3987012013-05-29 14:47:47 +00001357 Tok.Previous->isOneOf(tok::arrowstar, tok::periodstar))
Daniel Jasper9c3c7b32013-02-28 13:40:17 +00001358 return false;
Daniel Jasper9b4de852013-09-25 15:15:02 +00001359 if (!Style.SpaceBeforeAssignmentOperators &&
1360 Tok.getPrecedence() == prec::Assignment)
1361 return false;
Daniel Jasper1dc6f742013-08-07 16:29:23 +00001362 if ((Tok.Type == TT_BinaryOperator && !Tok.Previous->is(tok::l_paren)) ||
1363 Tok.Previous->Type == TT_BinaryOperator)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001364 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001365 if (Tok.Previous->Type == TT_TemplateCloser && Tok.is(tok::l_paren))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001366 return false;
Daniel Jaspera4dd9822013-08-28 08:24:04 +00001367 if (Tok.is(tok::less) && Tok.Previous->isNot(tok::l_paren) &&
1368 Line.First->is(tok::hash))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001369 return true;
1370 if (Tok.Type == TT_TrailingUnaryOperator)
1371 return false;
Manuel Klimekb3987012013-05-29 14:47:47 +00001372 return spaceRequiredBetween(Line, *Tok.Previous, Tok);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001373}
1374
Daniel Jasperebaa1712013-09-17 09:52:48 +00001375bool TokenAnnotator::mustBreakBefore(const AnnotatedLine &Line,
1376 const FormatToken &Right) {
1377 if (Right.is(tok::comment)) {
1378 return Right.NewlinesBefore > 0;
1379 } else if (Right.Previous->isTrailingComment() ||
1380 (Right.is(tok::string_literal) &&
1381 Right.Previous->is(tok::string_literal))) {
1382 return true;
1383 } else if (Right.Previous->IsUnterminatedLiteral) {
1384 return true;
1385 } else if (Right.is(tok::lessless) && Right.Next &&
1386 Right.Previous->is(tok::string_literal) &&
1387 Right.Next->is(tok::string_literal)) {
1388 return true;
1389 } else if (Right.Previous->ClosesTemplateDeclaration &&
1390 Right.Previous->MatchingParen &&
1391 Right.Previous->MatchingParen->BindingStrength == 1 &&
1392 Style.AlwaysBreakTemplateDeclarations) {
1393 // FIXME: Fix horrible hack of using BindingStrength to find top-level <>.
1394 return true;
1395 } else if (Right.Type == TT_CtorInitializerComma &&
Daniel Jasper19ccb122013-10-08 05:11:18 +00001396 Style.BreakConstructorInitializersBeforeComma &&
1397 !Style.ConstructorInitializerAllOnOneLineOrOnePerLine) {
Daniel Jasperebaa1712013-09-17 09:52:48 +00001398 return true;
1399 } else if (Right.Previous->BlockKind == BK_Block &&
Alexander Kornienko01fe9f92013-10-10 13:36:20 +00001400 Right.Previous->isNot(tok::r_brace) && Right.isNot(tok::r_brace)) {
Daniel Jasperebaa1712013-09-17 09:52:48 +00001401 return true;
1402 } else if (Right.is(tok::l_brace) && (Right.BlockKind == BK_Block)) {
1403 return Style.BreakBeforeBraces == FormatStyle::BS_Allman;
1404 }
1405 return false;
1406}
1407
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001408bool TokenAnnotator::canBreakBefore(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001409 const FormatToken &Right) {
1410 const FormatToken &Left = *Right.Previous;
Daniel Jasper6561f6a2013-07-09 07:43:55 +00001411 if (Right.Type == TT_StartOfName || Right.is(tok::kw_operator))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001412 return true;
Nico Weberf2ff8122013-05-26 05:39:26 +00001413 if (Right.is(tok::colon) &&
1414 (Right.Type == TT_ObjCDictLiteral || Right.Type == TT_ObjCMethodExpr))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001415 return false;
Nico Weberf2ff8122013-05-26 05:39:26 +00001416 if (Left.is(tok::colon) &&
1417 (Left.Type == TT_ObjCDictLiteral || Left.Type == TT_ObjCMethodExpr))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001418 return true;
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001419 if (Right.Type == TT_ObjCSelectorName)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001420 return true;
Daniel Jasperaa9e7b12013-08-01 13:46:58 +00001421 if (Left.is(tok::r_paren) && Line.Type == LT_ObjCProperty)
1422 return true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001423 if (Left.ClosesTemplateDeclaration)
1424 return true;
Daniel Jasperab3ce592013-08-01 22:05:00 +00001425 if ((Right.Type == TT_ConditionalExpr &&
1426 !(Right.is(tok::colon) && Left.is(tok::question))) ||
1427 Right.is(tok::question))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001428 return true;
Daniel Jasper6cabab42013-02-14 08:42:54 +00001429 if (Right.Type == TT_RangeBasedForLoopColon ||
Daniel Jasperc476ea92013-08-28 07:27:35 +00001430 Right.Type == TT_OverloadedOperatorLParen ||
1431 Right.Type == TT_OverloadedOperator)
Daniel Jasper6cabab42013-02-14 08:42:54 +00001432 return false;
Daniel Jasperc194c952013-05-06 06:45:09 +00001433 if (Left.Type == TT_RangeBasedForLoopColon)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001434 return true;
Daniel Jasper7d812812013-02-21 15:00:29 +00001435 if (Right.Type == TT_RangeBasedForLoopColon)
1436 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001437 if (Left.Type == TT_PointerOrReference || Left.Type == TT_TemplateCloser ||
1438 Left.Type == TT_UnaryOperator || Left.Type == TT_ConditionalExpr ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001439 Left.isOneOf(tok::question, tok::kw_operator))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001440 return false;
1441 if (Left.is(tok::equal) && Line.Type == LT_VirtualFunctionDecl)
1442 return false;
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001443 if (Left.Previous) {
1444 if (Left.is(tok::l_paren) && Right.is(tok::l_paren) &&
1445 Left.Previous->is(tok::kw___attribute))
1446 return false;
Daniel Jasper2ca37412013-07-09 14:36:48 +00001447 if (Left.is(tok::l_paren) && (Left.Previous->Type == TT_BinaryOperator ||
Daniel Jasper5e2169f2013-07-18 14:46:07 +00001448 Left.Previous->Type == TT_CastRParen))
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001449 return false;
1450 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001451
Daniel Jasper65d2c382013-06-06 16:08:57 +00001452 if (Right.isTrailingComment())
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001453 // We rely on MustBreakBefore being set correctly here as we should not
1454 // change the "binding" behavior of a comment.
1455 return false;
1456
Daniel Jasper567dcf92013-09-05 09:29:45 +00001457 if (Right.is(tok::r_paren) || Right.Type == TT_TemplateCloser)
1458 return false;
1459
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001460 // We only break before r_brace if there was a corresponding break before
1461 // the l_brace, which is tracked by BreakBeforeClosingBrace.
Daniel Jasper567dcf92013-09-05 09:29:45 +00001462 if (Right.is(tok::r_brace))
1463 return Right.MatchingParen && Right.MatchingParen->BlockKind == BK_Block;
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001464
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001465 // Allow breaking after a trailing 'const', e.g. after a method declaration,
1466 // unless it is follow by ';', '{' or '='.
Manuel Klimekb3987012013-05-29 14:47:47 +00001467 if (Left.is(tok::kw_const) && Left.Previous != NULL &&
1468 Left.Previous->is(tok::r_paren))
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001469 return !Right.isOneOf(tok::l_brace, tok::semi, tok::equal);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001470
Daniel Jasper8ef19a22013-03-14 09:50:46 +00001471 if (Right.is(tok::kw___attribute))
1472 return true;
1473
Daniel Jasper3a204412013-02-23 07:46:38 +00001474 if (Left.is(tok::identifier) && Right.is(tok::string_literal))
1475 return true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +00001476
1477 if (Left.Type == TT_CtorInitializerComma &&
1478 Style.BreakConstructorInitializersBeforeComma)
1479 return false;
Daniel Jasper19ccb122013-10-08 05:11:18 +00001480 if (Right.Type == TT_CtorInitializerComma &&
1481 Style.BreakConstructorInitializersBeforeComma)
1482 return true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +00001483 if (Right.isBinaryOperator() && Style.BreakBeforeBinaryOperators)
1484 return true;
Daniel Jasper26356cc2013-09-17 08:15:46 +00001485 if (Left.is(tok::greater) && Right.is(tok::greater) &&
1486 Left.Type != TT_TemplateCloser)
1487 return false;
Daniel Jaspera07aa662013-10-22 15:30:28 +00001488 if (Left.Type == TT_ArrayInitializerLSquare)
1489 return true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +00001490 return (Left.isBinaryOperator() && Left.isNot(tok::lessless) &&
1491 !Style.BreakBeforeBinaryOperators) ||
Daniel Jasper6b119d62013-04-05 17:22:09 +00001492 Left.isOneOf(tok::comma, tok::coloncolon, tok::semi, tok::l_brace,
1493 tok::kw_class, tok::kw_struct) ||
Daniel Jaspera07aa662013-10-22 15:30:28 +00001494 Right.isOneOf(tok::lessless, tok::arrow, tok::period, tok::colon,
1495 tok::l_square, tok::at) ||
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001496 (Left.is(tok::r_paren) &&
Daniel Jaspere033e872013-05-21 09:16:31 +00001497 Right.isOneOf(tok::identifier, tok::kw_const, tok::kw___attribute)) ||
Daniel Jaspera07aa662013-10-22 15:30:28 +00001498 (Left.is(tok::l_paren) && !Right.is(tok::r_paren));
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001499}
1500
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001501void TokenAnnotator::printDebugInfo(const AnnotatedLine &Line) {
1502 llvm::errs() << "AnnotatedTokens:\n";
Manuel Klimekb3987012013-05-29 14:47:47 +00001503 const FormatToken *Tok = Line.First;
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001504 while (Tok) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001505 llvm::errs() << " M=" << Tok->MustBreakBefore
Daniel Jasperc7bd68f2013-07-10 14:02:49 +00001506 << " C=" << Tok->CanBreakBefore << " T=" << Tok->Type
1507 << " S=" << Tok->SpacesRequiredBefore
1508 << " P=" << Tok->SplitPenalty << " Name=" << Tok->Tok.getName()
Manuel Klimekae76f7f2013-10-11 21:25:45 +00001509 << " L=" << Tok->TotalLength << " PPK=" << Tok->PackingKind
1510 << " FakeLParens=";
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001511 for (unsigned i = 0, e = Tok->FakeLParens.size(); i != e; ++i)
1512 llvm::errs() << Tok->FakeLParens[i] << "/";
1513 llvm::errs() << " FakeRParens=" << Tok->FakeRParens << "\n";
Manuel Klimekae76f7f2013-10-11 21:25:45 +00001514 if (Tok->Next == NULL)
1515 assert(Tok == Line.Last);
Manuel Klimekb3987012013-05-29 14:47:47 +00001516 Tok = Tok->Next;
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001517 }
1518 llvm::errs() << "----\n";
1519}
1520
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001521} // namespace format
1522} // namespace clang