blob: e3cd2105e63756470b17f699cec2748233aa525a [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 Jasper567dcf92013-09-05 09:29:45 +0000185 Contexts.back().CanBeExpression && Left->Type != TT_LambdaLSquare &&
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)) {
Daniel Jasperac2c9742013-09-05 10:04:31 +0000204 if (CurrentToken->Next && CurrentToken->Next->is(tok::l_paren) &&
205 Left->Type == TT_ObjCMethodExpr) {
Nico Webere8a97982013-02-06 06:20:11 +0000206 // An ObjC method call is rarely followed by an open parenthesis.
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000207 // FIXME: Do we incorrectly label ":" with this?
208 StartsObjCMethodExpr = false;
209 Left->Type = TT_Unknown;
210 }
Daniel Jasper01786732013-02-04 07:21:18 +0000211 if (StartsObjCMethodExpr) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000212 CurrentToken->Type = TT_ObjCMethodExpr;
Nico Webere8a97982013-02-06 06:20:11 +0000213 // determineStarAmpUsage() thinks that '*' '[' is allocating an
214 // array of pointers, but if '[' starts a selector then '*' is a
215 // binary operator.
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +0000216 if (Parent != NULL && Parent->Type == TT_PointerOrReference)
Nico Weber4ed7f3e2013-02-06 16:54:35 +0000217 Parent->Type = TT_BinaryOperator;
Nico Weber051860e2013-02-10 02:08:05 +0000218 } else if (StartsObjCArrayLiteral) {
219 CurrentToken->Type = TT_ObjCArrayLiteral;
Daniel Jasper01786732013-02-04 07:21:18 +0000220 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000221 Left->MatchingParen = CurrentToken;
222 CurrentToken->MatchingParen = Left;
Daniel Jasper4e778092013-02-06 10:05:46 +0000223 if (Contexts.back().FirstObjCSelectorName != NULL)
224 Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
225 Contexts.back().LongestObjCSelectorName;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000226 next();
227 return true;
228 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000229 if (CurrentToken->isOneOf(tok::r_paren, tok::r_brace))
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000230 return false;
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000231 updateParameterCount(Left, CurrentToken);
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000232 if (!consumeToken())
233 return false;
234 }
235 return false;
236 }
237
238 bool parseBrace() {
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000239 if (CurrentToken != NULL) {
240 ScopedContextCreator ContextCreator(*this, tok::l_brace, 1);
Manuel Klimekb3987012013-05-29 14:47:47 +0000241 FormatToken *Left = CurrentToken->Previous;
Nico Weberf2ff8122013-05-26 05:39:26 +0000242
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000243 FormatToken *Parent = Left->getPreviousNonComment();
Nico Weberf2ff8122013-05-26 05:39:26 +0000244 bool StartsObjCDictLiteral = Parent && Parent->is(tok::at);
245 if (StartsObjCDictLiteral) {
246 Contexts.back().ColonIsObjCDictLiteral = true;
247 Left->Type = TT_ObjCDictLiteral;
248 }
249
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000250 while (CurrentToken != NULL) {
251 if (CurrentToken->is(tok::r_brace)) {
Nico Weberf2ff8122013-05-26 05:39:26 +0000252 if (StartsObjCDictLiteral)
253 CurrentToken->Type = TT_ObjCDictLiteral;
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000254 Left->MatchingParen = CurrentToken;
255 CurrentToken->MatchingParen = Left;
256 next();
257 return true;
258 }
259 if (CurrentToken->isOneOf(tok::r_paren, tok::r_square))
260 return false;
261 updateParameterCount(Left, CurrentToken);
262 if (!consumeToken())
263 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000264 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000265 }
Daniel Jasper53e72cd2013-05-06 08:27:33 +0000266 // No closing "}" found, this probably starts a definition.
267 Line.StartsDefinition = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000268 return true;
269 }
Daniel Jasperc4615b72013-02-20 12:56:39 +0000270
Manuel Klimekb3987012013-05-29 14:47:47 +0000271 void updateParameterCount(FormatToken *Left, FormatToken *Current) {
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000272 if (Current->is(tok::comma)) {
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000273 ++Left->ParameterCount;
Daniel Jasperd4a03db2013-08-22 15:00:41 +0000274 if (!Left->Role)
275 Left->Role.reset(new CommaSeparatedList(Style));
276 Left->Role->CommaFound(Current);
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000277 } else if (Left->ParameterCount == 0 && Current->isNot(tok::comment)) {
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000278 Left->ParameterCount = 1;
Daniel Jasperc7bd68f2013-07-10 14:02:49 +0000279 }
Daniel Jasper9fc56f22013-02-14 15:01:34 +0000280 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000281
282 bool parseConditional() {
283 while (CurrentToken != NULL) {
284 if (CurrentToken->is(tok::colon)) {
285 CurrentToken->Type = TT_ConditionalExpr;
286 next();
287 return true;
288 }
289 if (!consumeToken())
290 return false;
291 }
292 return false;
293 }
294
295 bool parseTemplateDeclaration() {
296 if (CurrentToken != NULL && CurrentToken->is(tok::less)) {
297 CurrentToken->Type = TT_TemplateOpener;
298 next();
299 if (!parseAngle())
300 return false;
Daniel Jasper34511fb2013-02-19 17:14:38 +0000301 if (CurrentToken != NULL)
Manuel Klimekb3987012013-05-29 14:47:47 +0000302 CurrentToken->Previous->ClosesTemplateDeclaration = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000303 return true;
304 }
305 return false;
306 }
307
308 bool consumeToken() {
Manuel Klimekb3987012013-05-29 14:47:47 +0000309 FormatToken *Tok = CurrentToken;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000310 next();
Manuel Klimekb3987012013-05-29 14:47:47 +0000311 switch (Tok->Tok.getKind()) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000312 case tok::plus:
313 case tok::minus:
Manuel Klimekb3987012013-05-29 14:47:47 +0000314 if (Tok->Previous == NULL && Line.MustBeDeclaration)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000315 Tok->Type = TT_ObjCMethodSpecifier;
316 break;
317 case tok::colon:
Manuel Klimekb3987012013-05-29 14:47:47 +0000318 if (Tok->Previous == NULL)
Daniel Jaspercf6d76a2013-03-18 12:50:26 +0000319 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000320 // Colons from ?: are handled in parseConditional().
Manuel Klimekb3987012013-05-29 14:47:47 +0000321 if (Tok->Previous->is(tok::r_paren) && Contexts.size() == 1) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000322 Tok->Type = TT_CtorInitializerColon;
Nico Weberf2ff8122013-05-26 05:39:26 +0000323 } else if (Contexts.back().ColonIsObjCDictLiteral) {
324 Tok->Type = TT_ObjCDictLiteral;
Daniel Jasper4e778092013-02-06 10:05:46 +0000325 } else if (Contexts.back().ColonIsObjCMethodExpr ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000326 Line.First->Type == TT_ObjCMethodSpecifier) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000327 Tok->Type = TT_ObjCMethodExpr;
Manuel Klimekb3987012013-05-29 14:47:47 +0000328 Tok->Previous->Type = TT_ObjCSelectorName;
Alexander Kornienko00895102013-06-05 14:09:10 +0000329 if (Tok->Previous->CodePointCount >
330 Contexts.back().LongestObjCSelectorName) {
331 Contexts.back().LongestObjCSelectorName =
332 Tok->Previous->CodePointCount;
333 }
Daniel Jasper4e778092013-02-06 10:05:46 +0000334 if (Contexts.back().FirstObjCSelectorName == NULL)
Manuel Klimekb3987012013-05-29 14:47:47 +0000335 Contexts.back().FirstObjCSelectorName = Tok->Previous;
Daniel Jasper4e778092013-02-06 10:05:46 +0000336 } else if (Contexts.back().ColonIsForRangeExpr) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000337 Tok->Type = TT_RangeBasedForLoopColon;
Daniel Jasper6cabab42013-02-14 08:42:54 +0000338 } else if (Contexts.size() == 1) {
339 Tok->Type = TT_InheritanceColon;
Daniel Jasper923ebef2013-03-14 13:45:21 +0000340 } else if (Contexts.back().ContextKind == tok::l_paren) {
341 Tok->Type = TT_InlineASMColon;
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000342 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000343 break;
344 case tok::kw_if:
345 case tok::kw_while:
346 if (CurrentToken != NULL && CurrentToken->is(tok::l_paren)) {
347 next();
Nico Weber27268772013-06-26 00:30:14 +0000348 if (!parseParens(/*LookForDecls=*/true))
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000349 return false;
350 }
351 break;
352 case tok::kw_for:
Daniel Jasper4e778092013-02-06 10:05:46 +0000353 Contexts.back().ColonIsForRangeExpr = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000354 next();
355 if (!parseParens())
356 return false;
357 break;
358 case tok::l_paren:
359 if (!parseParens())
360 return false;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000361 if (Line.MustBeDeclaration && NameFound && !Contexts.back().IsExpression)
Daniel Jasper3c08a812013-02-24 18:54:32 +0000362 Line.MightBeFunctionDecl = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000363 break;
364 case tok::l_square:
365 if (!parseSquare())
366 return false;
367 break;
368 case tok::l_brace:
369 if (!parseBrace())
370 return false;
371 break;
372 case tok::less:
Daniel Jasper0236dd02013-07-30 22:37:19 +0000373 if (Tok->Previous && !Tok->Previous->Tok.isLiteral() && parseAngle())
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000374 Tok->Type = TT_TemplateOpener;
375 else {
376 Tok->Type = TT_BinaryOperator;
377 CurrentToken = Tok;
378 next();
379 }
380 break;
381 case tok::r_paren:
382 case tok::r_square:
383 return false;
384 case tok::r_brace:
385 // Lines can start with '}'.
Manuel Klimekb3987012013-05-29 14:47:47 +0000386 if (Tok->Previous != NULL)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000387 return false;
388 break;
389 case tok::greater:
390 Tok->Type = TT_BinaryOperator;
391 break;
392 case tok::kw_operator:
Daniel Jasper174f60f2013-09-02 09:20:39 +0000393 while (CurrentToken &&
394 !CurrentToken->isOneOf(tok::l_paren, tok::semi, tok::r_paren)) {
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000395 if (CurrentToken->isOneOf(tok::star, tok::amp))
Daniel Jasper2b4c9242013-02-11 08:01:18 +0000396 CurrentToken->Type = TT_PointerOrReference;
397 consumeToken();
Daniel Jasper174f60f2013-09-02 09:20:39 +0000398 if (CurrentToken && CurrentToken->Previous->Type == TT_BinaryOperator)
Daniel Jasperc476ea92013-08-28 07:27:35 +0000399 CurrentToken->Previous->Type = TT_OverloadedOperator;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000400 }
Daniel Jasper6ea933c2013-05-10 07:59:58 +0000401 if (CurrentToken) {
Daniel Jasper2b4c9242013-02-11 08:01:18 +0000402 CurrentToken->Type = TT_OverloadedOperatorLParen;
Manuel Klimekb3987012013-05-29 14:47:47 +0000403 if (CurrentToken->Previous->Type == TT_BinaryOperator)
404 CurrentToken->Previous->Type = TT_OverloadedOperator;
Daniel Jasper6ea933c2013-05-10 07:59:58 +0000405 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000406 break;
407 case tok::question:
408 parseConditional();
409 break;
410 case tok::kw_template:
411 parseTemplateDeclaration();
412 break;
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000413 case tok::identifier:
Manuel Klimekb3987012013-05-29 14:47:47 +0000414 if (Line.First->is(tok::kw_for) &&
415 Tok->Tok.getIdentifierInfo() == &Ident_in)
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000416 Tok->Type = TT_ObjCForIn;
417 break;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000418 case tok::comma:
419 if (Contexts.back().FirstStartOfName)
420 Contexts.back().FirstStartOfName->PartOfMultiVariableDeclStmt = true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000421 if (Contexts.back().InCtorInitializer)
422 Tok->Type = TT_CtorInitializerComma;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000423 break;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000424 default:
425 break;
426 }
427 return true;
428 }
429
430 void parseIncludeDirective() {
431 next();
432 if (CurrentToken != NULL && CurrentToken->is(tok::less)) {
433 next();
434 while (CurrentToken != NULL) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000435 if (CurrentToken->isNot(tok::comment) || CurrentToken->Next)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000436 CurrentToken->Type = TT_ImplicitStringLiteral;
437 next();
438 }
439 } else {
440 while (CurrentToken != NULL) {
Daniel Jasper3a204412013-02-23 07:46:38 +0000441 if (CurrentToken->is(tok::string_literal))
442 // Mark these string literals as "implicit" literals, too, so that
443 // they are not split or line-wrapped.
444 CurrentToken->Type = TT_ImplicitStringLiteral;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000445 next();
446 }
447 }
448 }
449
450 void parseWarningOrError() {
451 next();
452 // We still want to format the whitespace left of the first token of the
453 // warning or error.
454 next();
455 while (CurrentToken != NULL) {
456 CurrentToken->Type = TT_ImplicitStringLiteral;
457 next();
458 }
459 }
460
461 void parsePreprocessorDirective() {
462 next();
463 if (CurrentToken == NULL)
464 return;
465 // Hashes in the middle of a line can lead to any strange token
466 // sequence.
Manuel Klimekb3987012013-05-29 14:47:47 +0000467 if (CurrentToken->Tok.getIdentifierInfo() == NULL)
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000468 return;
Manuel Klimekb3987012013-05-29 14:47:47 +0000469 switch (CurrentToken->Tok.getIdentifierInfo()->getPPKeywordID()) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000470 case tok::pp_include:
471 case tok::pp_import:
472 parseIncludeDirective();
473 break;
474 case tok::pp_error:
475 case tok::pp_warning:
476 parseWarningOrError();
477 break;
Daniel Jasperaae7bad2013-04-23 13:54:04 +0000478 case tok::pp_if:
479 case tok::pp_elif:
480 parseLine();
481 break;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000482 default:
483 break;
484 }
Daniel Jasper5b7e7b02013-02-05 09:34:14 +0000485 while (CurrentToken != NULL)
486 next();
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000487 }
488
Nico Weber95e8e462013-02-12 16:17:07 +0000489public:
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000490 LineType parseLine() {
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000491 if (CurrentToken->is(tok::hash)) {
492 parsePreprocessorDirective();
493 return LT_PreprocessorDirective;
494 }
495 while (CurrentToken != NULL) {
496 if (CurrentToken->is(tok::kw_virtual))
497 KeywordVirtualFound = true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000498 if (!consumeToken())
499 return LT_Invalid;
500 }
501 if (KeywordVirtualFound)
502 return LT_VirtualFunctionDecl;
503
Manuel Klimekb3987012013-05-29 14:47:47 +0000504 if (Line.First->Type == TT_ObjCMethodSpecifier) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000505 if (Contexts.back().FirstObjCSelectorName != NULL)
506 Contexts.back().FirstObjCSelectorName->LongestObjCSelectorName =
507 Contexts.back().LongestObjCSelectorName;
Daniel Jasper63d7ced2013-02-05 10:07:47 +0000508 return LT_ObjCMethodDecl;
509 }
510
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000511 return LT_Other;
512 }
513
Nico Weber95e8e462013-02-12 16:17:07 +0000514private:
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000515 void next() {
Daniel Jasper01786732013-02-04 07:21:18 +0000516 if (CurrentToken != NULL) {
517 determineTokenType(*CurrentToken);
Daniel Jasper4e778092013-02-06 10:05:46 +0000518 CurrentToken->BindingStrength = Contexts.back().BindingStrength;
Daniel Jasper01786732013-02-04 07:21:18 +0000519 }
520
Manuel Klimekb3987012013-05-29 14:47:47 +0000521 if (CurrentToken != NULL)
522 CurrentToken = CurrentToken->Next;
Daniel Jasperd0f349b2013-02-18 12:44:35 +0000523
524 // Reset token type in case we have already looked at it and then recovered
525 // from an error (e.g. failure to find the matching >).
Daniel Jasper567dcf92013-09-05 09:29:45 +0000526 if (CurrentToken != NULL && CurrentToken->Type != TT_LambdaLSquare)
Daniel Jasperd0f349b2013-02-18 12:44:35 +0000527 CurrentToken->Type = TT_Unknown;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000528 }
529
Daniel Jasper4e778092013-02-06 10:05:46 +0000530 /// \brief A struct to hold information valid in a specific context, e.g.
531 /// a pair of parenthesis.
532 struct Context {
Daniel Jasper923ebef2013-03-14 13:45:21 +0000533 Context(tok::TokenKind ContextKind, unsigned BindingStrength,
534 bool IsExpression)
535 : ContextKind(ContextKind), BindingStrength(BindingStrength),
536 LongestObjCSelectorName(0), ColonIsForRangeExpr(false),
Nico Weberf2ff8122013-05-26 05:39:26 +0000537 ColonIsObjCDictLiteral(false), ColonIsObjCMethodExpr(false),
538 FirstObjCSelectorName(NULL), FirstStartOfName(NULL),
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000539 IsExpression(IsExpression), CanBeExpression(true),
540 InCtorInitializer(false) {}
Daniel Jasper01786732013-02-04 07:21:18 +0000541
Daniel Jasper923ebef2013-03-14 13:45:21 +0000542 tok::TokenKind ContextKind;
Daniel Jasper4e778092013-02-06 10:05:46 +0000543 unsigned BindingStrength;
544 unsigned LongestObjCSelectorName;
545 bool ColonIsForRangeExpr;
Nico Weberf2ff8122013-05-26 05:39:26 +0000546 bool ColonIsObjCDictLiteral;
Daniel Jasper4e778092013-02-06 10:05:46 +0000547 bool ColonIsObjCMethodExpr;
Manuel Klimekb3987012013-05-29 14:47:47 +0000548 FormatToken *FirstObjCSelectorName;
549 FormatToken *FirstStartOfName;
Daniel Jasper4e778092013-02-06 10:05:46 +0000550 bool IsExpression;
Daniel Jasper6f21a982013-03-13 07:49:51 +0000551 bool CanBeExpression;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000552 bool InCtorInitializer;
Daniel Jasper4e778092013-02-06 10:05:46 +0000553 };
554
555 /// \brief Puts a new \c Context onto the stack \c Contexts for the lifetime
556 /// of each instance.
557 struct ScopedContextCreator {
558 AnnotatingParser &P;
559
Daniel Jasper923ebef2013-03-14 13:45:21 +0000560 ScopedContextCreator(AnnotatingParser &P, tok::TokenKind ContextKind,
561 unsigned Increase)
562 : P(P) {
Daniel Jasper2a409b62013-07-08 14:34:09 +0000563 P.Contexts.push_back(Context(ContextKind,
564 P.Contexts.back().BindingStrength + Increase,
565 P.Contexts.back().IsExpression));
Daniel Jasper4e778092013-02-06 10:05:46 +0000566 }
567
568 ~ScopedContextCreator() { P.Contexts.pop_back(); }
569 };
Daniel Jasper01786732013-02-04 07:21:18 +0000570
Manuel Klimekb3987012013-05-29 14:47:47 +0000571 void determineTokenType(FormatToken &Current) {
572 if (Current.getPrecedence() == prec::Assignment &&
Daniel Jasper53352602013-08-12 12:16:34 +0000573 !Line.First->isOneOf(tok::kw_template, tok::kw_using) &&
Manuel Klimekb3987012013-05-29 14:47:47 +0000574 (!Current.Previous || Current.Previous->isNot(tok::kw_operator))) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000575 Contexts.back().IsExpression = true;
Manuel Klimekb3987012013-05-29 14:47:47 +0000576 for (FormatToken *Previous = Current.Previous;
Nico Weber95e8e462013-02-12 16:17:07 +0000577 Previous && Previous->isNot(tok::comma);
Manuel Klimekb3987012013-05-29 14:47:47 +0000578 Previous = Previous->Previous) {
Daniel Jasper9c65b062013-02-27 11:43:50 +0000579 if (Previous->is(tok::r_square))
580 Previous = Previous->MatchingParen;
Daniel Jasper01786732013-02-04 07:21:18 +0000581 if (Previous->Type == TT_BinaryOperator &&
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000582 Previous->isOneOf(tok::star, tok::amp)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000583 Previous->Type = TT_PointerOrReference;
584 }
Daniel Jasper01786732013-02-04 07:21:18 +0000585 }
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000586 } else if (Current.isOneOf(tok::kw_return, tok::kw_throw) ||
Nico Weber95e8e462013-02-12 16:17:07 +0000587 (Current.is(tok::l_paren) && !Line.MustBeDeclaration &&
Daniel Jasper378d93d2013-05-13 07:14:40 +0000588 !Line.InPPDirective &&
Manuel Klimek2530fd52013-08-12 03:51:17 +0000589 (!Current.Previous ||
Daniel Jasper53352602013-08-12 12:16:34 +0000590 !Current.Previous->isOneOf(tok::kw_for, tok::kw_catch)))) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000591 Contexts.back().IsExpression = true;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000592 } else if (Current.isOneOf(tok::r_paren, tok::greater, tok::comma)) {
Manuel Klimekb3987012013-05-29 14:47:47 +0000593 for (FormatToken *Previous = Current.Previous;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000594 Previous && Previous->isOneOf(tok::star, tok::amp);
Manuel Klimekb3987012013-05-29 14:47:47 +0000595 Previous = Previous->Previous)
Nico Weber95e8e462013-02-12 16:17:07 +0000596 Previous->Type = TT_PointerOrReference;
Manuel Klimekb3987012013-05-29 14:47:47 +0000597 } else if (Current.Previous &&
598 Current.Previous->Type == TT_CtorInitializerColon) {
Daniel Jasperd0f349b2013-02-18 12:44:35 +0000599 Contexts.back().IsExpression = true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +0000600 Contexts.back().InCtorInitializer = true;
Daniel Jasper6f21a982013-03-13 07:49:51 +0000601 } else if (Current.is(tok::kw_new)) {
602 Contexts.back().CanBeExpression = false;
Daniel Jasper16a69ef2013-05-03 14:41:24 +0000603 } else if (Current.is(tok::semi)) {
604 // This should be the condition or increment in a for-loop.
605 Contexts.back().IsExpression = true;
Nico Weber95e8e462013-02-12 16:17:07 +0000606 }
Daniel Jasper01786732013-02-04 07:21:18 +0000607
608 if (Current.Type == TT_Unknown) {
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000609 if (isStartOfName(Current)) {
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000610 Contexts.back().FirstStartOfName = &Current;
Daniel Jasper3c08a812013-02-24 18:54:32 +0000611 Current.Type = TT_StartOfName;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000612 NameFound = true;
Daniel Jasper2ca37412013-07-09 14:36:48 +0000613 } else if (Current.is(tok::kw_auto)) {
614 AutoFound = true;
Daniel Jasper3262f4c2013-07-11 14:33:06 +0000615 } else if (Current.is(tok::arrow) && AutoFound &&
616 Line.MustBeDeclaration) {
Daniel Jasper2ca37412013-07-09 14:36:48 +0000617 Current.Type = TT_TrailingReturnArrow;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000618 } else if (Current.isOneOf(tok::star, tok::amp, tok::ampamp)) {
Daniel Jasper4e778092013-02-06 10:05:46 +0000619 Current.Type =
Daniel Jasperd6104f62013-07-05 13:30:40 +0000620 determineStarAmpUsage(Current, Contexts.back().CanBeExpression &&
621 Contexts.back().IsExpression);
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000622 } else if (Current.isOneOf(tok::minus, tok::plus, tok::caret)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000623 Current.Type = determinePlusMinusCaretUsage(Current);
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000624 } else if (Current.isOneOf(tok::minusminus, tok::plusplus)) {
Daniel Jasper01786732013-02-04 07:21:18 +0000625 Current.Type = determineIncrementUsage(Current);
626 } else if (Current.is(tok::exclaim)) {
627 Current.Type = TT_UnaryOperator;
Manuel Klimek31e44f72013-09-04 08:20:47 +0000628 } else if (Current.isBinaryOperator() &&
629 (!Current.Previous ||
630 Current.Previous->isNot(tok::l_square))) {
Daniel Jasper01786732013-02-04 07:21:18 +0000631 Current.Type = TT_BinaryOperator;
632 } else if (Current.is(tok::comment)) {
Alexander Kornienko00895102013-06-05 14:09:10 +0000633 if (Current.TokenText.startswith("//"))
Daniel Jasper01786732013-02-04 07:21:18 +0000634 Current.Type = TT_LineComment;
635 else
636 Current.Type = TT_BlockComment;
Nico Weber37d69312013-02-13 04:13:13 +0000637 } else if (Current.is(tok::r_paren)) {
Daniel Jasperb8b42952013-05-31 16:14:28 +0000638 FormatToken *LeftOfParens = NULL;
639 if (Current.MatchingParen)
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000640 LeftOfParens = Current.MatchingParen->getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000641 bool IsCast = false;
642 bool ParensAreEmpty = Current.Previous == Current.MatchingParen;
643 bool ParensAreType = !Current.Previous ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000644 Current.Previous->Type == TT_PointerOrReference ||
Daniel Jasperb8b42952013-05-31 16:14:28 +0000645 Current.Previous->Type == TT_TemplateCloser ||
646 isSimpleTypeSpecifier(*Current.Previous);
Nico Weber37d69312013-02-13 04:13:13 +0000647 bool ParensCouldEndDecl =
Manuel Klimekb3987012013-05-29 14:47:47 +0000648 Current.Next &&
649 Current.Next->isOneOf(tok::equal, tok::semi, tok::l_brace);
Daniel Jasper6a365aa2013-03-13 17:13:53 +0000650 bool IsSizeOfOrAlignOf =
Daniel Jasperb8b42952013-05-31 16:14:28 +0000651 LeftOfParens &&
652 LeftOfParens->isOneOf(tok::kw_sizeof, tok::kw_alignof);
653 if (ParensAreType && !ParensCouldEndDecl && !IsSizeOfOrAlignOf &&
Daniel Jasper0c368782013-07-15 15:04:42 +0000654 (Contexts.back().IsExpression ||
655 (Current.Next && Current.Next->isBinaryOperator())))
Daniel Jasperb8b42952013-05-31 16:14:28 +0000656 IsCast = true;
Daniel Jasper2a409b62013-07-08 14:34:09 +0000657 if (Current.Next && Current.Next->isNot(tok::string_literal) &&
Daniel Jasperb8b42952013-05-31 16:14:28 +0000658 (Current.Next->Tok.isLiteral() ||
659 Current.Next->isOneOf(tok::kw_sizeof, tok::kw_alignof)))
660 IsCast = true;
661 // If there is an identifier after the (), it is likely a cast, unless
662 // there is also an identifier before the ().
Daniel Jasperff1a2e52013-06-06 08:20:20 +0000663 if (LeftOfParens && (LeftOfParens->Tok.getIdentifierInfo() == NULL ||
664 LeftOfParens->is(tok::kw_return)) &&
Daniel Jasper526df0f2013-07-08 14:58:01 +0000665 LeftOfParens->Type != TT_OverloadedOperator &&
Nico Weber465e8612013-06-25 00:55:57 +0000666 LeftOfParens->Type != TT_TemplateCloser && Current.Next &&
667 Current.Next->is(tok::identifier))
Daniel Jasperb8b42952013-05-31 16:14:28 +0000668 IsCast = true;
669 if (IsCast && !ParensAreEmpty)
Nico Weber37d69312013-02-13 04:13:13 +0000670 Current.Type = TT_CastRParen;
Manuel Klimekb3987012013-05-29 14:47:47 +0000671 } else if (Current.is(tok::at) && Current.Next) {
672 switch (Current.Next->Tok.getObjCKeywordID()) {
Daniel Jasper01786732013-02-04 07:21:18 +0000673 case tok::objc_interface:
674 case tok::objc_implementation:
675 case tok::objc_protocol:
676 Current.Type = TT_ObjCDecl;
677 break;
678 case tok::objc_property:
679 Current.Type = TT_ObjCProperty;
680 break;
681 default:
682 break;
683 }
Daniel Jasper5ad390d2013-05-28 11:30:49 +0000684 } else if (Current.is(tok::period)) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000685 FormatToken *PreviousNoComment = Current.getPreviousNonComment();
Daniel Jasper5ad390d2013-05-28 11:30:49 +0000686 if (PreviousNoComment &&
687 PreviousNoComment->isOneOf(tok::comma, tok::l_brace))
688 Current.Type = TT_DesignatedInitializerPeriod;
Daniel Jasper01786732013-02-04 07:21:18 +0000689 }
690 }
691 }
692
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000693 /// \brief Take a guess at whether \p Tok starts a name of a function or
694 /// variable declaration.
695 ///
696 /// This is a heuristic based on whether \p Tok is an identifier following
697 /// something that is likely a type.
698 bool isStartOfName(const FormatToken &Tok) {
699 if (Tok.isNot(tok::identifier) || Tok.Previous == NULL)
700 return false;
701
702 // Skip "const" as it does not have an influence on whether this is a name.
703 FormatToken *PreviousNotConst = Tok.Previous;
704 while (PreviousNotConst != NULL && PreviousNotConst->is(tok::kw_const))
705 PreviousNotConst = PreviousNotConst->Previous;
706
707 if (PreviousNotConst == NULL)
708 return false;
709
Daniel Jasper2a409b62013-07-08 14:34:09 +0000710 bool IsPPKeyword = PreviousNotConst->is(tok::identifier) &&
711 PreviousNotConst->Previous &&
712 PreviousNotConst->Previous->is(tok::hash);
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000713
Daniel Jasper92495a82013-08-19 10:16:18 +0000714 if (PreviousNotConst->Type == TT_TemplateCloser)
715 return PreviousNotConst && PreviousNotConst->MatchingParen &&
716 PreviousNotConst->MatchingParen->Previous &&
717 PreviousNotConst->MatchingParen->Previous->isNot(tok::kw_template);
718
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000719 return (!IsPPKeyword && PreviousNotConst->is(tok::identifier)) ||
720 PreviousNotConst->Type == TT_PointerOrReference ||
Daniel Jasper6ac431c2013-07-02 09:47:29 +0000721 isSimpleTypeSpecifier(*PreviousNotConst);
722 }
723
Daniel Jasper01786732013-02-04 07:21:18 +0000724 /// \brief Return the type of the given token assuming it is * or &.
Manuel Klimekb3987012013-05-29 14:47:47 +0000725 TokenType determineStarAmpUsage(const FormatToken &Tok, bool IsExpression) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000726 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasper01786732013-02-04 07:21:18 +0000727 if (PrevToken == NULL)
728 return TT_UnaryOperator;
729
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000730 const FormatToken *NextToken = Tok.getNextNonComment();
Daniel Jasper01786732013-02-04 07:21:18 +0000731 if (NextToken == NULL)
732 return TT_Unknown;
733
Daniel Jasper431f5912013-05-28 08:33:00 +0000734 if (PrevToken->is(tok::coloncolon) ||
735 (PrevToken->is(tok::l_paren) && !IsExpression))
Daniel Jasper8a5d7cd2013-03-01 17:13:29 +0000736 return TT_PointerOrReference;
737
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000738 if (PrevToken->isOneOf(tok::l_paren, tok::l_square, tok::l_brace,
Daniel Jasperd3cf17b2013-03-14 10:50:25 +0000739 tok::comma, tok::semi, tok::kw_return, tok::colon,
Daniel Jasperdbef71e2013-05-07 14:17:18 +0000740 tok::equal, tok::kw_delete) ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000741 PrevToken->Type == TT_BinaryOperator ||
Daniel Jasper01786732013-02-04 07:21:18 +0000742 PrevToken->Type == TT_UnaryOperator || PrevToken->Type == TT_CastRParen)
743 return TT_UnaryOperator;
744
Nico Webere8a97982013-02-06 06:20:11 +0000745 if (NextToken->is(tok::l_square))
746 return TT_PointerOrReference;
747
Manuel Klimekb3987012013-05-29 14:47:47 +0000748 if (PrevToken->Tok.isLiteral() ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000749 PrevToken->isOneOf(tok::r_paren, tok::r_square) ||
Manuel Klimekb3987012013-05-29 14:47:47 +0000750 NextToken->Tok.isLiteral() || NextToken->isUnaryOperator())
Daniel Jasper01786732013-02-04 07:21:18 +0000751 return TT_BinaryOperator;
752
Daniel Jasper01786732013-02-04 07:21:18 +0000753 // It is very unlikely that we are going to find a pointer or reference type
754 // definition on the RHS of an assignment.
755 if (IsExpression)
756 return TT_BinaryOperator;
757
758 return TT_PointerOrReference;
759 }
760
Manuel Klimekb3987012013-05-29 14:47:47 +0000761 TokenType determinePlusMinusCaretUsage(const FormatToken &Tok) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000762 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000763 if (PrevToken == NULL || PrevToken->Type == TT_CastRParen)
Daniel Jasper01786732013-02-04 07:21:18 +0000764 return TT_UnaryOperator;
765
766 // Use heuristics to recognize unary operators.
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000767 if (PrevToken->isOneOf(tok::equal, tok::l_paren, tok::comma, tok::l_square,
768 tok::question, tok::colon, tok::kw_return,
769 tok::kw_case, tok::at, tok::l_brace))
Daniel Jasper01786732013-02-04 07:21:18 +0000770 return TT_UnaryOperator;
771
Nico Weberee0feec2013-02-05 16:21:00 +0000772 // There can't be two consecutive binary operators.
Daniel Jasper01786732013-02-04 07:21:18 +0000773 if (PrevToken->Type == TT_BinaryOperator)
774 return TT_UnaryOperator;
775
776 // Fall back to marking the token as binary operator.
777 return TT_BinaryOperator;
778 }
779
780 /// \brief Determine whether ++/-- are pre- or post-increments/-decrements.
Manuel Klimekb3987012013-05-29 14:47:47 +0000781 TokenType determineIncrementUsage(const FormatToken &Tok) {
Alexander Kornienko0bdc6432013-07-04 14:47:51 +0000782 const FormatToken *PrevToken = Tok.getPreviousNonComment();
Daniel Jasperb8b42952013-05-31 16:14:28 +0000783 if (PrevToken == NULL || PrevToken->Type == TT_CastRParen)
Daniel Jasper01786732013-02-04 07:21:18 +0000784 return TT_UnaryOperator;
Alexander Kornienkoe74de282013-03-13 14:41:29 +0000785 if (PrevToken->isOneOf(tok::r_paren, tok::r_square, tok::identifier))
Daniel Jasper01786732013-02-04 07:21:18 +0000786 return TT_TrailingUnaryOperator;
787
788 return TT_UnaryOperator;
789 }
Daniel Jasper4e778092013-02-06 10:05:46 +0000790
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000791 // FIXME: This is copy&pasted from Sema. Put it in a common place and remove
792 // duplication.
793 /// \brief Determine whether the token kind starts a simple-type-specifier.
Manuel Klimekb3987012013-05-29 14:47:47 +0000794 bool isSimpleTypeSpecifier(const FormatToken &Tok) const {
795 switch (Tok.Tok.getKind()) {
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000796 case tok::kw_short:
797 case tok::kw_long:
798 case tok::kw___int64:
799 case tok::kw___int128:
800 case tok::kw_signed:
801 case tok::kw_unsigned:
802 case tok::kw_void:
803 case tok::kw_char:
804 case tok::kw_int:
805 case tok::kw_half:
806 case tok::kw_float:
807 case tok::kw_double:
808 case tok::kw_wchar_t:
809 case tok::kw_bool:
810 case tok::kw___underlying_type:
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000811 case tok::annot_typename:
812 case tok::kw_char16_t:
813 case tok::kw_char32_t:
814 case tok::kw_typeof:
815 case tok::kw_decltype:
Alexander Kornienko00895102013-06-05 14:09:10 +0000816 return true;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000817 default:
Alexander Kornienko00895102013-06-05 14:09:10 +0000818 return false;
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000819 }
Daniel Jasper8ed9f2b2013-04-03 13:36:17 +0000820 }
821
Daniel Jasper4e778092013-02-06 10:05:46 +0000822 SmallVector<Context, 8> Contexts;
823
Daniel Jasperd4a03db2013-08-22 15:00:41 +0000824 const FormatStyle &Style;
Daniel Jasper4e778092013-02-06 10:05:46 +0000825 AnnotatedLine &Line;
Manuel Klimekb3987012013-05-29 14:47:47 +0000826 FormatToken *CurrentToken;
Daniel Jasper4e778092013-02-06 10:05:46 +0000827 bool KeywordVirtualFound;
Daniel Jasper1407bee2013-04-11 14:29:13 +0000828 bool NameFound;
Daniel Jasper2ca37412013-07-09 14:36:48 +0000829 bool AutoFound;
Nico Weberc2e6d2a2013-02-11 15:32:15 +0000830 IdentifierInfo &Ident_in;
Daniel Jasper32d28ee2013-01-29 21:01:14 +0000831};
832
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000833static int PrecedenceUnaryOperator = prec::PointerToMember + 1;
834static int PrecedenceArrowAndPeriod = prec::PointerToMember + 2;
835
Daniel Jasper29f123b2013-02-08 15:28:42 +0000836/// \brief Parses binary expressions by inserting fake parenthesis based on
837/// operator precedence.
838class ExpressionParser {
839public:
Daniel Jasper9acb8b42013-06-06 09:11:58 +0000840 ExpressionParser(AnnotatedLine &Line) : Current(Line.First) {
841 // Skip leading "}", e.g. in "} else if (...) {".
842 if (Current->is(tok::r_brace))
843 next();
844 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000845
846 /// \brief Parse expressions with the given operatore precedence.
Daniel Jasper237d4c12013-02-23 21:01:55 +0000847 void parse(int Precedence = 0) {
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000848 if (Current == NULL || Precedence > PrecedenceArrowAndPeriod)
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000849 return;
850
Daniel Jasperc01897c2013-05-31 14:56:12 +0000851 // Conditional expressions need to be parsed separately for proper nesting.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000852 if (Precedence == prec::Conditional) {
Daniel Jasperc01897c2013-05-31 14:56:12 +0000853 parseConditionalExpr();
854 return;
855 }
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000856
857 // Parse unary operators, which all have a higher precedence than binary
858 // operators.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000859 if (Precedence == PrecedenceUnaryOperator) {
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000860 parseUnaryOperator();
Daniel Jasper29f123b2013-02-08 15:28:42 +0000861 return;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000862 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000863
Manuel Klimekb3987012013-05-29 14:47:47 +0000864 FormatToken *Start = Current;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000865 FormatToken *LatestOperator = NULL;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000866
Daniel Jasper237d4c12013-02-23 21:01:55 +0000867 while (Current) {
Daniel Jasper29f123b2013-02-08 15:28:42 +0000868 // Consume operators with higher precedence.
Daniel Jasperbf71ba22013-04-08 20:33:42 +0000869 parse(Precedence + 1);
Daniel Jasper29f123b2013-02-08 15:28:42 +0000870
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000871 int CurrentPrecedence = getCurrentPrecedence();
872
873 if (Current && Current->Type == TT_ObjCSelectorName &&
874 Precedence == CurrentPrecedence)
875 Start = Current;
Daniel Jasper237d4c12013-02-23 21:01:55 +0000876
Daniel Jasper29f123b2013-02-08 15:28:42 +0000877 // At the end of the line or when an operator with higher precedence is
878 // found, insert fake parenthesis and return.
Daniel Jasperac3223e2013-04-10 09:49:49 +0000879 if (Current == NULL || Current->closesScope() ||
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000880 (CurrentPrecedence != -1 && CurrentPrecedence < Precedence)) {
881 if (LatestOperator) {
882 if (Precedence == PrecedenceArrowAndPeriod) {
883 LatestOperator->LastInChainOfCalls = true;
884 // Call expressions don't have a binary operator precedence.
885 addFakeParenthesis(Start, prec::Unknown);
886 } else {
887 addFakeParenthesis(Start, prec::Level(Precedence));
888 }
889 }
Daniel Jasper29f123b2013-02-08 15:28:42 +0000890 return;
891 }
892
893 // Consume scopes: (), [], <> and {}
Daniel Jasperac3223e2013-04-10 09:49:49 +0000894 if (Current->opensScope()) {
895 while (Current && !Current->closesScope()) {
Daniel Jasper29f123b2013-02-08 15:28:42 +0000896 next();
897 parse();
898 }
899 next();
900 } else {
901 // Operator found.
Daniel Jasper237d4c12013-02-23 21:01:55 +0000902 if (CurrentPrecedence == Precedence)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000903 LatestOperator = Current;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000904
905 next();
906 }
907 }
908 }
909
910private:
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000911 /// \brief Gets the precedence (+1) of the given token for binary operators
912 /// and other tokens that we treat like binary operators.
913 int getCurrentPrecedence() {
914 if (Current) {
915 if (Current->Type == TT_ConditionalExpr)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000916 return prec::Conditional;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000917 else if (Current->is(tok::semi) || Current->Type == TT_InlineASMColon)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000918 return 0;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000919 else if (Current->Type == TT_BinaryOperator || Current->is(tok::comma))
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000920 return Current->getPrecedence();
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000921 else if (Current->Type == TT_ObjCSelectorName)
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000922 return prec::Assignment;
923 else if (Current->isOneOf(tok::period, tok::arrow))
924 return PrecedenceArrowAndPeriod;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000925 }
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000926 return -1;
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000927 }
928
Daniel Jasperc01897c2013-05-31 14:56:12 +0000929 void addFakeParenthesis(FormatToken *Start, prec::Level Precedence) {
930 Start->FakeLParens.push_back(Precedence);
Daniel Jasperdb4813a2013-09-06 08:08:14 +0000931 if (Precedence > prec::Unknown)
932 Start->StartsBinaryExpression = true;
933 if (Current) {
Daniel Jasperc01897c2013-05-31 14:56:12 +0000934 ++Current->Previous->FakeRParens;
Daniel Jasperdb4813a2013-09-06 08:08:14 +0000935 if (Precedence > prec::Unknown)
936 Current->Previous->EndsBinaryExpression = true;
937 }
Daniel Jasperc01897c2013-05-31 14:56:12 +0000938 }
939
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000940 /// \brief Parse unary operator expressions and surround them with fake
941 /// parentheses if appropriate.
942 void parseUnaryOperator() {
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000943 if (Current == NULL || Current->Type != TT_UnaryOperator) {
944 parse(PrecedenceArrowAndPeriod);
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000945 return;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000946 }
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000947
948 FormatToken *Start = Current;
949 next();
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000950 parseUnaryOperator();
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000951
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000952 // The actual precedence doesn't matter.
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000953 addFakeParenthesis(Start, prec::Unknown);
Daniel Jasper3618e6f2013-08-23 15:14:03 +0000954 }
955
Daniel Jasperc01897c2013-05-31 14:56:12 +0000956 void parseConditionalExpr() {
957 FormatToken *Start = Current;
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000958 parse(prec::LogicalOr);
Daniel Jasperc01897c2013-05-31 14:56:12 +0000959 if (!Current || !Current->is(tok::question))
960 return;
961 next();
Daniel Jasperd489f8c2013-08-27 11:09:05 +0000962 parse(prec::LogicalOr);
Daniel Jasperc01897c2013-05-31 14:56:12 +0000963 if (!Current || Current->Type != TT_ConditionalExpr)
964 return;
965 next();
966 parseConditionalExpr();
967 addFakeParenthesis(Start, prec::Conditional);
968 }
969
Daniel Jasper29f123b2013-02-08 15:28:42 +0000970 void next() {
Alexander Kornienkod71b15b2013-06-17 13:19:53 +0000971 if (Current)
972 Current = Current->Next;
973 while (Current && Current->isTrailingComment())
Manuel Klimekb3987012013-05-29 14:47:47 +0000974 Current = Current->Next;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000975 }
976
Manuel Klimekb3987012013-05-29 14:47:47 +0000977 FormatToken *Current;
Daniel Jasper29f123b2013-02-08 15:28:42 +0000978};
979
Craig Topper14e66492013-07-01 04:03:19 +0000980} // end anonymous namespace
981
Daniel Jasperb77d7412013-09-06 07:54:20 +0000982void
983TokenAnnotator::setCommentLineLevels(SmallVectorImpl<AnnotatedLine *> &Lines) {
984 if (Lines.empty())
985 return;
986
987 const AnnotatedLine *NextNonCommentLine = NULL;
988 for (unsigned i = Lines.size() - 1; i > 0; --i) {
989 if (NextNonCommentLine && Lines[i]->First->is(tok::comment) &&
990 !Lines[i]->First->Next)
991 Lines[i]->Level = NextNonCommentLine->Level;
992 else
993 NextNonCommentLine =
994 Lines[i]->First->isNot(tok::r_brace) ? Lines[i] : NULL;
995 }
996}
997
Daniel Jasper8ff690a2013-02-06 14:22:40 +0000998void TokenAnnotator::annotate(AnnotatedLine &Line) {
Daniel Jasperb77d7412013-09-06 07:54:20 +0000999 setCommentLineLevels(Line.Children);
1000 for (SmallVectorImpl<AnnotatedLine *>::iterator I = Line.Children.begin(),
1001 E = Line.Children.end();
Daniel Jasper567dcf92013-09-05 09:29:45 +00001002 I != E; ++I) {
1003 annotate(**I);
1004 }
Daniel Jasperd4a03db2013-08-22 15:00:41 +00001005 AnnotatingParser Parser(Style, Line, Ident_in);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001006 Line.Type = Parser.parseLine();
1007 if (Line.Type == LT_Invalid)
1008 return;
1009
Daniel Jasper29f123b2013-02-08 15:28:42 +00001010 ExpressionParser ExprParser(Line);
1011 ExprParser.parse();
1012
Manuel Klimekb3987012013-05-29 14:47:47 +00001013 if (Line.First->Type == TT_ObjCMethodSpecifier)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001014 Line.Type = LT_ObjCMethodDecl;
Manuel Klimekb3987012013-05-29 14:47:47 +00001015 else if (Line.First->Type == TT_ObjCDecl)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001016 Line.Type = LT_ObjCDecl;
Manuel Klimekb3987012013-05-29 14:47:47 +00001017 else if (Line.First->Type == TT_ObjCProperty)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001018 Line.Type = LT_ObjCProperty;
1019
Manuel Klimekb3987012013-05-29 14:47:47 +00001020 Line.First->SpacesRequiredBefore = 1;
1021 Line.First->CanBreakBefore = Line.First->MustBreakBefore;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001022}
1023
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001024void TokenAnnotator::calculateFormattingInformation(AnnotatedLine &Line) {
Alexander Kornienko00895102013-06-05 14:09:10 +00001025 Line.First->TotalLength = Line.First->CodePointCount;
Manuel Klimekb3987012013-05-29 14:47:47 +00001026 if (!Line.First->Next)
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001027 return;
Manuel Klimekb3987012013-05-29 14:47:47 +00001028 FormatToken *Current = Line.First->Next;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001029 while (Current != NULL) {
Daniel Jasper729a7432013-02-11 12:36:37 +00001030 if (Current->Type == TT_LineComment)
1031 Current->SpacesRequiredBefore = Style.SpacesBeforeTrailingComments;
1032 else
1033 Current->SpacesRequiredBefore =
1034 spaceRequiredBefore(Line, *Current) ? 1 : 0;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001035
Daniel Jasper561211d2013-07-16 20:28:33 +00001036 if (Current->is(tok::comment)) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001037 Current->MustBreakBefore = Current->NewlinesBefore > 0;
1038 } else if (Current->Previous->isTrailingComment() ||
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001039 (Current->is(tok::string_literal) &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001040 Current->Previous->is(tok::string_literal))) {
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001041 Current->MustBreakBefore = true;
Daniel Jasper561211d2013-07-16 20:28:33 +00001042 } else if (Current->Previous->IsUnterminatedLiteral) {
1043 Current->MustBreakBefore = true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001044 } else if (Current->is(tok::lessless) && Current->Next &&
1045 Current->Previous->is(tok::string_literal) &&
1046 Current->Next->is(tok::string_literal)) {
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001047 Current->MustBreakBefore = true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001048 } else if (Current->Previous->ClosesTemplateDeclaration &&
Daniel Jasperbbc87762013-05-29 12:07:31 +00001049 Style.AlwaysBreakTemplateDeclarations) {
1050 Current->MustBreakBefore = true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +00001051 } else if (Current->Type == TT_CtorInitializerComma &&
1052 Style.BreakConstructorInitializersBeforeComma) {
1053 Current->MustBreakBefore = true;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001054 }
1055 Current->CanBreakBefore =
1056 Current->MustBreakBefore || canBreakBefore(Line, *Current);
Daniel Jasper567dcf92013-09-05 09:29:45 +00001057 if (Current->MustBreakBefore || !Current->Children.empty() ||
Alexander Kornienko4b762a92013-09-02 13:58:14 +00001058 (Current->is(tok::string_literal) && Current->isMultiline()))
Manuel Klimekb3987012013-05-29 14:47:47 +00001059 Current->TotalLength = Current->Previous->TotalLength + Style.ColumnLimit;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001060 else
Daniel Jasper2a409b62013-07-08 14:34:09 +00001061 Current->TotalLength = Current->Previous->TotalLength +
1062 Current->CodePointCount +
1063 Current->SpacesRequiredBefore;
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001064 // FIXME: Only calculate this if CanBreakBefore is true once static
1065 // initializers etc. are sorted out.
1066 // FIXME: Move magic numbers to a better place.
1067 Current->SplitPenalty =
1068 20 * Current->BindingStrength + splitPenalty(Line, *Current);
1069
Manuel Klimekb3987012013-05-29 14:47:47 +00001070 Current = Current->Next;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001071 }
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001072
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001073 calculateUnbreakableTailLengths(Line);
Daniel Jasperd4a03db2013-08-22 15:00:41 +00001074 for (Current = Line.First; Current != NULL; Current = Current->Next) {
1075 if (Current->Role)
1076 Current->Role->precomputeFormattingInfos(Current);
1077 }
1078
Daniel Jasper567dcf92013-09-05 09:29:45 +00001079 DEBUG({ printDebugInfo(Line); });
1080
Daniel Jasperb77d7412013-09-06 07:54:20 +00001081 for (SmallVectorImpl<AnnotatedLine *>::iterator I = Line.Children.begin(),
1082 E = Line.Children.end();
Daniel Jasper567dcf92013-09-05 09:29:45 +00001083 I != E; ++I) {
1084 calculateFormattingInformation(**I);
1085 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001086}
1087
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001088void TokenAnnotator::calculateUnbreakableTailLengths(AnnotatedLine &Line) {
1089 unsigned UnbreakableTailLength = 0;
Manuel Klimekb3987012013-05-29 14:47:47 +00001090 FormatToken *Current = Line.Last;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001091 while (Current != NULL) {
1092 Current->UnbreakableTailLength = UnbreakableTailLength;
1093 if (Current->CanBreakBefore ||
1094 Current->isOneOf(tok::comment, tok::string_literal)) {
1095 UnbreakableTailLength = 0;
1096 } else {
1097 UnbreakableTailLength +=
Alexander Kornienko00895102013-06-05 14:09:10 +00001098 Current->CodePointCount + Current->SpacesRequiredBefore;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001099 }
Manuel Klimekb3987012013-05-29 14:47:47 +00001100 Current = Current->Previous;
Manuel Klimeke573c3f2013-05-22 12:51:29 +00001101 }
1102}
1103
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001104unsigned TokenAnnotator::splitPenalty(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001105 const FormatToken &Tok) {
1106 const FormatToken &Left = *Tok.Previous;
1107 const FormatToken &Right = Tok;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001108
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001109 if (Left.is(tok::semi))
1110 return 0;
1111 if (Left.is(tok::comma))
1112 return 1;
Daniel Jasper011c35d2013-07-12 11:19:37 +00001113 if (Right.is(tok::l_square))
1114 return 150;
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001115
Daniel Jasper6561f6a2013-07-09 07:43:55 +00001116 if (Right.Type == TT_StartOfName || Right.is(tok::kw_operator)) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001117 if (Line.First->is(tok::kw_for) && Right.PartOfMultiVariableDeclStmt)
Daniel Jasper3c08a812013-02-24 18:54:32 +00001118 return 3;
Daniel Jasperc18cff32013-07-11 12:34:23 +00001119 if (Left.Type == TT_StartOfName)
1120 return 20;
Daniel Jasper92495a82013-08-19 10:16:18 +00001121 if (Line.MightBeFunctionDecl && Right.BindingStrength == 1)
Daniel Jasper3c08a812013-02-24 18:54:32 +00001122 // FIXME: Clean up hack of using BindingStrength to find top-level names.
1123 return Style.PenaltyReturnTypeOnItsOwnLine;
Daniel Jasper92495a82013-08-19 10:16:18 +00001124 return 200;
Daniel Jasper3c08a812013-02-24 18:54:32 +00001125 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001126 if (Left.is(tok::equal) && Right.is(tok::l_brace))
1127 return 150;
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001128 if (Left.Type == TT_CastRParen)
1129 return 100;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001130 if (Left.is(tok::coloncolon))
1131 return 500;
Daniel Jasper6b119d62013-04-05 17:22:09 +00001132 if (Left.isOneOf(tok::kw_class, tok::kw_struct))
1133 return 5000;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001134
Daniel Jasper6cabab42013-02-14 08:42:54 +00001135 if (Left.Type == TT_RangeBasedForLoopColon ||
1136 Left.Type == TT_InheritanceColon)
Daniel Jasper84a1a632013-02-26 13:18:08 +00001137 return 2;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001138
Daniel Jasperd3fef0f2013-08-27 14:24:43 +00001139 if (Right.isMemberAccess()) {
Daniel Jasper2f0a0202013-09-06 08:54:24 +00001140 if (Left.isOneOf(tok::r_paren, tok::r_square) && Left.MatchingParen &&
1141 Left.MatchingParen->ParameterCount > 0)
Daniel Jasper518ee342013-02-26 13:59:14 +00001142 return 20; // Should be smaller than breaking at a nested comma.
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001143 return 150;
1144 }
1145
Daniel Jasper20a0f8c2013-07-11 21:02:56 +00001146 // Breaking before a trailing 'const' or not-function-like annotation is bad.
Daniel Jasperaa9e7b12013-08-01 13:46:58 +00001147 if (Left.is(tok::r_paren) && Line.Type != LT_ObjCProperty &&
Daniel Jasper20a0f8c2013-07-11 21:02:56 +00001148 (Right.is(tok::kw_const) || (Right.is(tok::identifier) && Right.Next &&
1149 Right.Next->isNot(tok::l_paren))))
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001150 return 150;
1151
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001152 // In for-loops, prefer breaking at ',' and ';'.
Manuel Klimekb3987012013-05-29 14:47:47 +00001153 if (Line.First->is(tok::kw_for) && Left.is(tok::equal))
Daniel Jasper7d812812013-02-21 15:00:29 +00001154 return 4;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001155
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001156 // In Objective-C method expressions, prefer breaking before "param:" over
1157 // breaking after it.
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001158 if (Right.Type == TT_ObjCSelectorName)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001159 return 0;
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001160 if (Left.is(tok::colon) && Left.Type == TT_ObjCMethodExpr)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001161 return 20;
1162
Daniel Jasper1407bee2013-04-11 14:29:13 +00001163 if (Left.is(tok::l_paren) && Line.MightBeFunctionDecl)
1164 return 100;
Daniel Jasperac3223e2013-04-10 09:49:49 +00001165 if (Left.opensScope())
Daniel Jaspere60084d2013-08-13 06:50:04 +00001166 return Left.ParameterCount > 1 ? prec::Comma : 19;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001167
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001168 if (Right.is(tok::lessless)) {
1169 if (Left.is(tok::string_literal)) {
Alexander Kornienko00895102013-06-05 14:09:10 +00001170 StringRef Content = Left.TokenText;
Daniel Jasperbfa1edd2013-03-14 14:00:17 +00001171 Content = Content.drop_back(1).drop_front(1).trim();
1172 if (Content.size() > 1 &&
1173 (Content.back() == ':' || Content.back() == '='))
Daniel Jasper9637dda2013-07-15 14:33:14 +00001174 return 25;
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001175 }
Daniel Jasper0c368782013-07-15 15:04:42 +00001176 return 1; // Breaking at a << is really cheap.
Daniel Jasper4e8a7b42013-02-06 21:04:05 +00001177 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001178 if (Left.Type == TT_ConditionalExpr)
Daniel Jasper518ee342013-02-26 13:59:14 +00001179 return prec::Conditional;
Manuel Klimekb3987012013-05-29 14:47:47 +00001180 prec::Level Level = Left.getPrecedence();
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001181
1182 if (Level != prec::Unknown)
1183 return Level;
Daniel Jasper24849712013-03-01 16:48:32 +00001184
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001185 return 3;
1186}
1187
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001188bool TokenAnnotator::spaceRequiredBetween(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001189 const FormatToken &Left,
1190 const FormatToken &Right) {
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001191 if (Right.is(tok::hashhash))
1192 return Left.is(tok::hash);
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001193 if (Left.isOneOf(tok::hashhash, tok::hash))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001194 return Right.is(tok::hash);
Daniel Jasper7df56bf2013-08-20 12:36:34 +00001195 if (Left.is(tok::l_paren) && Right.is(tok::r_paren))
1196 return Style.SpaceInEmptyParentheses;
1197 if (Left.is(tok::l_paren) || Right.is(tok::r_paren))
Daniel Jasper34f3d052013-08-21 08:39:01 +00001198 return (Right.Type == TT_CastRParen ||
1199 (Left.MatchingParen && Left.MatchingParen->Type == TT_CastRParen))
Daniel Jasper7df56bf2013-08-20 12:36:34 +00001200 ? Style.SpacesInCStyleCastParentheses
1201 : Style.SpacesInParentheses;
1202 if (Right.isOneOf(tok::semi, tok::comma))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001203 return false;
1204 if (Right.is(tok::less) &&
1205 (Left.is(tok::kw_template) ||
1206 (Line.Type == LT_ObjCDecl && Style.ObjCSpaceBeforeProtocolList)))
1207 return true;
1208 if (Left.is(tok::arrow) || Right.is(tok::arrow))
1209 return false;
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001210 if (Left.isOneOf(tok::exclaim, tok::tilde))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001211 return false;
1212 if (Left.is(tok::at) &&
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001213 Right.isOneOf(tok::identifier, tok::string_literal, tok::char_constant,
1214 tok::numeric_constant, tok::l_paren, tok::l_brace,
1215 tok::kw_true, tok::kw_false))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001216 return false;
1217 if (Left.is(tok::coloncolon))
1218 return false;
1219 if (Right.is(tok::coloncolon))
Daniel Jasper081fc172013-08-28 07:07:07 +00001220 return !Left.isOneOf(tok::identifier, tok::greater, tok::l_paren,
1221 tok::r_paren);
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001222 if (Left.is(tok::less) || Right.isOneOf(tok::greater, tok::less))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001223 return false;
Daniel Jasperc47d7f12013-07-01 09:47:25 +00001224 if (Right.is(tok::ellipsis))
1225 return false;
Manuel Klimek31e44f72013-09-04 08:20:47 +00001226 if (Left.is(tok::l_square) && Right.is(tok::amp))
1227 return false;
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001228 if (Right.Type == TT_PointerOrReference)
Manuel Klimekb3987012013-05-29 14:47:47 +00001229 return Left.Tok.isLiteral() ||
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001230 ((Left.Type != TT_PointerOrReference) && Left.isNot(tok::l_paren) &&
1231 !Style.PointerBindsToType);
Daniel Jasper3ff4a2f2013-05-28 15:27:10 +00001232 if (Right.Type == TT_FunctionTypeLParen && Left.isNot(tok::l_paren) &&
Daniel Jasper395228f2013-05-08 14:58:20 +00001233 (Left.Type != TT_PointerOrReference || Style.PointerBindsToType))
1234 return true;
Alexander Kornienko3fd9ccd2013-03-12 16:28:18 +00001235 if (Left.Type == TT_PointerOrReference)
Daniel Jasper3a1847e2013-07-01 09:34:09 +00001236 return Right.Tok.isLiteral() || Right.Type == TT_BlockComment ||
Daniel Jasper9322aae2013-03-20 09:53:18 +00001237 ((Right.Type != TT_PointerOrReference) &&
Daniel Jasper81d2d382013-04-01 17:13:26 +00001238 Right.isNot(tok::l_paren) && Style.PointerBindsToType &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001239 Left.Previous &&
1240 !Left.Previous->isOneOf(tok::l_paren, tok::coloncolon));
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001241 if (Right.is(tok::star) && Left.is(tok::l_paren))
1242 return false;
Nico Weber051860e2013-02-10 02:08:05 +00001243 if (Left.is(tok::l_square))
1244 return Left.Type == TT_ObjCArrayLiteral && Right.isNot(tok::r_square);
1245 if (Right.is(tok::r_square))
1246 return Right.Type == TT_ObjCArrayLiteral;
Daniel Jasperec172262013-08-30 10:36:58 +00001247 if (Right.is(tok::l_square) && Right.Type != TT_ObjCMethodExpr &&
Daniel Jasper567dcf92013-09-05 09:29:45 +00001248 Right.Type != TT_LambdaLSquare && Left.isNot(tok::numeric_constant))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001249 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001250 if (Left.is(tok::colon))
1251 return Left.Type != TT_ObjCMethodExpr;
1252 if (Right.is(tok::colon))
Daniel Jasperab3ce592013-08-01 22:05:00 +00001253 return Right.Type != TT_ObjCMethodExpr && !Left.is(tok::question);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001254 if (Right.is(tok::l_paren)) {
Daniel Jaspere0fa4c52013-07-17 20:25:02 +00001255 if (Left.is(tok::r_paren) && Left.MatchingParen &&
1256 Left.MatchingParen->Previous &&
1257 Left.MatchingParen->Previous->is(tok::kw___attribute))
1258 return true;
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001259 return Line.Type == LT_ObjCDecl ||
Daniel Jasper34f3d052013-08-21 08:39:01 +00001260 Left.isOneOf(tok::kw_return, tok::kw_new, tok::kw_delete,
1261 tok::semi) ||
1262 (Style.SpaceAfterControlStatementKeyword &&
1263 Left.isOneOf(tok::kw_if, tok::kw_for, tok::kw_while, tok::kw_switch,
1264 tok::kw_catch));
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001265 }
Manuel Klimekb3987012013-05-29 14:47:47 +00001266 if (Left.is(tok::at) && Right.Tok.getObjCKeywordID() != tok::objc_not_keyword)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001267 return false;
1268 if (Left.is(tok::l_brace) && Right.is(tok::r_brace))
Daniel Jasper567dcf92013-09-05 09:29:45 +00001269 return !Left.Children.empty(); // No spaces in "{}".
Daniel Jasper2424eef2013-05-23 10:15:45 +00001270 if (Left.is(tok::l_brace) || Right.is(tok::r_brace))
Daniel Jasperb5dc3f42013-07-16 18:22:10 +00001271 return !Style.Cpp11BracedListStyle;
Daniel Jasper1bee0732013-05-23 18:05:18 +00001272 if (Right.Type == TT_UnaryOperator)
1273 return !Left.isOneOf(tok::l_paren, tok::l_square, tok::at) &&
1274 (Left.isNot(tok::colon) || Left.Type != TT_ObjCMethodExpr);
Daniel Jasperce933562013-05-23 21:35:49 +00001275 if (Left.isOneOf(tok::identifier, tok::greater, tok::r_square) &&
Manuel Klimek31e44f72013-09-04 08:20:47 +00001276 Right.is(tok::l_brace) && Right.getNextNonComment() &&
1277 Right.BlockKind != BK_Block)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001278 return false;
Daniel Jasper5ad390d2013-05-28 11:30:49 +00001279 if (Left.is(tok::period) || Right.is(tok::period))
1280 return false;
Nico Weber861576b2013-06-26 00:15:19 +00001281 if (Left.Type == TT_BlockComment && Left.TokenText.endswith("=*/"))
1282 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001283 return true;
1284}
1285
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001286bool TokenAnnotator::spaceRequiredBefore(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001287 const FormatToken &Tok) {
1288 if (Tok.Tok.getIdentifierInfo() && Tok.Previous->Tok.getIdentifierInfo())
Daniel Jasper2b4c9242013-02-11 08:01:18 +00001289 return true; // Never ever merge two identifiers.
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001290 if (Line.Type == LT_ObjCMethodDecl) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001291 if (Tok.Previous->Type == TT_ObjCMethodSpecifier)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001292 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001293 if (Tok.Previous->is(tok::r_paren) && Tok.is(tok::identifier))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001294 // Don't space between ')' and <id>
1295 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001296 }
1297 if (Line.Type == LT_ObjCProperty &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001298 (Tok.is(tok::equal) || Tok.Previous->is(tok::equal)))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001299 return false;
1300
Daniel Jasper2ca37412013-07-09 14:36:48 +00001301 if (Tok.Type == TT_TrailingReturnArrow ||
1302 Tok.Previous->Type == TT_TrailingReturnArrow)
1303 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001304 if (Tok.Previous->is(tok::comma))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001305 return true;
Daniel Jasper9c3c7b32013-02-28 13:40:17 +00001306 if (Tok.is(tok::comma))
1307 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001308 if (Tok.Type == TT_CtorInitializerColon || Tok.Type == TT_ObjCBlockLParen)
1309 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001310 if (Tok.Previous->Tok.is(tok::kw_operator))
Daniel Jasper2b4c9242013-02-11 08:01:18 +00001311 return false;
1312 if (Tok.Type == TT_OverloadedOperatorLParen)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001313 return false;
1314 if (Tok.is(tok::colon))
Manuel Klimekb3987012013-05-29 14:47:47 +00001315 return !Line.First->isOneOf(tok::kw_case, tok::kw_default) &&
Daniel Jasperab3ce592013-08-01 22:05:00 +00001316 Tok.getNextNonComment() != NULL && Tok.Type != TT_ObjCMethodExpr &&
1317 !Tok.Previous->is(tok::question);
Manuel Klimekb3987012013-05-29 14:47:47 +00001318 if (Tok.Previous->Type == TT_UnaryOperator ||
1319 Tok.Previous->Type == TT_CastRParen)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001320 return false;
Manuel Klimekb3987012013-05-29 14:47:47 +00001321 if (Tok.Previous->is(tok::greater) && Tok.is(tok::greater)) {
Daniel Jasper29f123b2013-02-08 15:28:42 +00001322 return Tok.Type == TT_TemplateCloser &&
Manuel Klimekb3987012013-05-29 14:47:47 +00001323 Tok.Previous->Type == TT_TemplateCloser &&
Daniel Jasper29f123b2013-02-08 15:28:42 +00001324 Style.Standard != FormatStyle::LS_Cpp11;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001325 }
Alexander Kornienko54a38bd2013-03-20 16:41:56 +00001326 if (Tok.isOneOf(tok::arrowstar, tok::periodstar) ||
Manuel Klimekb3987012013-05-29 14:47:47 +00001327 Tok.Previous->isOneOf(tok::arrowstar, tok::periodstar))
Daniel Jasper9c3c7b32013-02-28 13:40:17 +00001328 return false;
Daniel Jasper1dc6f742013-08-07 16:29:23 +00001329 if ((Tok.Type == TT_BinaryOperator && !Tok.Previous->is(tok::l_paren)) ||
1330 Tok.Previous->Type == TT_BinaryOperator)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001331 return true;
Manuel Klimekb3987012013-05-29 14:47:47 +00001332 if (Tok.Previous->Type == TT_TemplateCloser && Tok.is(tok::l_paren))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001333 return false;
Daniel Jaspera4dd9822013-08-28 08:24:04 +00001334 if (Tok.is(tok::less) && Tok.Previous->isNot(tok::l_paren) &&
1335 Line.First->is(tok::hash))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001336 return true;
1337 if (Tok.Type == TT_TrailingUnaryOperator)
1338 return false;
Manuel Klimekb3987012013-05-29 14:47:47 +00001339 return spaceRequiredBetween(Line, *Tok.Previous, Tok);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001340}
1341
Daniel Jasper8ff690a2013-02-06 14:22:40 +00001342bool TokenAnnotator::canBreakBefore(const AnnotatedLine &Line,
Manuel Klimekb3987012013-05-29 14:47:47 +00001343 const FormatToken &Right) {
1344 const FormatToken &Left = *Right.Previous;
Daniel Jasper6561f6a2013-07-09 07:43:55 +00001345 if (Right.Type == TT_StartOfName || Right.is(tok::kw_operator))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001346 return true;
Nico Weberf2ff8122013-05-26 05:39:26 +00001347 if (Right.is(tok::colon) &&
1348 (Right.Type == TT_ObjCDictLiteral || Right.Type == TT_ObjCMethodExpr))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001349 return false;
Nico Weberf2ff8122013-05-26 05:39:26 +00001350 if (Left.is(tok::colon) &&
1351 (Left.Type == TT_ObjCDictLiteral || Left.Type == TT_ObjCMethodExpr))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001352 return true;
Daniel Jasper63d7ced2013-02-05 10:07:47 +00001353 if (Right.Type == TT_ObjCSelectorName)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001354 return true;
Daniel Jasperaa9e7b12013-08-01 13:46:58 +00001355 if (Left.is(tok::r_paren) && Line.Type == LT_ObjCProperty)
1356 return true;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001357 if (Left.ClosesTemplateDeclaration)
1358 return true;
Daniel Jasperab3ce592013-08-01 22:05:00 +00001359 if ((Right.Type == TT_ConditionalExpr &&
1360 !(Right.is(tok::colon) && Left.is(tok::question))) ||
1361 Right.is(tok::question))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001362 return true;
Daniel Jasper6cabab42013-02-14 08:42:54 +00001363 if (Right.Type == TT_RangeBasedForLoopColon ||
Daniel Jasperc476ea92013-08-28 07:27:35 +00001364 Right.Type == TT_OverloadedOperatorLParen ||
1365 Right.Type == TT_OverloadedOperator)
Daniel Jasper6cabab42013-02-14 08:42:54 +00001366 return false;
Daniel Jasperc194c952013-05-06 06:45:09 +00001367 if (Left.Type == TT_RangeBasedForLoopColon)
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001368 return true;
Daniel Jasper7d812812013-02-21 15:00:29 +00001369 if (Right.Type == TT_RangeBasedForLoopColon)
1370 return false;
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001371 if (Left.Type == TT_PointerOrReference || Left.Type == TT_TemplateCloser ||
1372 Left.Type == TT_UnaryOperator || Left.Type == TT_ConditionalExpr ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001373 Left.isOneOf(tok::question, tok::kw_operator))
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001374 return false;
1375 if (Left.is(tok::equal) && Line.Type == LT_VirtualFunctionDecl)
1376 return false;
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001377 if (Left.Previous) {
1378 if (Left.is(tok::l_paren) && Right.is(tok::l_paren) &&
1379 Left.Previous->is(tok::kw___attribute))
1380 return false;
Daniel Jasper2ca37412013-07-09 14:36:48 +00001381 if (Left.is(tok::l_paren) && (Left.Previous->Type == TT_BinaryOperator ||
Daniel Jasper5e2169f2013-07-18 14:46:07 +00001382 Left.Previous->Type == TT_CastRParen))
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001383 return false;
1384 }
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001385
Daniel Jasper65d2c382013-06-06 16:08:57 +00001386 if (Right.isTrailingComment())
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001387 // We rely on MustBreakBefore being set correctly here as we should not
1388 // change the "binding" behavior of a comment.
1389 return false;
1390
Daniel Jasper567dcf92013-09-05 09:29:45 +00001391 if (Right.is(tok::r_paren) || Right.Type == TT_TemplateCloser)
1392 return false;
1393
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001394 // We only break before r_brace if there was a corresponding break before
1395 // the l_brace, which is tracked by BreakBeforeClosingBrace.
Daniel Jasper567dcf92013-09-05 09:29:45 +00001396 if (Right.is(tok::r_brace))
1397 return Right.MatchingParen && Right.MatchingParen->BlockKind == BK_Block;
Daniel Jasper5ad72bb2013-05-22 08:28:26 +00001398
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001399 // Allow breaking after a trailing 'const', e.g. after a method declaration,
1400 // unless it is follow by ';', '{' or '='.
Manuel Klimekb3987012013-05-29 14:47:47 +00001401 if (Left.is(tok::kw_const) && Left.Previous != NULL &&
1402 Left.Previous->is(tok::r_paren))
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001403 return !Right.isOneOf(tok::l_brace, tok::semi, tok::equal);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001404
Daniel Jasper8ef19a22013-03-14 09:50:46 +00001405 if (Right.is(tok::kw___attribute))
1406 return true;
1407
Daniel Jasper3a204412013-02-23 07:46:38 +00001408 if (Left.is(tok::identifier) && Right.is(tok::string_literal))
1409 return true;
Daniel Jaspere8b10d32013-07-26 16:56:36 +00001410
1411 if (Left.Type == TT_CtorInitializerComma &&
1412 Style.BreakConstructorInitializersBeforeComma)
1413 return false;
1414 if (Right.isBinaryOperator() && Style.BreakBeforeBinaryOperators)
1415 return true;
1416 return (Left.isBinaryOperator() && Left.isNot(tok::lessless) &&
1417 !Style.BreakBeforeBinaryOperators) ||
Daniel Jasper6b119d62013-04-05 17:22:09 +00001418 Left.isOneOf(tok::comma, tok::coloncolon, tok::semi, tok::l_brace,
1419 tok::kw_class, tok::kw_struct) ||
Alexander Kornienkoe74de282013-03-13 14:41:29 +00001420 Right.isOneOf(tok::lessless, tok::arrow, tok::period, tok::colon) ||
Daniel Jasper198c8bf2013-07-05 07:58:34 +00001421 (Left.is(tok::r_paren) &&
Daniel Jaspere033e872013-05-21 09:16:31 +00001422 Right.isOneOf(tok::identifier, tok::kw_const, tok::kw___attribute)) ||
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001423 (Left.is(tok::l_paren) && !Right.is(tok::r_paren)) ||
Daniel Jasper011c35d2013-07-12 11:19:37 +00001424 Right.is(tok::l_square);
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001425}
1426
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001427void TokenAnnotator::printDebugInfo(const AnnotatedLine &Line) {
1428 llvm::errs() << "AnnotatedTokens:\n";
Manuel Klimekb3987012013-05-29 14:47:47 +00001429 const FormatToken *Tok = Line.First;
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001430 while (Tok) {
Manuel Klimekb3987012013-05-29 14:47:47 +00001431 llvm::errs() << " M=" << Tok->MustBreakBefore
Daniel Jasperc7bd68f2013-07-10 14:02:49 +00001432 << " C=" << Tok->CanBreakBefore << " T=" << Tok->Type
1433 << " S=" << Tok->SpacesRequiredBefore
1434 << " P=" << Tok->SplitPenalty << " Name=" << Tok->Tok.getName()
1435 << " PPK=" << Tok->PackingKind << " FakeLParens=";
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001436 for (unsigned i = 0, e = Tok->FakeLParens.size(); i != e; ++i)
1437 llvm::errs() << Tok->FakeLParens[i] << "/";
1438 llvm::errs() << " FakeRParens=" << Tok->FakeRParens << "\n";
Manuel Klimekb3987012013-05-29 14:47:47 +00001439 Tok = Tok->Next;
Daniel Jasperbf71ba22013-04-08 20:33:42 +00001440 }
1441 llvm::errs() << "----\n";
1442}
1443
Daniel Jasper32d28ee2013-01-29 21:01:14 +00001444} // namespace format
1445} // namespace clang