blob: b438a3f64e71335fd168d43161bd0a12bae8be4e [file] [log] [blame]
Daniel Jasperde0328a2013-08-16 11:20:30 +00001//===--- ContinuationIndenter.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 the continuation indenter.
12///
13//===----------------------------------------------------------------------===//
14
Daniel Jasperde0328a2013-08-16 11:20:30 +000015#include "ContinuationIndenter.h"
Manuel Klimek89628f62017-09-20 09:51:03 +000016#include "BreakableToken.h"
Daniel Jasperde0328a2013-08-16 11:20:30 +000017#include "WhitespaceManager.h"
18#include "clang/Basic/OperatorPrecedence.h"
19#include "clang/Basic/SourceManager.h"
20#include "clang/Format/Format.h"
21#include "llvm/Support/Debug.h"
Daniel Jasperde0328a2013-08-16 11:20:30 +000022
Krasimir Georgiev91834222017-01-25 13:58:58 +000023#define DEBUG_TYPE "format-indenter"
Chandler Carruth10346662014-04-22 03:17:02 +000024
Daniel Jasperde0328a2013-08-16 11:20:30 +000025namespace clang {
26namespace format {
27
28// Returns the length of everything up to the first possible line break after
29// the ), ], } or > matching \c Tok.
30static unsigned getLengthToMatchingParen(const FormatToken &Tok) {
Craig Topper2145bc02014-05-09 08:15:10 +000031 if (!Tok.MatchingParen)
Daniel Jasperde0328a2013-08-16 11:20:30 +000032 return 0;
33 FormatToken *End = Tok.MatchingParen;
34 while (End->Next && !End->Next->CanBreakBefore) {
35 End = End->Next;
36 }
37 return End->TotalLength - Tok.TotalLength + 1;
38}
39
Daniel Jasper801cdb22016-01-05 13:03:59 +000040static unsigned getLengthToNextOperator(const FormatToken &Tok) {
41 if (!Tok.NextOperator)
42 return 0;
43 return Tok.NextOperator->TotalLength - Tok.TotalLength;
44}
45
Daniel Jasper4c6e0052013-08-27 14:24:43 +000046// Returns \c true if \c Tok is the "." or "->" of a call and starts the next
47// segment of a builder type call.
48static bool startsSegmentOfBuilderTypeCall(const FormatToken &Tok) {
49 return Tok.isMemberAccess() && Tok.Previous && Tok.Previous->closesScope();
50}
51
Daniel Jasperec01cd62013-10-08 05:11:18 +000052// Returns \c true if \c Current starts a new parameter.
53static bool startsNextParameter(const FormatToken &Current,
54 const FormatStyle &Style) {
55 const FormatToken &Previous = *Current.Previous;
Daniel Jaspera98b7b02014-11-25 10:05:17 +000056 if (Current.is(TT_CtorInitializerComma) &&
Francois Ferranda6b6d512017-05-24 11:36:58 +000057 Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma)
Daniel Jasperec01cd62013-10-08 05:11:18 +000058 return true;
Krasimir Georgievff747be2017-06-27 13:43:07 +000059 if (Style.Language == FormatStyle::LK_Proto && Current.is(TT_SelectorName))
60 return true;
Daniel Jasperec01cd62013-10-08 05:11:18 +000061 return Previous.is(tok::comma) && !Current.isTrailingComment() &&
Andi-Bogdan Postelnicu0ef8ee12017-03-10 15:10:37 +000062 ((Previous.isNot(TT_CtorInitializerComma) ||
Francois Ferranda6b6d512017-05-24 11:36:58 +000063 Style.BreakConstructorInitializers !=
64 FormatStyle::BCIS_BeforeComma) &&
Andi-Bogdan Postelnicu0ef8ee12017-03-10 15:10:37 +000065 (Previous.isNot(TT_InheritanceComma) ||
Francois Ferranda6b6d512017-05-24 11:36:58 +000066 !Style.BreakBeforeInheritanceComma));
Daniel Jasperec01cd62013-10-08 05:11:18 +000067}
68
Krasimir Georgiev26b144c2017-07-03 15:05:14 +000069static bool opensProtoMessageField(const FormatToken &LessTok,
70 const FormatStyle &Style) {
71 if (LessTok.isNot(tok::less))
72 return false;
73 return Style.Language == FormatStyle::LK_TextProto ||
74 (Style.Language == FormatStyle::LK_Proto &&
75 (LessTok.NestingLevel > 0 ||
76 (LessTok.Previous && LessTok.Previous->is(tok::equal))));
77}
78
Daniel Jasperde0328a2013-08-16 11:20:30 +000079ContinuationIndenter::ContinuationIndenter(const FormatStyle &Style,
Daniel Jasperd0ec0d62014-11-04 12:41:02 +000080 const AdditionalKeywords &Keywords,
Eric Liu635423e2016-04-28 07:52:03 +000081 const SourceManager &SourceMgr,
Daniel Jasperde0328a2013-08-16 11:20:30 +000082 WhitespaceManager &Whitespaces,
83 encoding::Encoding Encoding,
84 bool BinPackInconclusiveFunctions)
Daniel Jasperd0ec0d62014-11-04 12:41:02 +000085 : Style(Style), Keywords(Keywords), SourceMgr(SourceMgr),
86 Whitespaces(Whitespaces), Encoding(Encoding),
Alexander Kornienkoce9161a2014-01-02 15:13:14 +000087 BinPackInconclusiveFunctions(BinPackInconclusiveFunctions),
88 CommentPragmasRegex(Style.CommentPragmas) {}
Daniel Jasperde0328a2013-08-16 11:20:30 +000089
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +000090LineState ContinuationIndenter::getInitialState(unsigned FirstIndent,
Daniel Jasper1c5d9df2013-09-06 07:54:20 +000091 const AnnotatedLine *Line,
92 bool DryRun) {
Daniel Jasperde0328a2013-08-16 11:20:30 +000093 LineState State;
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +000094 State.FirstIndent = FirstIndent;
Daniel Jasperde0328a2013-08-16 11:20:30 +000095 State.Column = FirstIndent;
Krasimir Georgievad47c902017-08-30 14:34:57 +000096 // With preprocessor directive indentation, the line starts on column 0
97 // since it's indented after the hash, but FirstIndent is set to the
98 // preprocessor indent.
99 if (Style.IndentPPDirectives == FormatStyle::PPDIS_AfterHash &&
100 (Line->Type == LT_PreprocessorDirective ||
101 Line->Type == LT_ImportStatement))
102 State.Column = 0;
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +0000103 State.Line = Line;
104 State.NextToken = Line->First;
Daniel Jasper7d42f3f2017-01-31 11:25:01 +0000105 State.Stack.push_back(ParenState(FirstIndent, FirstIndent,
Daniel Jasperde0328a2013-08-16 11:20:30 +0000106 /*AvoidBinPacking=*/false,
107 /*NoLineBreak=*/false));
108 State.LineContainsContinuedForLoopSection = false;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000109 State.StartOfStringLiteral = 0;
Daniel Jasper05cd5862014-05-08 12:21:30 +0000110 State.StartOfLineLevel = 0;
111 State.LowestLevelOnLine = 0;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000112 State.IgnoreStackForComparison = false;
113
Krasimir Georgiev26b144c2017-07-03 15:05:14 +0000114 if (Style.Language == FormatStyle::LK_TextProto) {
115 // We need this in order to deal with the bin packing of text fields at
116 // global scope.
117 State.Stack.back().AvoidBinPacking = true;
118 State.Stack.back().BreakBeforeParameter = true;
119 }
120
Daniel Jasperde0328a2013-08-16 11:20:30 +0000121 // The first token has already been indented and thus consumed.
Daniel Jasper1c5d9df2013-09-06 07:54:20 +0000122 moveStateToNextToken(State, DryRun, /*Newline=*/false);
Daniel Jasperde0328a2013-08-16 11:20:30 +0000123 return State;
124}
125
126bool ContinuationIndenter::canBreak(const LineState &State) {
127 const FormatToken &Current = *State.NextToken;
128 const FormatToken &Previous = *Current.Previous;
129 assert(&Previous == Current.Previous);
Manuel Klimek89628f62017-09-20 09:51:03 +0000130 if (!Current.CanBreakBefore && !(State.Stack.back().BreakBeforeClosingBrace &&
131 Current.closesBlockOrBlockTypeList(Style)))
Daniel Jasperde0328a2013-08-16 11:20:30 +0000132 return false;
133 // The opening "{" of a braced list has to be on the same line as the first
134 // element if it is nested in another braced init list or function call.
135 if (!Current.MustBreakBefore && Previous.is(tok::l_brace) &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000136 Previous.isNot(TT_DictLiteral) && Previous.BlockKind == BK_BracedInit &&
Daniel Jasperb05a81d2014-05-09 13:11:16 +0000137 Previous.Previous &&
Daniel Jasperde0328a2013-08-16 11:20:30 +0000138 Previous.Previous->isOneOf(tok::l_brace, tok::l_paren, tok::comma))
139 return false;
140 // This prevents breaks like:
141 // ...
142 // SomeParameter, OtherParameter).DoSomething(
143 // ...
144 // As they hide "DoSomething" and are generally bad for readability.
Daniel Jasper8f59ae52014-03-11 11:03:26 +0000145 if (Previous.opensScope() && Previous.isNot(tok::l_brace) &&
Daniel Jasperfcfac102014-07-15 09:00:34 +0000146 State.LowestLevelOnLine < State.StartOfLineLevel &&
147 State.LowestLevelOnLine < Current.NestingLevel)
Daniel Jasperde0328a2013-08-16 11:20:30 +0000148 return false;
Daniel Jasper4c6e0052013-08-27 14:24:43 +0000149 if (Current.isMemberAccess() && State.Stack.back().ContainsUnwrappedBuilder)
150 return false;
Daniel Jasper114a2bc2014-06-03 12:02:45 +0000151
152 // Don't create a 'hanging' indent if there are multiple blocks in a single
153 // statement.
Daniel Jasper4b444492014-11-21 13:38:53 +0000154 if (Previous.is(tok::l_brace) && State.Stack.size() > 1 &&
155 State.Stack[State.Stack.size() - 2].NestedBlockInlined &&
Daniel Jasper114a2bc2014-06-03 12:02:45 +0000156 State.Stack[State.Stack.size() - 2].HasMultipleNestedBlocks)
157 return false;
158
Daniel Jaspere068ac72014-10-27 17:13:59 +0000159 // Don't break after very short return types (e.g. "void") as that is often
160 // unexpected.
Zachary Turner448592e2015-12-18 22:20:15 +0000161 if (Current.is(TT_FunctionDeclarationName) && State.Column < 6) {
162 if (Style.AlwaysBreakAfterReturnType == FormatStyle::RTBS_None)
163 return false;
164 }
Daniel Jaspere068ac72014-10-27 17:13:59 +0000165
Daniel Jasper240527c2017-01-16 13:13:15 +0000166 // If binary operators are moved to the next line (including commas for some
167 // styles of constructor initializers), that's always ok.
168 if (!Current.isOneOf(TT_BinaryOperator, tok::comma) &&
169 State.Stack.back().NoLineBreakInOperand)
170 return false;
171
Daniel Jasperde0328a2013-08-16 11:20:30 +0000172 return !State.Stack.back().NoLineBreak;
173}
174
175bool ContinuationIndenter::mustBreak(const LineState &State) {
176 const FormatToken &Current = *State.NextToken;
177 const FormatToken &Previous = *Current.Previous;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000178 if (Current.MustBreakBefore || Current.is(TT_InlineASMColon))
Daniel Jasperde0328a2013-08-16 11:20:30 +0000179 return true;
Daniel Jasper1db6c382013-10-22 15:30:28 +0000180 if (State.Stack.back().BreakBeforeClosingBrace &&
Daniel Jasperbd73bcf2015-10-27 13:42:08 +0000181 Current.closesBlockOrBlockTypeList(Style))
Daniel Jasperde0328a2013-08-16 11:20:30 +0000182 return true;
183 if (Previous.is(tok::semi) && State.LineContainsContinuedForLoopSection)
184 return true;
Daniel Jasperec01cd62013-10-08 05:11:18 +0000185 if ((startsNextParameter(Current, Style) || Previous.is(tok::semi) ||
Daniel Jasper00693b082016-01-09 15:56:47 +0000186 (Previous.is(TT_TemplateCloser) && Current.is(TT_StartOfName) &&
Daniel Jasper1dbc2102017-03-31 13:30:24 +0000187 Style.isCpp() &&
Daniel Jasper06ca0fc2016-01-11 11:01:05 +0000188 // FIXME: This is a temporary workaround for the case where clang-format
189 // sets BreakBeforeParameter to avoid bin packing and this creates a
190 // completely unnecessary line break after a template type that isn't
191 // line-wrapped.
192 (Previous.NestingLevel == 1 || Style.BinPackParameters)) ||
Daniel Jasper3e0dcc22015-05-27 05:37:40 +0000193 (Style.BreakBeforeTernaryOperators && Current.is(TT_ConditionalExpr) &&
194 Previous.isNot(tok::question)) ||
Daniel Jasper165b29e2013-11-08 00:57:11 +0000195 (!Style.BreakBeforeTernaryOperators &&
Daniel Jasper3e0dcc22015-05-27 05:37:40 +0000196 Previous.is(TT_ConditionalExpr))) &&
Daniel Jasperde0328a2013-08-16 11:20:30 +0000197 State.Stack.back().BreakBeforeParameter && !Current.isTrailingComment() &&
198 !Current.isOneOf(tok::r_paren, tok::r_brace))
199 return true;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000200 if (((Previous.is(TT_DictLiteral) && Previous.is(tok::l_brace)) ||
Daniel Jasperccff4d12016-01-04 07:27:33 +0000201 (Previous.is(TT_ArrayInitializerLSquare) &&
Krasimir Georgiev26b144c2017-07-03 15:05:14 +0000202 Previous.ParameterCount > 1) ||
203 opensProtoMessageField(Previous, Style)) &&
Daniel Jasperdb8804b2014-04-14 12:11:07 +0000204 Style.ColumnLimit > 0 &&
Daniel Jasper199d0c92015-06-02 15:14:21 +0000205 getLengthToMatchingParen(Previous) + State.Column - 1 >
206 getColumnLimit(State))
Daniel Jasperd489dd32013-10-20 16:45:46 +0000207 return true;
Francois Ferranda6b6d512017-05-24 11:36:58 +0000208
209 const FormatToken &BreakConstructorInitializersToken =
210 Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon
211 ? Previous
212 : Current;
213 if (BreakConstructorInitializersToken.is(TT_CtorInitializerColon) &&
214 (State.Column + State.Line->Last->TotalLength - Previous.TotalLength >
Daniel Jasper7b259cd2015-08-27 11:59:31 +0000215 getColumnLimit(State) ||
216 State.Stack.back().BreakBeforeParameter) &&
Francois Ferranda6b6d512017-05-24 11:36:58 +0000217 (Style.AllowShortFunctionsOnASingleLine != FormatStyle::SFS_All ||
218 Style.BreakConstructorInitializers != FormatStyle::BCIS_BeforeColon ||
219 Style.ColumnLimit != 0))
Daniel Jasper5d2587d2014-03-27 16:14:13 +0000220 return true;
Francois Ferranda6b6d512017-05-24 11:36:58 +0000221
Daniel Jasperfd36f0b2016-11-12 07:38:22 +0000222 if (Current.is(TT_ObjCMethodExpr) && !Previous.is(TT_SelectorName) &&
223 State.Line->startsWith(TT_ObjCMethodSpecifier))
224 return true;
Daniel Jasper2746a302015-05-06 13:13:03 +0000225 if (Current.is(TT_SelectorName) && State.Stack.back().ObjCSelectorNameFound &&
226 State.Stack.back().BreakBeforeParameter)
227 return true;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000228
Daniel Jasper2aaedd32015-06-18 09:12:47 +0000229 unsigned NewLineColumn = getNewLineColumn(State);
Daniel Jaspera3cd21642016-01-14 13:36:46 +0000230 if (Current.isMemberAccess() && Style.ColumnLimit != 0 &&
Daniel Jasper28024562016-01-11 11:00:58 +0000231 State.Column + getLengthToNextOperator(Current) > Style.ColumnLimit &&
232 (State.Column > NewLineColumn ||
233 Current.NestingLevel < State.StartOfLineLevel))
234 return true;
235
Daniel Jaspere61f9f92017-01-13 23:18:16 +0000236 if (startsSegmentOfBuilderTypeCall(Current) &&
237 (State.Stack.back().CallContinuation != 0 ||
Daniel Jasper51c868e2017-01-30 07:08:40 +0000238 State.Stack.back().BreakBeforeParameter) &&
239 // JavaScript is treated different here as there is a frequent pattern:
240 // SomeFunction(function() {
241 // ...
242 // }.bind(...));
243 // FIXME: We should find a more generic solution to this problem.
Martin Probstb2f06ea2017-05-29 07:50:52 +0000244 !(State.Column <= NewLineColumn &&
Daniel Jasper51c868e2017-01-30 07:08:40 +0000245 Style.Language == FormatStyle::LK_JavaScript))
Daniel Jaspere61f9f92017-01-13 23:18:16 +0000246 return true;
247
Daniel Jasper411af722016-01-05 16:10:39 +0000248 if (State.Column <= NewLineColumn)
Daniel Jasper5d2587d2014-03-27 16:14:13 +0000249 return false;
Daniel Jasper173504e2015-05-10 21:15:07 +0000250
Daniel Jasper2aaedd32015-06-18 09:12:47 +0000251 if (Style.AlwaysBreakBeforeMultilineStrings &&
Daniel Jasper1bf729c2015-06-18 16:05:17 +0000252 (NewLineColumn == State.FirstIndent + Style.ContinuationIndentWidth ||
Daniel Jasper9fb676a2015-06-19 10:32:28 +0000253 Previous.is(tok::comma) || Current.NestingLevel < 2) &&
Daniel Jasper2aaedd32015-06-18 09:12:47 +0000254 !Previous.isOneOf(tok::kw_return, tok::lessless, tok::at) &&
255 !Previous.isOneOf(TT_InlineASMColon, TT_ConditionalExpr) &&
256 nextIsMultilineString(State))
257 return true;
258
Daniel Jasper173504e2015-05-10 21:15:07 +0000259 // Using CanBreakBefore here and below takes care of the decision whether the
260 // current style uses wrapping before or after operators for the given
261 // operator.
262 if (Previous.is(TT_BinaryOperator) && Current.CanBreakBefore) {
Daniel Jasperde0328a2013-08-16 11:20:30 +0000263 // If we need to break somewhere inside the LHS of a binary expression, we
264 // should also break after the operator. Otherwise, the formatting would
265 // hide the operator precedence, e.g. in:
266 // if (aaaaaaaaaaaaaa ==
267 // bbbbbbbbbbbbbb && c) {..
268 // For comparisons, we only apply this rule, if the LHS is a binary
269 // expression itself as otherwise, the line breaks seem superfluous.
270 // We need special cases for ">>" which we have split into two ">" while
271 // lexing in order to make template parsing easier.
Daniel Jasperde0328a2013-08-16 11:20:30 +0000272 bool IsComparison = (Previous.getPrecedence() == prec::Relational ||
273 Previous.getPrecedence() == prec::Equality) &&
274 Previous.Previous &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000275 Previous.Previous->isNot(TT_BinaryOperator); // For >>.
Daniel Jasperde0328a2013-08-16 11:20:30 +0000276 bool LHSIsBinaryExpr =
Daniel Jasper562ecd42013-09-06 08:08:14 +0000277 Previous.Previous && Previous.Previous->EndsBinaryExpression;
Daniel Jasper173504e2015-05-10 21:15:07 +0000278 if ((!IsComparison || LHSIsBinaryExpr) && !Current.isTrailingComment() &&
Daniel Jasperde0328a2013-08-16 11:20:30 +0000279 Previous.getPrecedence() != prec::Assignment &&
280 State.Stack.back().BreakBeforeParameter)
281 return true;
Daniel Jasper173504e2015-05-10 21:15:07 +0000282 } else if (Current.is(TT_BinaryOperator) && Current.CanBreakBefore &&
283 State.Stack.back().BreakBeforeParameter) {
284 return true;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000285 }
286
287 // Same as above, but for the first "<<" operator.
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000288 if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator) &&
Alexander Kornienko86b2dfd2014-03-06 15:13:08 +0000289 State.Stack.back().BreakBeforeParameter &&
Daniel Jasperde0328a2013-08-16 11:20:30 +0000290 State.Stack.back().FirstLessLess == 0)
291 return true;
292
Daniel Jasper211e1322014-12-08 20:08:04 +0000293 if (Current.NestingLevel == 0 && !Current.isTrailingComment()) {
Daniel Jasperf090f032015-05-18 09:47:22 +0000294 // Always break after "template <...>" and leading annotations. This is only
295 // for cases where the entire line does not fit on a single line as a
296 // different LineFormatter would be used otherwise.
Daniel Jasper211e1322014-12-08 20:08:04 +0000297 if (Previous.ClosesTemplateDeclaration)
298 return true;
Daniel Jasper47bbda02015-05-18 13:47:23 +0000299 if (Previous.is(TT_FunctionAnnotationRParen))
Daniel Jasperf090f032015-05-18 09:47:22 +0000300 return true;
Nico Weberbeb03932015-01-09 23:25:06 +0000301 if (Previous.is(TT_LeadingJavaAnnotation) && Current.isNot(tok::l_paren) &&
302 Current.isNot(TT_LeadingJavaAnnotation))
Daniel Jasper211e1322014-12-08 20:08:04 +0000303 return true;
304 }
Daniel Jasperde0328a2013-08-16 11:20:30 +0000305
Daniel Jasper4355e7f2014-07-09 07:50:33 +0000306 // If the return type spans multiple lines, wrap before the function name.
Birunthan Mohanathas525579d2015-07-15 19:11:58 +0000307 if ((Current.is(TT_FunctionDeclarationName) ||
308 (Current.is(tok::kw_operator) && !Previous.is(tok::coloncolon))) &&
Daniel Jasper0c9772e2016-02-05 14:17:16 +0000309 !Previous.is(tok::kw_template) && State.Stack.back().BreakBeforeParameter)
Daniel Jasperde0328a2013-08-16 11:20:30 +0000310 return true;
Daniel Jasper4355e7f2014-07-09 07:50:33 +0000311
Daniel Jasper96972812014-01-05 12:38:10 +0000312 // The following could be precomputed as they do not depend on the state.
313 // However, as they should take effect only if the UnwrappedLine does not fit
314 // into the ColumnLimit, they are checked here in the ContinuationIndenter.
Daniel Jasper35995672014-04-29 14:05:20 +0000315 if (Style.ColumnLimit != 0 && Previous.BlockKind == BK_Block &&
316 Previous.is(tok::l_brace) && !Current.isOneOf(tok::r_brace, tok::comment))
Daniel Jasper96972812014-01-05 12:38:10 +0000317 return true;
Daniel Jasper96972812014-01-05 12:38:10 +0000318
Daniel Jasper0a589412016-01-05 13:06:27 +0000319 if (Current.is(tok::lessless) &&
320 ((Previous.is(tok::identifier) && Previous.TokenText == "endl") ||
321 (Previous.Tok.isLiteral() && (Previous.TokenText.endswith("\\n\"") ||
322 Previous.TokenText == "\'\\n\'"))))
Daniel Jasper69963122015-02-17 10:05:15 +0000323 return true;
324
Daniel Jasperde0328a2013-08-16 11:20:30 +0000325 return false;
326}
327
328unsigned ContinuationIndenter::addTokenToState(LineState &State, bool Newline,
Daniel Jasper8de9ed02013-08-22 15:00:41 +0000329 bool DryRun,
330 unsigned ExtraSpaces) {
Daniel Jasperde0328a2013-08-16 11:20:30 +0000331 const FormatToken &Current = *State.NextToken;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000332
Manuel Klimek819788d2014-03-18 11:22:45 +0000333 assert(!State.Stack.empty());
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000334 if ((Current.is(TT_ImplicitStringLiteral) &&
Craig Topper2145bc02014-05-09 08:15:10 +0000335 (Current.Previous->Tok.getIdentifierInfo() == nullptr ||
Daniel Jasper98857842013-10-30 13:54:53 +0000336 Current.Previous->Tok.getIdentifierInfo()->getPPKeywordID() ==
337 tok::pp_not_keyword))) {
Daniel Jasper30526e72015-01-31 07:05:46 +0000338 unsigned EndColumn =
339 SourceMgr.getSpellingColumnNumber(Current.WhitespaceRange.getEnd());
340 if (Current.LastNewlineOffset != 0) {
341 // If there is a newline within this token, the final column will solely
342 // determined by the current end column.
343 State.Column = EndColumn;
344 } else {
345 unsigned StartColumn =
346 SourceMgr.getSpellingColumnNumber(Current.WhitespaceRange.getBegin());
347 assert(EndColumn >= StartColumn);
348 State.Column += EndColumn - StartColumn;
349 }
Daniel Jasper240dfda2014-03-31 14:23:49 +0000350 moveStateToNextToken(State, DryRun, /*Newline=*/false);
Daniel Jasperde0328a2013-08-16 11:20:30 +0000351 return 0;
352 }
353
Alexander Kornienko1f803962013-10-01 14:41:18 +0000354 unsigned Penalty = 0;
355 if (Newline)
356 Penalty = addTokenOnNewLine(State, DryRun);
357 else
Daniel Jasper48437ce2013-11-20 14:54:39 +0000358 addTokenOnCurrentLine(State, DryRun, ExtraSpaces);
Alexander Kornienko1f803962013-10-01 14:41:18 +0000359
360 return moveStateToNextToken(State, DryRun, Newline) + Penalty;
361}
362
Daniel Jasper48437ce2013-11-20 14:54:39 +0000363void ContinuationIndenter::addTokenOnCurrentLine(LineState &State, bool DryRun,
364 unsigned ExtraSpaces) {
Manuel Klimek71814b42013-10-11 21:25:45 +0000365 FormatToken &Current = *State.NextToken;
Alexander Kornienko1f803962013-10-01 14:41:18 +0000366 const FormatToken &Previous = *State.NextToken->Previous;
367 if (Current.is(tok::equal) &&
Daniel Jasper05cd5862014-05-08 12:21:30 +0000368 (State.Line->First->is(tok::kw_for) || Current.NestingLevel == 0) &&
Alexander Kornienko1f803962013-10-01 14:41:18 +0000369 State.Stack.back().VariablePos == 0) {
370 State.Stack.back().VariablePos = State.Column;
371 // Move over * and & if they are bound to the variable name.
372 const FormatToken *Tok = &Previous;
373 while (Tok && State.Stack.back().VariablePos >= Tok->ColumnWidth) {
374 State.Stack.back().VariablePos -= Tok->ColumnWidth;
375 if (Tok->SpacesRequiredBefore != 0)
376 break;
377 Tok = Tok->Previous;
378 }
379 if (Previous.PartOfMultiVariableDeclStmt)
380 State.Stack.back().LastSpace = State.Stack.back().VariablePos;
381 }
382
383 unsigned Spaces = Current.SpacesRequiredBefore + ExtraSpaces;
384
Krasimir Georgievad47c902017-08-30 14:34:57 +0000385 // Indent preprocessor directives after the hash if required.
386 int PPColumnCorrection = 0;
387 if (Style.IndentPPDirectives == FormatStyle::PPDIS_AfterHash &&
388 Previous.is(tok::hash) && State.FirstIndent > 0 &&
389 (State.Line->Type == LT_PreprocessorDirective ||
390 State.Line->Type == LT_ImportStatement)) {
391 Spaces += State.FirstIndent;
392
393 // For preprocessor indent with tabs, State.Column will be 1 because of the
394 // hash. This causes second-level indents onward to have an extra space
395 // after the tabs. We avoid this misalignment by subtracting 1 from the
396 // column value passed to replaceWhitespace().
397 if (Style.UseTab != FormatStyle::UT_Never)
398 PPColumnCorrection = -1;
399 }
400
Alexander Kornienko1f803962013-10-01 14:41:18 +0000401 if (!DryRun)
Daniel Jasper7d42f3f2017-01-31 11:25:01 +0000402 Whitespaces.replaceWhitespace(Current, /*Newlines=*/0, Spaces,
Krasimir Georgievad47c902017-08-30 14:34:57 +0000403 State.Column + Spaces + PPColumnCorrection);
Alexander Kornienko1f803962013-10-01 14:41:18 +0000404
Andi-Bogdan Postelnicu0ef8ee12017-03-10 15:10:37 +0000405 // If "BreakBeforeInheritanceComma" mode, don't break within the inheritance
406 // declaration unless there is multiple inheritance.
407 if (Style.BreakBeforeInheritanceComma && Current.is(TT_InheritanceColon))
408 State.Stack.back().NoLineBreak = true;
409
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000410 if (Current.is(TT_SelectorName) &&
Daniel Jasperb88b25f2013-12-23 07:29:06 +0000411 !State.Stack.back().ObjCSelectorNameFound) {
Daniel Jasper06a26952016-01-04 07:29:07 +0000412 unsigned MinIndent =
413 std::max(State.FirstIndent + Style.ContinuationIndentWidth,
414 State.Stack.back().Indent);
415 unsigned FirstColonPos = State.Column + Spaces + Current.ColumnWidth;
Daniel Jasperb88b25f2013-12-23 07:29:06 +0000416 if (Current.LongestObjCSelectorName == 0)
417 State.Stack.back().AlignColons = false;
Daniel Jasper06a26952016-01-04 07:29:07 +0000418 else if (MinIndent + Current.LongestObjCSelectorName > FirstColonPos)
419 State.Stack.back().ColonPos = MinIndent + Current.LongestObjCSelectorName;
Alexander Kornienko1f803962013-10-01 14:41:18 +0000420 else
Daniel Jasper06a26952016-01-04 07:29:07 +0000421 State.Stack.back().ColonPos = FirstColonPos;
Alexander Kornienko1f803962013-10-01 14:41:18 +0000422 }
423
Daniel Jasper6501f7e2015-10-27 12:38:37 +0000424 // In "AlwaysBreak" mode, enforce wrapping directly after the parenthesis by
425 // disallowing any further line breaks if there is no line break after the
426 // opening parenthesis. Don't break if it doesn't conserve columns.
427 if (Style.AlignAfterOpenBracket == FormatStyle::BAS_AlwaysBreak &&
Daniel Jasperb618a982016-02-02 10:28:11 +0000428 Previous.isOneOf(tok::l_paren, TT_TemplateOpener, tok::l_square) &&
429 State.Column > getNewLineColumn(State) &&
Manuel Klimek89628f62017-09-20 09:51:03 +0000430 (!Previous.Previous || !Previous.Previous->isOneOf(
431 tok::kw_for, tok::kw_while, tok::kw_switch)) &&
Daniel Jasper710f8492016-03-17 12:00:22 +0000432 // Don't do this for simple (no expressions) one-argument function calls
433 // as that feels like needlessly wasting whitespace, e.g.:
434 //
435 // caaaaaaaaaaaall(
436 // caaaaaaaaaaaall(
437 // caaaaaaaaaaaall(
438 // caaaaaaaaaaaaaaaaaaaaaaall(aaaaaaaaaaaaaa, aaaaaaaaa))));
439 Current.FakeLParens.size() > 0 &&
440 Current.FakeLParens.back() > prec::Unknown)
Daniel Jasper6501f7e2015-10-27 12:38:37 +0000441 State.Stack.back().NoLineBreak = true;
Daniel Jasper98e0b122017-02-20 14:51:16 +0000442 if (Previous.is(TT_TemplateString) && Previous.opensScope())
443 State.Stack.back().NoLineBreak = true;
Daniel Jasper6501f7e2015-10-27 12:38:37 +0000444
445 if (Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign &&
446 Previous.opensScope() && Previous.isNot(TT_ObjCMethodExpr) &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000447 (Current.isNot(TT_LineComment) || Previous.BlockKind == BK_BracedInit))
Alexander Kornienko1f803962013-10-01 14:41:18 +0000448 State.Stack.back().Indent = State.Column + Spaces;
Daniel Jasperec01cd62013-10-08 05:11:18 +0000449 if (State.Stack.back().AvoidBinPacking && startsNextParameter(Current, Style))
Alexander Kornienko1f803962013-10-01 14:41:18 +0000450 State.Stack.back().NoLineBreak = true;
Daniel Jasper775954b2015-04-24 10:08:09 +0000451 if (startsSegmentOfBuilderTypeCall(Current) &&
452 State.Column > getNewLineColumn(State))
Alexander Kornienko1f803962013-10-01 14:41:18 +0000453 State.Stack.back().ContainsUnwrappedBuilder = true;
454
Daniel Jasper6f2b88a2015-06-05 13:18:09 +0000455 if (Current.is(TT_LambdaArrow) && Style.Language == FormatStyle::LK_Java)
Daniel Jasper05cd9292015-03-26 18:46:28 +0000456 State.Stack.back().NoLineBreak = true;
Daniel Jasperd6f17d82014-09-12 16:35:28 +0000457 if (Current.isMemberAccess() && Previous.is(tok::r_paren) &&
458 (Previous.MatchingParen &&
Daniel Jasper98e0b122017-02-20 14:51:16 +0000459 (Previous.TotalLength - Previous.MatchingParen->TotalLength > 10)))
Daniel Jasperd6f17d82014-09-12 16:35:28 +0000460 // If there is a function call with long parameters, break before trailing
461 // calls. This prevents things like:
462 // EXPECT_CALL(SomeLongParameter).Times(
463 // 2);
464 // We don't want to do this for short parameters as they can just be
465 // indexes.
466 State.Stack.back().NoLineBreak = true;
Daniel Jasperd6f17d82014-09-12 16:35:28 +0000467
Daniel Jasper240527c2017-01-16 13:13:15 +0000468 // Don't allow the RHS of an operator to be split over multiple lines unless
469 // there is a line-break right after the operator.
470 // Exclude relational operators, as there, it is always more desirable to
471 // have the LHS 'left' of the RHS.
472 const FormatToken *P = Current.getPreviousNonComment();
473 if (!Current.is(tok::comment) && P &&
474 (P->isOneOf(TT_BinaryOperator, tok::comma) ||
475 (P->is(TT_ConditionalExpr) && P->is(tok::colon))) &&
476 !P->isOneOf(TT_OverloadedOperator, TT_CtorInitializerComma) &&
477 P->getPrecedence() != prec::Assignment &&
478 P->getPrecedence() != prec::Relational) {
479 bool BreakBeforeOperator =
Daniel Jasperb1270392017-02-01 23:27:37 +0000480 P->MustBreakBefore || P->is(tok::lessless) ||
Daniel Jasper240527c2017-01-16 13:13:15 +0000481 (P->is(TT_BinaryOperator) &&
482 Style.BreakBeforeBinaryOperators != FormatStyle::BOS_None) ||
483 (P->is(TT_ConditionalExpr) && Style.BreakBeforeTernaryOperators);
Daniel Jasper21f7dea2017-02-01 09:23:39 +0000484 // Don't do this if there are only two operands. In these cases, there is
485 // always a nice vertical separation between them and the extra line break
486 // does not help.
487 bool HasTwoOperands =
488 P->OperatorIndex == 0 && !P->NextOperator && !P->is(TT_ConditionalExpr);
Daniel Jasperc3aa05c2017-02-02 08:30:21 +0000489 if ((!BreakBeforeOperator && !(HasTwoOperands && Style.AlignOperands)) ||
Daniel Jasper240527c2017-01-16 13:13:15 +0000490 (!State.Stack.back().LastOperatorWrapped && BreakBeforeOperator))
491 State.Stack.back().NoLineBreakInOperand = true;
492 }
493
Alexander Kornienko1f803962013-10-01 14:41:18 +0000494 State.Column += Spaces;
Daniel Jasper8acf8222014-05-07 09:23:05 +0000495 if (Current.isNot(tok::comment) && Previous.is(tok::l_paren) &&
Daniel Jasper11a0ac62014-12-12 09:40:58 +0000496 Previous.Previous &&
Daniel Jasper6a7d5a72017-06-19 07:40:49 +0000497 (Previous.Previous->isOneOf(tok::kw_if, tok::kw_for) ||
498 Previous.Previous->endsSequence(tok::kw_constexpr, tok::kw_if))) {
Alexander Kornienko1f803962013-10-01 14:41:18 +0000499 // Treat the condition inside an if as if it was a second function
Daniel Jasper6633ab82013-10-18 10:38:14 +0000500 // parameter, i.e. let nested calls have a continuation indent.
Daniel Jasper8acf8222014-05-07 09:23:05 +0000501 State.Stack.back().LastSpace = State.Column;
Daniel Jasper11a0ac62014-12-12 09:40:58 +0000502 State.Stack.back().NestedBlockIndent = State.Column;
503 } else if (!Current.isOneOf(tok::comment, tok::caret) &&
Daniel Jasper804a2762016-01-09 15:56:40 +0000504 ((Previous.is(tok::comma) &&
505 !Previous.is(TT_OverloadedOperator)) ||
Daniel Jasper11a0ac62014-12-12 09:40:58 +0000506 (Previous.is(tok::colon) && Previous.is(TT_ObjCMethodExpr)))) {
Alexander Kornienko1f803962013-10-01 14:41:18 +0000507 State.Stack.back().LastSpace = State.Column;
Francois Ferranda6b6d512017-05-24 11:36:58 +0000508 } else if (Previous.is(TT_CtorInitializerColon) &&
509 Style.BreakConstructorInitializers ==
510 FormatStyle::BCIS_AfterColon) {
511 State.Stack.back().Indent = State.Column;
512 State.Stack.back().LastSpace = State.Column;
Daniel Jasper11a0ac62014-12-12 09:40:58 +0000513 } else if ((Previous.isOneOf(TT_BinaryOperator, TT_ConditionalExpr,
514 TT_CtorInitializerColon)) &&
515 ((Previous.getPrecedence() != prec::Assignment &&
516 (Previous.isNot(tok::lessless) || Previous.OperatorIndex != 0 ||
Daniel Jasper00492f92016-01-05 13:03:50 +0000517 Previous.NextOperator)) ||
Daniel Jasper11a0ac62014-12-12 09:40:58 +0000518 Current.StartsBinaryExpression)) {
Daniel Jasper602a7272016-02-11 13:15:14 +0000519 // Indent relative to the RHS of the expression unless this is a simple
520 // assignment without binary expression on the RHS. Also indent relative to
521 // unary operators and the colons of constructor initializers.
Alexander Kornienko1f803962013-10-01 14:41:18 +0000522 State.Stack.back().LastSpace = State.Column;
Daniel Jasper11a0ac62014-12-12 09:40:58 +0000523 } else if (Previous.is(TT_InheritanceColon)) {
Alexander Kornienko1f803962013-10-01 14:41:18 +0000524 State.Stack.back().Indent = State.Column;
Daniel Jasperf9a5e402013-10-08 16:24:07 +0000525 State.Stack.back().LastSpace = State.Column;
526 } else if (Previous.opensScope()) {
Alexander Kornienko1f803962013-10-01 14:41:18 +0000527 // If a function has a trailing call, indent all parameters from the
528 // opening parenthesis. This avoids confusing indents like:
529 // OuterFunction(InnerFunctionCall( // break
530 // ParameterToInnerFunction)) // break
531 // .SecondInnerFunctionCall();
532 bool HasTrailingCall = false;
533 if (Previous.MatchingParen) {
534 const FormatToken *Next = Previous.MatchingParen->getNextNonComment();
535 HasTrailingCall = Next && Next->isMemberAccess();
536 }
Daniel Jasperd97d5d52015-02-17 09:58:03 +0000537 if (HasTrailingCall && State.Stack.size() > 1 &&
Alexander Kornienko1f803962013-10-01 14:41:18 +0000538 State.Stack[State.Stack.size() - 2].CallContinuation == 0)
539 State.Stack.back().LastSpace = State.Column;
540 }
541}
542
543unsigned ContinuationIndenter::addTokenOnNewLine(LineState &State,
544 bool DryRun) {
Manuel Klimek71814b42013-10-11 21:25:45 +0000545 FormatToken &Current = *State.NextToken;
Alexander Kornienko1f803962013-10-01 14:41:18 +0000546 const FormatToken &Previous = *State.NextToken->Previous;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000547
Alexander Kornienko1f803962013-10-01 14:41:18 +0000548 // Extra penalty that needs to be added because of the way certain line
549 // breaks are chosen.
550 unsigned Penalty = 0;
Daniel Jasper8de9ed02013-08-22 15:00:41 +0000551
Daniel Jaspera0407742014-02-11 10:08:11 +0000552 const FormatToken *PreviousNonComment = Current.getPreviousNonComment();
553 const FormatToken *NextNonComment = Previous.getNextNonComment();
554 if (!NextNonComment)
555 NextNonComment = &Current;
Daniel Jasper05cd5862014-05-08 12:21:30 +0000556 // The first line break on any NestingLevel causes an extra penalty in order
Alexander Kornienko1f803962013-10-01 14:41:18 +0000557 // prefer similar line breaks.
558 if (!State.Stack.back().ContainsLineBreak)
559 Penalty += 15;
560 State.Stack.back().ContainsLineBreak = true;
Daniel Jasper8de9ed02013-08-22 15:00:41 +0000561
Alexander Kornienko1f803962013-10-01 14:41:18 +0000562 Penalty += State.NextToken->SplitPenalty;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000563
Alexander Kornienko1f803962013-10-01 14:41:18 +0000564 // Breaking before the first "<<" is generally not desirable if the LHS is
Daniel Jasper7aacf462016-12-19 11:14:23 +0000565 // short. Also always add the penalty if the LHS is split over multiple lines
Daniel Jasper2b7556e2014-04-03 12:00:27 +0000566 // to avoid unnecessary line breaks that just work around this penalty.
Daniel Jaspera0407742014-02-11 10:08:11 +0000567 if (NextNonComment->is(tok::lessless) &&
568 State.Stack.back().FirstLessLess == 0 &&
Daniel Jasper004177e2013-12-19 16:06:40 +0000569 (State.Column <= Style.ColumnLimit / 3 ||
Daniel Jasper48437ce2013-11-20 14:54:39 +0000570 State.Stack.back().BreakBeforeParameter))
Alexander Kornienko1f803962013-10-01 14:41:18 +0000571 Penalty += Style.PenaltyBreakFirstLessLess;
572
Daniel Jasper9f388d02014-03-27 14:33:30 +0000573 State.Column = getNewLineColumn(State);
Daniel Jaspered3f3952015-06-18 12:32:59 +0000574
575 // Indent nested blocks relative to this column, unless in a very specific
576 // JavaScript special case where:
577 //
578 // var loooooong_name =
579 // function() {
580 // // code
581 // }
582 //
Daniel Jasper87448c52016-06-13 07:48:45 +0000583 // is common and should be formatted like a free-standing function. The same
584 // goes for wrapping before the lambda return type arrow.
585 if (!Current.is(TT_LambdaArrow) &&
586 (Style.Language != FormatStyle::LK_JavaScript ||
587 Current.NestingLevel != 0 || !PreviousNonComment ||
588 !PreviousNonComment->is(tok::equal) ||
589 !Current.isOneOf(Keywords.kw_async, Keywords.kw_function)))
Daniel Jaspered3f3952015-06-18 12:32:59 +0000590 State.Stack.back().NestedBlockIndent = State.Column;
591
Daniel Jasper9f388d02014-03-27 14:33:30 +0000592 if (NextNonComment->isMemberAccess()) {
593 if (State.Stack.back().CallContinuation == 0)
Alexander Kornienko1f803962013-10-01 14:41:18 +0000594 State.Stack.back().CallContinuation = State.Column;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000595 } else if (NextNonComment->is(TT_SelectorName)) {
Daniel Jasperb88b25f2013-12-23 07:29:06 +0000596 if (!State.Stack.back().ObjCSelectorNameFound) {
Daniel Jaspera0407742014-02-11 10:08:11 +0000597 if (NextNonComment->LongestObjCSelectorName == 0) {
Daniel Jasperb88b25f2013-12-23 07:29:06 +0000598 State.Stack.back().AlignColons = false;
599 } else {
600 State.Stack.back().ColonPos =
Daniel Jaspera2a4d9c2015-05-13 09:38:25 +0000601 (Style.IndentWrappedFunctionNames
602 ? std::max(State.Stack.back().Indent,
603 State.FirstIndent + Style.ContinuationIndentWidth)
604 : State.Stack.back().Indent) +
605 NextNonComment->LongestObjCSelectorName;
Daniel Jasperb88b25f2013-12-23 07:29:06 +0000606 }
Daniel Jasper9f388d02014-03-27 14:33:30 +0000607 } else if (State.Stack.back().AlignColons &&
608 State.Stack.back().ColonPos <= NextNonComment->ColumnWidth) {
Daniel Jaspera0407742014-02-11 10:08:11 +0000609 State.Stack.back().ColonPos = State.Column + NextNonComment->ColumnWidth;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000610 }
Daniel Jasper1fd6f1f2014-03-17 14:32:47 +0000611 } else if (PreviousNonComment && PreviousNonComment->is(tok::colon) &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000612 PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) {
Daniel Jasperb88b25f2013-12-23 07:29:06 +0000613 // FIXME: This is hacky, find a better way. The problem is that in an ObjC
614 // method expression, the block should be aligned to the line starting it,
615 // e.g.:
616 // [aaaaaaaaaaaaaaa aaaaaaaaa: \\ break for some reason
617 // ^(int *i) {
618 // // ...
619 // }];
Daniel Jasper05cd5862014-05-08 12:21:30 +0000620 // Thus, we set LastSpace of the next higher NestingLevel, to which we move
Daniel Jasperb88b25f2013-12-23 07:29:06 +0000621 // when we consume all of the "}"'s FakeRParens at the "{".
Daniel Jasper9a26e772013-12-23 11:25:40 +0000622 if (State.Stack.size() > 1)
Daniel Jasper9f388d02014-03-27 14:33:30 +0000623 State.Stack[State.Stack.size() - 2].LastSpace =
624 std::max(State.Stack.back().LastSpace, State.Stack.back().Indent) +
625 Style.ContinuationIndentWidth;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000626 }
627
Daniel Jasper35e41222016-11-29 09:40:01 +0000628 if ((PreviousNonComment &&
629 PreviousNonComment->isOneOf(tok::comma, tok::semi) &&
Alexander Kornienko1f803962013-10-01 14:41:18 +0000630 !State.Stack.back().AvoidBinPacking) ||
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000631 Previous.is(TT_BinaryOperator))
Alexander Kornienko1f803962013-10-01 14:41:18 +0000632 State.Stack.back().BreakBeforeParameter = false;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000633 if (Previous.isOneOf(TT_TemplateCloser, TT_JavaAnnotation) &&
Daniel Jasper39af6cd2014-11-03 02:27:28 +0000634 Current.NestingLevel == 0)
Alexander Kornienko1f803962013-10-01 14:41:18 +0000635 State.Stack.back().BreakBeforeParameter = false;
Daniel Jaspera0407742014-02-11 10:08:11 +0000636 if (NextNonComment->is(tok::question) ||
Daniel Jasper165b29e2013-11-08 00:57:11 +0000637 (PreviousNonComment && PreviousNonComment->is(tok::question)))
638 State.Stack.back().BreakBeforeParameter = true;
Daniel Jasper5962fa82015-06-03 09:26:03 +0000639 if (Current.is(TT_BinaryOperator) && Current.CanBreakBefore)
640 State.Stack.back().BreakBeforeParameter = false;
Alexander Kornienko1f803962013-10-01 14:41:18 +0000641
642 if (!DryRun) {
Daniel Jaspera69ca9b2014-06-04 12:40:57 +0000643 unsigned Newlines = std::max(
644 1u, std::min(Current.NewlinesBefore, Style.MaxEmptyLinesToKeep + 1));
Krasimir Georgieva1c30932017-05-19 10:34:57 +0000645 bool ContinuePPDirective =
646 State.Line->InPPDirective && State.Line->Type != LT_ImportStatement;
Daniel Jasper7d42f3f2017-01-31 11:25:01 +0000647 Whitespaces.replaceWhitespace(Current, Newlines, State.Column, State.Column,
Krasimir Georgieva1c30932017-05-19 10:34:57 +0000648 ContinuePPDirective);
Alexander Kornienko1f803962013-10-01 14:41:18 +0000649 }
650
651 if (!Current.isTrailingComment())
652 State.Stack.back().LastSpace = State.Column;
Daniel Jasper602a7272016-02-11 13:15:14 +0000653 if (Current.is(tok::lessless))
654 // If we are breaking before a "<<", we always want to indent relative to
655 // RHS. This is necessary only for "<<", as we special-case it and don't
656 // always indent relative to the RHS.
657 State.Stack.back().LastSpace += 3; // 3 -> width of "<< ".
658
Daniel Jasper05cd5862014-05-08 12:21:30 +0000659 State.StartOfLineLevel = Current.NestingLevel;
660 State.LowestLevelOnLine = Current.NestingLevel;
Alexander Kornienko1f803962013-10-01 14:41:18 +0000661
662 // Any break on this level means that the parent level has been broken
663 // and we need to avoid bin packing there.
Daniel Jasper4b444492014-11-21 13:38:53 +0000664 bool NestedBlockSpecialCase =
Daniel Jasper1dbc2102017-03-31 13:30:24 +0000665 !Style.isCpp() && Current.is(tok::r_brace) && State.Stack.size() > 1 &&
Daniel Jasper4b444492014-11-21 13:38:53 +0000666 State.Stack[State.Stack.size() - 2].NestedBlockInlined;
Daniel Jasper1699eca2015-06-01 09:56:32 +0000667 if (!NestedBlockSpecialCase)
668 for (unsigned i = 0, e = State.Stack.size() - 1; i != e; ++i)
Daniel Jasperb16b9692014-05-21 12:51:23 +0000669 State.Stack[i].BreakBeforeParameter = true;
Daniel Jasperb16b9692014-05-21 12:51:23 +0000670
Daniel Jasper9e5ede02013-11-08 19:56:28 +0000671 if (PreviousNonComment &&
Francois Ferranda6b6d512017-05-24 11:36:58 +0000672 !PreviousNonComment->isOneOf(tok::comma, tok::colon, tok::semi) &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000673 (PreviousNonComment->isNot(TT_TemplateCloser) ||
Daniel Jasper6c0ee172014-11-14 13:14:45 +0000674 Current.NestingLevel != 0) &&
Daniel Jasper47bbda02015-05-18 13:47:23 +0000675 !PreviousNonComment->isOneOf(
676 TT_BinaryOperator, TT_FunctionAnnotationRParen, TT_JavaAnnotation,
677 TT_LeadingJavaAnnotation) &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000678 Current.isNot(TT_BinaryOperator) && !PreviousNonComment->opensScope())
Alexander Kornienko1f803962013-10-01 14:41:18 +0000679 State.Stack.back().BreakBeforeParameter = true;
680
Daniel Jasper1db6c382013-10-22 15:30:28 +0000681 // If we break after { or the [ of an array initializer, we should also break
682 // before the corresponding } or ].
Daniel Jasper90818052014-06-10 10:42:26 +0000683 if (PreviousNonComment &&
Daniel Jasper98e0b122017-02-20 14:51:16 +0000684 (PreviousNonComment->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) ||
Martin Probstc10d97f2017-08-29 08:30:07 +0000685 opensProtoMessageField(*PreviousNonComment, Style)))
Alexander Kornienko1f803962013-10-01 14:41:18 +0000686 State.Stack.back().BreakBeforeClosingBrace = true;
687
688 if (State.Stack.back().AvoidBinPacking) {
689 // If we are breaking after '(', '{', '<', this is not bin packing
Daniel Jasper2a958322014-05-21 13:26:58 +0000690 // unless AllowAllParametersOfDeclarationOnNextLine is false or this is a
691 // dict/object literal.
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000692 if (!Previous.isOneOf(tok::l_paren, tok::l_brace, TT_BinaryOperator) ||
Alexander Kornienko1f803962013-10-01 14:41:18 +0000693 (!Style.AllowAllParametersOfDeclarationOnNextLine &&
Daniel Jasper2a958322014-05-21 13:26:58 +0000694 State.Line->MustBeDeclaration) ||
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000695 Previous.is(TT_DictLiteral))
Alexander Kornienko1f803962013-10-01 14:41:18 +0000696 State.Stack.back().BreakBeforeParameter = true;
697 }
698
699 return Penalty;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000700}
701
Daniel Jasper9f388d02014-03-27 14:33:30 +0000702unsigned ContinuationIndenter::getNewLineColumn(const LineState &State) {
Daniel Jasper5d2587d2014-03-27 16:14:13 +0000703 if (!State.NextToken || !State.NextToken->Previous)
704 return 0;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000705 FormatToken &Current = *State.NextToken;
Daniel Jasper4281c5a2014-10-07 14:45:34 +0000706 const FormatToken &Previous = *Current.Previous;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000707 // If we are continuing an expression, we want to use the continuation indent.
708 unsigned ContinuationIndent =
709 std::max(State.Stack.back().LastSpace, State.Stack.back().Indent) +
710 Style.ContinuationIndentWidth;
711 const FormatToken *PreviousNonComment = Current.getPreviousNonComment();
712 const FormatToken *NextNonComment = Previous.getNextNonComment();
713 if (!NextNonComment)
714 NextNonComment = &Current;
Daniel Jasper50b4bd72014-11-02 19:16:41 +0000715
716 // Java specific bits.
Daniel Jasperd0ec0d62014-11-04 12:41:02 +0000717 if (Style.Language == FormatStyle::LK_Java &&
718 Current.isOneOf(Keywords.kw_implements, Keywords.kw_extends))
Daniel Jasper50b4bd72014-11-02 19:16:41 +0000719 return std::max(State.Stack.back().LastSpace,
720 State.Stack.back().Indent + Style.ContinuationIndentWidth);
721
Daniel Jasperb05a81d2014-05-09 13:11:16 +0000722 if (NextNonComment->is(tok::l_brace) && NextNonComment->BlockKind == BK_Block)
Daniel Jasper05cd5862014-05-08 12:21:30 +0000723 return Current.NestingLevel == 0 ? State.FirstIndent
724 : State.Stack.back().Indent;
Krasimir Georgievff747be2017-06-27 13:43:07 +0000725 if ((Current.isOneOf(tok::r_brace, tok::r_square) ||
Krasimir Georgiev26b144c2017-07-03 15:05:14 +0000726 (Current.is(tok::greater) &&
727 (Style.Language == FormatStyle::LK_Proto ||
728 Style.Language == FormatStyle::LK_TextProto))) &&
Krasimir Georgievff747be2017-06-27 13:43:07 +0000729 State.Stack.size() > 1) {
Daniel Jasperbd73bcf2015-10-27 13:42:08 +0000730 if (Current.closesBlockOrBlockTypeList(Style))
Daniel Jasper11a0ac62014-12-12 09:40:58 +0000731 return State.Stack[State.Stack.size() - 2].NestedBlockIndent;
732 if (Current.MatchingParen &&
733 Current.MatchingParen->BlockKind == BK_BracedInit)
Daniel Jasper9f388d02014-03-27 14:33:30 +0000734 return State.Stack[State.Stack.size() - 2].LastSpace;
Daniel Jasper24a14772014-12-10 17:24:34 +0000735 return State.FirstIndent;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000736 }
Martin Probstb2f06ea2017-05-29 07:50:52 +0000737 // Indent a closing parenthesis at the previous level if followed by a semi or
738 // opening brace. This allows indentations such as:
739 // foo(
740 // a,
741 // );
742 // function foo(
743 // a,
744 // ) {
745 // code(); //
746 // }
747 if (Current.is(tok::r_paren) && State.Stack.size() > 1 &&
748 (!Current.Next || Current.Next->isOneOf(tok::semi, tok::l_brace)))
Martin Probst2c1cdae2017-05-15 11:15:29 +0000749 return State.Stack[State.Stack.size() - 2].LastSpace;
Daniel Jasper98e0b122017-02-20 14:51:16 +0000750 if (NextNonComment->is(TT_TemplateString) && NextNonComment->closesScope())
751 return State.Stack[State.Stack.size() - 2].LastSpace;
Daniel Jasper783bac62014-04-15 09:54:30 +0000752 if (Current.is(tok::identifier) && Current.Next &&
Krasimir Georgievddb19242017-08-03 14:17:29 +0000753 (Current.Next->is(TT_DictLiteral) ||
754 ((Style.Language == FormatStyle::LK_Proto ||
755 Style.Language == FormatStyle::LK_TextProto) &&
756 Current.Next->isOneOf(TT_TemplateOpener, tok::l_brace))))
Daniel Jasper783bac62014-04-15 09:54:30 +0000757 return State.Stack.back().Indent;
Daniel Jasper09285532015-05-17 08:13:23 +0000758 if (NextNonComment->is(TT_ObjCStringLiteral) &&
759 State.StartOfStringLiteral != 0)
760 return State.StartOfStringLiteral - 1;
Alexander Kornienkod4fa2e62017-04-11 09:55:00 +0000761 if (NextNonComment->isStringLiteral() && State.StartOfStringLiteral != 0)
762 return State.StartOfStringLiteral;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000763 if (NextNonComment->is(tok::lessless) &&
764 State.Stack.back().FirstLessLess != 0)
765 return State.Stack.back().FirstLessLess;
766 if (NextNonComment->isMemberAccess()) {
Daniel Jasper24a14772014-12-10 17:24:34 +0000767 if (State.Stack.back().CallContinuation == 0)
Daniel Jasper9f388d02014-03-27 14:33:30 +0000768 return ContinuationIndent;
Daniel Jasper24a14772014-12-10 17:24:34 +0000769 return State.Stack.back().CallContinuation;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000770 }
771 if (State.Stack.back().QuestionColumn != 0 &&
Daniel Jasperc0d606a2014-04-14 11:08:45 +0000772 ((NextNonComment->is(tok::colon) &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000773 NextNonComment->is(TT_ConditionalExpr)) ||
774 Previous.is(TT_ConditionalExpr)))
Daniel Jasper9f388d02014-03-27 14:33:30 +0000775 return State.Stack.back().QuestionColumn;
776 if (Previous.is(tok::comma) && State.Stack.back().VariablePos != 0)
777 return State.Stack.back().VariablePos;
Daniel Jaspere9ab42d2014-10-31 18:23:49 +0000778 if ((PreviousNonComment &&
779 (PreviousNonComment->ClosesTemplateDeclaration ||
Daniel Jasper47bbda02015-05-18 13:47:23 +0000780 PreviousNonComment->isOneOf(
781 TT_AttributeParen, TT_FunctionAnnotationRParen, TT_JavaAnnotation,
782 TT_LeadingJavaAnnotation))) ||
Daniel Jasperc75e1ef2014-07-09 08:42:42 +0000783 (!Style.IndentWrappedFunctionNames &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000784 NextNonComment->isOneOf(tok::kw_operator, TT_FunctionDeclarationName)))
Daniel Jasper9f388d02014-03-27 14:33:30 +0000785 return std::max(State.Stack.back().LastSpace, State.Stack.back().Indent);
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000786 if (NextNonComment->is(TT_SelectorName)) {
Daniel Jasper9f388d02014-03-27 14:33:30 +0000787 if (!State.Stack.back().ObjCSelectorNameFound) {
Daniel Jasper24a14772014-12-10 17:24:34 +0000788 if (NextNonComment->LongestObjCSelectorName == 0)
Daniel Jasper9f388d02014-03-27 14:33:30 +0000789 return State.Stack.back().Indent;
Daniel Jaspera2a4d9c2015-05-13 09:38:25 +0000790 return (Style.IndentWrappedFunctionNames
791 ? std::max(State.Stack.back().Indent,
792 State.FirstIndent + Style.ContinuationIndentWidth)
793 : State.Stack.back().Indent) +
Daniel Jasper24a14772014-12-10 17:24:34 +0000794 NextNonComment->LongestObjCSelectorName -
795 NextNonComment->ColumnWidth;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000796 }
Daniel Jasper24a14772014-12-10 17:24:34 +0000797 if (!State.Stack.back().AlignColons)
798 return State.Stack.back().Indent;
799 if (State.Stack.back().ColonPos > NextNonComment->ColumnWidth)
800 return State.Stack.back().ColonPos - NextNonComment->ColumnWidth;
801 return State.Stack.back().Indent;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000802 }
Daniel Jasperfd36f0b2016-11-12 07:38:22 +0000803 if (NextNonComment->is(tok::colon) && NextNonComment->is(TT_ObjCMethodExpr))
804 return State.Stack.back().ColonPos;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000805 if (NextNonComment->is(TT_ArraySubscriptLSquare)) {
Daniel Jasper9f388d02014-03-27 14:33:30 +0000806 if (State.Stack.back().StartOfArraySubscripts != 0)
807 return State.Stack.back().StartOfArraySubscripts;
Daniel Jasper24a14772014-12-10 17:24:34 +0000808 return ContinuationIndent;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000809 }
Daniel Jasper9c950132015-05-07 14:19:59 +0000810
811 // This ensure that we correctly format ObjC methods calls without inputs,
812 // i.e. where the last element isn't selector like: [callee method];
813 if (NextNonComment->is(tok::identifier) && NextNonComment->FakeRParens == 0 &&
814 NextNonComment->Next && NextNonComment->Next->is(TT_ObjCMethodExpr))
Daniel Jaspereb536682015-05-06 12:48:06 +0000815 return State.Stack.back().Indent;
Daniel Jasper9c950132015-05-07 14:19:59 +0000816
Daniel Jasperb754a742015-03-12 15:04:53 +0000817 if (NextNonComment->isOneOf(TT_StartOfName, TT_PointerOrReference) ||
Daniel Jasperb2328b12015-07-06 14:07:51 +0000818 Previous.isOneOf(tok::coloncolon, tok::equal, TT_JsTypeColon))
Daniel Jasper9f388d02014-03-27 14:33:30 +0000819 return ContinuationIndent;
Daniel Jasper9f388d02014-03-27 14:33:30 +0000820 if (PreviousNonComment && PreviousNonComment->is(tok::colon) &&
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000821 PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral))
Daniel Jasper9f388d02014-03-27 14:33:30 +0000822 return ContinuationIndent;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000823 if (NextNonComment->is(TT_CtorInitializerComma))
Daniel Jasper9f388d02014-03-27 14:33:30 +0000824 return State.Stack.back().Indent;
Francois Ferranda6b6d512017-05-24 11:36:58 +0000825 if (PreviousNonComment && PreviousNonComment->is(TT_CtorInitializerColon) &&
826 Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon)
827 return State.Stack.back().Indent;
Andi-Bogdan Postelnicu0ef8ee12017-03-10 15:10:37 +0000828 if (NextNonComment->isOneOf(TT_CtorInitializerColon, TT_InheritanceColon,
829 TT_InheritanceComma))
830 return State.FirstIndent + Style.ConstructorInitializerIndentWidth;
Daniel Jasper316ab382014-08-06 13:14:58 +0000831 if (Previous.is(tok::r_paren) && !Current.isBinaryOperator() &&
Daniel Jasper119ff532014-11-14 12:31:14 +0000832 !Current.isOneOf(tok::colon, tok::comment))
Daniel Jasper316ab382014-08-06 13:14:58 +0000833 return ContinuationIndent;
Daniel Jasper5d2587d2014-03-27 16:14:13 +0000834 if (State.Stack.back().Indent == State.FirstIndent && PreviousNonComment &&
Daniel Jasper9f388d02014-03-27 14:33:30 +0000835 PreviousNonComment->isNot(tok::r_brace))
836 // Ensure that we fall back to the continuation indent width instead of
837 // just flushing continuations left.
838 return State.Stack.back().Indent + Style.ContinuationIndentWidth;
839 return State.Stack.back().Indent;
840}
841
Daniel Jasperde0328a2013-08-16 11:20:30 +0000842unsigned ContinuationIndenter::moveStateToNextToken(LineState &State,
843 bool DryRun, bool Newline) {
Daniel Jasperde0328a2013-08-16 11:20:30 +0000844 assert(State.Stack.size());
Daniel Jasper60553be2014-05-26 13:10:39 +0000845 const FormatToken &Current = *State.NextToken;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000846
Daniel Jasper240527c2017-01-16 13:13:15 +0000847 if (Current.isOneOf(tok::comma, TT_BinaryOperator))
848 State.Stack.back().NoLineBreakInOperand = false;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000849 if (Current.is(TT_InheritanceColon))
Daniel Jasperde0328a2013-08-16 11:20:30 +0000850 State.Stack.back().AvoidBinPacking = true;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000851 if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator)) {
Daniel Jasperc0d606a2014-04-14 11:08:45 +0000852 if (State.Stack.back().FirstLessLess == 0)
853 State.Stack.back().FirstLessLess = State.Column;
854 else
855 State.Stack.back().LastOperatorWrapped = Newline;
856 }
Daniel Jasper240527c2017-01-16 13:13:15 +0000857 if (Current.is(TT_BinaryOperator) && Current.isNot(tok::lessless))
858 State.Stack.back().LastOperatorWrapped = Newline;
859 if (Current.is(TT_ConditionalExpr) && Current.Previous &&
860 !Current.Previous->is(TT_ConditionalExpr))
Daniel Jasperc0d606a2014-04-14 11:08:45 +0000861 State.Stack.back().LastOperatorWrapped = Newline;
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000862 if (Current.is(TT_ArraySubscriptLSquare) &&
Daniel Jasperde0328a2013-08-16 11:20:30 +0000863 State.Stack.back().StartOfArraySubscripts == 0)
864 State.Stack.back().StartOfArraySubscripts = State.Column;
Daniel Jasper45860fa2016-02-03 17:27:10 +0000865 if (Style.BreakBeforeTernaryOperators && Current.is(tok::question))
Daniel Jasperde0328a2013-08-16 11:20:30 +0000866 State.Stack.back().QuestionColumn = State.Column;
Daniel Jasper45860fa2016-02-03 17:27:10 +0000867 if (!Style.BreakBeforeTernaryOperators && Current.isNot(tok::colon)) {
868 const FormatToken *Previous = Current.Previous;
869 while (Previous && Previous->isTrailingComment())
870 Previous = Previous->Previous;
871 if (Previous && Previous->is(tok::question))
872 State.Stack.back().QuestionColumn = State.Column;
873 }
Daniel Jaspercab46172017-04-24 14:28:49 +0000874 if (!Current.opensScope() && !Current.closesScope() &&
875 !Current.is(TT_PointerOrReference))
Daniel Jasperde0328a2013-08-16 11:20:30 +0000876 State.LowestLevelOnLine =
Daniel Jasper05cd5862014-05-08 12:21:30 +0000877 std::min(State.LowestLevelOnLine, Current.NestingLevel);
Daniel Jasper4c6e0052013-08-27 14:24:43 +0000878 if (Current.isMemberAccess())
Daniel Jasperde0328a2013-08-16 11:20:30 +0000879 State.Stack.back().StartOfFunctionCall =
Daniel Jasper00492f92016-01-05 13:03:50 +0000880 !Current.NextOperator ? 0 : State.Column;
Daniel Jasper3c44c222015-07-16 22:58:24 +0000881 if (Current.is(TT_SelectorName)) {
Daniel Jasperb88b25f2013-12-23 07:29:06 +0000882 State.Stack.back().ObjCSelectorNameFound = true;
Daniel Jasper3c44c222015-07-16 22:58:24 +0000883 if (Style.IndentWrappedFunctionNames) {
884 State.Stack.back().Indent =
885 State.FirstIndent + Style.ContinuationIndentWidth;
886 }
887 }
Francois Ferranda6b6d512017-05-24 11:36:58 +0000888 if (Current.is(TT_CtorInitializerColon) &&
889 Style.BreakConstructorInitializers != FormatStyle::BCIS_AfterColon) {
Daniel Jasperde0328a2013-08-16 11:20:30 +0000890 // Indent 2 from the column, so:
891 // SomeClass::SomeClass()
892 // : First(...), ...
893 // Next(...)
894 // ^ line up here.
895 State.Stack.back().Indent =
Manuel Klimek89628f62017-09-20 09:51:03 +0000896 State.Column +
897 (Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma
898 ? 0
899 : 2);
Daniel Jasperd6a1cab2015-01-12 10:23:24 +0000900 State.Stack.back().NestedBlockIndent = State.Stack.back().Indent;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000901 if (Style.ConstructorInitializerAllOnOneLineOrOnePerLine)
902 State.Stack.back().AvoidBinPacking = true;
903 State.Stack.back().BreakBeforeParameter = false;
904 }
Francois Ferranda6b6d512017-05-24 11:36:58 +0000905 if (Current.is(TT_CtorInitializerColon) &&
906 Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon) {
907 State.Stack.back().Indent =
908 State.FirstIndent + Style.ConstructorInitializerIndentWidth;
909 State.Stack.back().NestedBlockIndent = State.Stack.back().Indent;
910 if (Style.ConstructorInitializerAllOnOneLineOrOnePerLine)
Manuel Klimek89628f62017-09-20 09:51:03 +0000911 State.Stack.back().AvoidBinPacking = true;
Francois Ferranda6b6d512017-05-24 11:36:58 +0000912 }
Andi-Bogdan Postelnicu0ef8ee12017-03-10 15:10:37 +0000913 if (Current.is(TT_InheritanceColon))
914 State.Stack.back().Indent =
915 State.FirstIndent + Style.ContinuationIndentWidth;
Daniel Jasperc4144ea2015-05-13 16:09:21 +0000916 if (Current.isOneOf(TT_BinaryOperator, TT_ConditionalExpr) && Newline)
917 State.Stack.back().NestedBlockIndent =
918 State.Column + Current.ColumnWidth + 1;
Daniel Jasperd9b319e2017-02-20 12:43:48 +0000919 if (Current.isOneOf(TT_LambdaLSquare, TT_LambdaArrow))
920 State.Stack.back().LastSpace = State.Column;
Daniel Jasperde0328a2013-08-16 11:20:30 +0000921
Daniel Jasperde0328a2013-08-16 11:20:30 +0000922 // Insert scopes created by fake parenthesis.
923 const FormatToken *Previous = Current.getPreviousNonComment();
Daniel Jasperb16b9692014-05-21 12:51:23 +0000924
925 // Add special behavior to support a format commonly used for JavaScript
926 // closures:
927 // SomeFunction(function() {
928 // foo();
929 // bar();
930 // }, a, b, c);
Daniel Jasperb2ad4d42015-06-15 09:23:17 +0000931 if (Current.isNot(tok::comment) && Previous &&
932 Previous->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) &&
Daniel Jasperccff4d12016-01-04 07:27:33 +0000933 !Previous->is(TT_DictLiteral) && State.Stack.size() > 1) {
Daniel Jasper1699eca2015-06-01 09:56:32 +0000934 if (State.Stack[State.Stack.size() - 2].NestedBlockInlined && Newline)
935 for (unsigned i = 0, e = State.Stack.size() - 1; i != e; ++i)
Daniel Jasper4b444492014-11-21 13:38:53 +0000936 State.Stack[i].NoLineBreak = true;
Daniel Jasper4b444492014-11-21 13:38:53 +0000937 State.Stack[State.Stack.size() - 2].NestedBlockInlined = false;
938 }
Manuel Klimek89628f62017-09-20 09:51:03 +0000939 if (Previous &&
940 (Previous->isOneOf(tok::l_paren, tok::comma, tok::colon) ||
941 Previous->isOneOf(TT_BinaryOperator, TT_ConditionalExpr)) &&
Daniel Jasper4b444492014-11-21 13:38:53 +0000942 !Previous->isOneOf(TT_DictLiteral, TT_ObjCMethodExpr)) {
943 State.Stack.back().NestedBlockInlined =
944 !Newline &&
Daniel Jasper11a0ac62014-12-12 09:40:58 +0000945 (Previous->isNot(tok::l_paren) || Previous->ParameterCount > 1);
Daniel Jasperb16b9692014-05-21 12:51:23 +0000946 }
947
Daniel Jasper60553be2014-05-26 13:10:39 +0000948 moveStatePastFakeLParens(State, Newline);
Daniel Jasper60553be2014-05-26 13:10:39 +0000949 moveStatePastScopeCloser(State);
Krasimir Georgievbc05eba2017-03-08 12:54:50 +0000950 bool CanBreakProtrudingToken = !State.Stack.back().NoLineBreak &&
951 !State.Stack.back().NoLineBreakInOperand;
Daniel Jasperc06f6da2017-02-03 14:32:38 +0000952 moveStatePastScopeOpener(State, Newline);
Daniel Jasper60553be2014-05-26 13:10:39 +0000953 moveStatePastFakeRParens(State);
954
Daniel Jasper09285532015-05-17 08:13:23 +0000955 if (Current.is(TT_ObjCStringLiteral) && State.StartOfStringLiteral == 0)
956 State.StartOfStringLiteral = State.Column + 1;
Alexander Kornienkod4fa2e62017-04-11 09:55:00 +0000957 else if (Current.isStringLiteral() && State.StartOfStringLiteral == 0)
958 State.StartOfStringLiteral = State.Column;
Daniel Jasper09285532015-05-17 08:13:23 +0000959 else if (!Current.isOneOf(tok::comment, tok::identifier, tok::hash) &&
Daniel Jaspere6fcf7d2015-06-17 13:08:06 +0000960 !Current.isStringLiteral())
Daniel Jasper60553be2014-05-26 13:10:39 +0000961 State.StartOfStringLiteral = 0;
Daniel Jasper60553be2014-05-26 13:10:39 +0000962
963 State.Column += Current.ColumnWidth;
964 State.NextToken = State.NextToken->Next;
Krasimir Georgievbc05eba2017-03-08 12:54:50 +0000965 unsigned Penalty = 0;
966 if (CanBreakProtrudingToken)
967 Penalty = breakProtrudingToken(Current, State, DryRun);
Daniel Jasper60553be2014-05-26 13:10:39 +0000968 if (State.Column > getColumnLimit(State)) {
969 unsigned ExcessCharacters = State.Column - getColumnLimit(State);
970 Penalty += Style.PenaltyExcessCharacter * ExcessCharacters;
971 }
972
973 if (Current.Role)
974 Current.Role->formatFromToken(State, this, DryRun);
975 // If the previous has a special role, let it consume tokens as appropriate.
976 // It is necessary to start at the previous token for the only implemented
977 // role (comma separated list). That way, the decision whether or not to break
978 // after the "{" is already done and both options are tried and evaluated.
979 // FIXME: This is ugly, find a better way.
980 if (Previous && Previous->Role)
981 Penalty += Previous->Role->formatAfterToken(State, this, DryRun);
982
983 return Penalty;
984}
985
986void ContinuationIndenter::moveStatePastFakeLParens(LineState &State,
987 bool Newline) {
988 const FormatToken &Current = *State.NextToken;
989 const FormatToken *Previous = Current.getPreviousNonComment();
990
Daniel Jasperde0328a2013-08-16 11:20:30 +0000991 // Don't add extra indentation for the first fake parenthesis after
Dinesh Dwivedi0db806b2014-05-01 17:19:34 +0000992 // 'return', assignments or opening <({[. The indentation for these cases
Daniel Jasperde0328a2013-08-16 11:20:30 +0000993 // is special cased.
994 bool SkipFirstExtraIndent =
Daniel Jasper98f8ae32015-03-06 10:57:12 +0000995 (Previous && (Previous->opensScope() ||
996 Previous->isOneOf(tok::semi, tok::kw_return) ||
Daniel Jasper3219e432014-12-02 13:24:51 +0000997 (Previous->getPrecedence() == prec::Assignment &&
998 Style.AlignOperands) ||
Daniel Jaspera98b7b02014-11-25 10:05:17 +0000999 Previous->is(TT_ObjCMethodExpr)));
Daniel Jasperde0328a2013-08-16 11:20:30 +00001000 for (SmallVectorImpl<prec::Level>::const_reverse_iterator
1001 I = Current.FakeLParens.rbegin(),
1002 E = Current.FakeLParens.rend();
1003 I != E; ++I) {
1004 ParenState NewParenState = State.Stack.back();
1005 NewParenState.ContainsLineBreak = false;
Daniel Jasper04bbda92017-03-16 07:54:11 +00001006 NewParenState.LastOperatorWrapped = true;
Daniel Jasper240527c2017-01-16 13:13:15 +00001007 NewParenState.NoLineBreak =
1008 NewParenState.NoLineBreak || State.Stack.back().NoLineBreakInOperand;
Daniel Jaspereabede62013-09-30 08:29:03 +00001009
Daniel Jasper988e7e42017-05-08 15:07:52 +00001010 // Don't propagate AvoidBinPacking into subexpressions of arg/param lists.
1011 if (*I > prec::Comma)
1012 NewParenState.AvoidBinPacking = false;
1013
Daniel Jasper3aa9a6a2014-11-18 23:55:27 +00001014 // Indent from 'LastSpace' unless these are fake parentheses encapsulating
1015 // a builder type call after 'return' or, if the alignment after opening
1016 // brackets is disabled.
Daniel Jasper4281c5a2014-10-07 14:45:34 +00001017 if (!Current.isTrailingComment() &&
Daniel Jasper3219e432014-12-02 13:24:51 +00001018 (Style.AlignOperands || *I < prec::Assignment) &&
Daniel Jasper6cab6782014-11-20 09:54:49 +00001019 (!Previous || Previous->isNot(tok::kw_return) ||
1020 (Style.Language != FormatStyle::LK_Java && *I > 0)) &&
Daniel Jasper6501f7e2015-10-27 12:38:37 +00001021 (Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign ||
1022 *I != prec::Comma || Current.NestingLevel == 0))
Daniel Jaspereabede62013-09-30 08:29:03 +00001023 NewParenState.Indent =
1024 std::max(std::max(State.Column, NewParenState.Indent),
1025 State.Stack.back().LastSpace);
1026
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00001027 // Do not indent relative to the fake parentheses inserted for "." or "->".
1028 // This is a special case to make the following to statements consistent:
1029 // OuterFunction(InnerFunctionCall( // break
1030 // ParameterToInnerFunction));
1031 // OuterFunction(SomeObject.InnerFunctionCall( // break
1032 // ParameterToInnerFunction));
1033 if (*I > prec::Unknown)
1034 NewParenState.LastSpace = std::max(NewParenState.LastSpace, State.Column);
Daniel Jasper2a9f7202016-02-08 09:52:54 +00001035 if (*I != prec::Conditional && !Current.is(TT_UnaryOperator) &&
1036 Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign)
Daniel Jaspera536df42014-12-08 21:28:31 +00001037 NewParenState.StartOfFunctionCall = State.Column;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001038
1039 // Always indent conditional expressions. Never indent expression where
1040 // the 'operator' is ',', ';' or an assignment (i.e. *I <=
1041 // prec::Assignment) as those have different indentation rules. Indent
1042 // other expression, unless the indentation needs to be skipped.
1043 if (*I == prec::Conditional ||
1044 (!SkipFirstExtraIndent && *I > prec::Assignment &&
Daniel Jasper8c6e9ef2014-12-02 09:46:56 +00001045 !Current.isTrailingComment()))
Daniel Jasper6633ab82013-10-18 10:38:14 +00001046 NewParenState.Indent += Style.ContinuationIndentWidth;
Daniel Jasper7bec87c2016-01-07 18:11:54 +00001047 if ((Previous && !Previous->opensScope()) || *I != prec::Comma)
Daniel Jasperde0328a2013-08-16 11:20:30 +00001048 NewParenState.BreakBeforeParameter = false;
1049 State.Stack.push_back(NewParenState);
1050 SkipFirstExtraIndent = false;
1051 }
Daniel Jasper60553be2014-05-26 13:10:39 +00001052}
Daniel Jasperde0328a2013-08-16 11:20:30 +00001053
Daniel Jasper11a0ac62014-12-12 09:40:58 +00001054void ContinuationIndenter::moveStatePastFakeRParens(LineState &State) {
1055 for (unsigned i = 0, e = State.NextToken->FakeRParens; i != e; ++i) {
Daniel Jasper335ff262014-05-28 09:11:53 +00001056 unsigned VariablePos = State.Stack.back().VariablePos;
Daniel Jasper335ff262014-05-28 09:11:53 +00001057 if (State.Stack.size() == 1) {
1058 // Do not pop the last element.
1059 break;
1060 }
1061 State.Stack.pop_back();
1062 State.Stack.back().VariablePos = VariablePos;
1063 }
1064}
1065
Daniel Jasper60553be2014-05-26 13:10:39 +00001066void ContinuationIndenter::moveStatePastScopeOpener(LineState &State,
1067 bool Newline) {
1068 const FormatToken &Current = *State.NextToken;
1069 if (!Current.opensScope())
1070 return;
1071
1072 if (Current.MatchingParen && Current.BlockKind == BK_Block) {
1073 moveStateToNewBlock(State);
1074 return;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001075 }
1076
Daniel Jasper60553be2014-05-26 13:10:39 +00001077 unsigned NewIndent;
Daniel Jasperde7ca752015-05-04 07:39:00 +00001078 unsigned LastSpace = State.Stack.back().LastSpace;
Daniel Jasper60553be2014-05-26 13:10:39 +00001079 bool AvoidBinPacking;
1080 bool BreakBeforeParameter = false;
Daniel Jasperea40cee2015-07-14 11:26:14 +00001081 unsigned NestedBlockIndent = std::max(State.Stack.back().StartOfFunctionCall,
1082 State.Stack.back().NestedBlockIndent);
Krasimir Georgievff747be2017-06-27 13:43:07 +00001083 if (Current.isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) ||
Krasimir Georgiev26b144c2017-07-03 15:05:14 +00001084 opensProtoMessageField(Current, Style)) {
Daniel Jasperbd73bcf2015-10-27 13:42:08 +00001085 if (Current.opensBlockOrBlockTypeList(Style)) {
Martin Probst38423272017-06-06 12:38:29 +00001086 NewIndent = Style.IndentWidth +
1087 std::min(State.Column, State.Stack.back().NestedBlockIndent);
Daniel Jasper60553be2014-05-26 13:10:39 +00001088 } else {
Daniel Jasper11a0ac62014-12-12 09:40:58 +00001089 NewIndent = State.Stack.back().LastSpace + Style.ContinuationIndentWidth;
Daniel Jasper60553be2014-05-26 13:10:39 +00001090 }
1091 const FormatToken *NextNoComment = Current.getNextNonComment();
Daniel Jasper50780ce2016-01-13 16:41:34 +00001092 bool EndsInComma = Current.MatchingParen &&
1093 Current.MatchingParen->Previous &&
1094 Current.MatchingParen->Previous->is(tok::comma);
Manuel Klimek89628f62017-09-20 09:51:03 +00001095 AvoidBinPacking = EndsInComma || Current.is(TT_DictLiteral) ||
1096 Style.Language == FormatStyle::LK_Proto ||
1097 Style.Language == FormatStyle::LK_TextProto ||
1098 !Style.BinPackArguments ||
1099 (NextNoComment &&
1100 NextNoComment->isOneOf(TT_DesignatedInitializerPeriod,
1101 TT_DesignatedInitializerLSquare));
Francois Ferrandd2130f52017-06-30 20:00:02 +00001102 BreakBeforeParameter = EndsInComma;
Daniel Jasperea40cee2015-07-14 11:26:14 +00001103 if (Current.ParameterCount > 1)
1104 NestedBlockIndent = std::max(NestedBlockIndent, State.Column + 1);
Daniel Jasper60553be2014-05-26 13:10:39 +00001105 } else {
1106 NewIndent = Style.ContinuationIndentWidth +
1107 std::max(State.Stack.back().LastSpace,
1108 State.Stack.back().StartOfFunctionCall);
Daniel Jasperde7ca752015-05-04 07:39:00 +00001109
1110 // Ensure that different different brackets force relative alignment, e.g.:
1111 // void SomeFunction(vector< // break
1112 // int> v);
1113 // FIXME: We likely want to do this for more combinations of brackets.
1114 // Verify that it is wanted for ObjC, too.
Daniel Jasper3f119412017-01-31 14:39:33 +00001115 if (Current.is(tok::less) && Current.ParentBracket == tok::l_paren) {
Daniel Jasperde7ca752015-05-04 07:39:00 +00001116 NewIndent = std::max(NewIndent, State.Stack.back().Indent);
1117 LastSpace = std::max(LastSpace, State.Stack.back().Indent);
1118 }
1119
Martin Probst2c1cdae2017-05-15 11:15:29 +00001120 bool EndsInComma =
1121 Current.MatchingParen &&
1122 Current.MatchingParen->getPreviousNonComment() &&
1123 Current.MatchingParen->getPreviousNonComment()->is(tok::comma);
1124
Daniel Jasper18210d72014-10-09 09:52:05 +00001125 AvoidBinPacking =
Martin Probst2c1cdae2017-05-15 11:15:29 +00001126 (Style.Language == FormatStyle::LK_JavaScript && EndsInComma) ||
Daniel Jasper18210d72014-10-09 09:52:05 +00001127 (State.Line->MustBeDeclaration && !Style.BinPackParameters) ||
1128 (!State.Line->MustBeDeclaration && !Style.BinPackArguments) ||
1129 (Style.ExperimentalAutoDetectBinPacking &&
1130 (Current.PackingKind == PPK_OnePerLine ||
1131 (!BinPackInconclusiveFunctions &&
1132 Current.PackingKind == PPK_Inconclusive)));
Martin Probst2c1cdae2017-05-15 11:15:29 +00001133
Daniel Jasper289afc02015-04-23 09:23:17 +00001134 if (Current.is(TT_ObjCMethodExpr) && Current.MatchingParen) {
1135 if (Style.ColumnLimit) {
1136 // If this '[' opens an ObjC call, determine whether all parameters fit
1137 // into one line and put one per line if they don't.
1138 if (getLengthToMatchingParen(Current) + State.Column >
Daniel Jasper60553be2014-05-26 13:10:39 +00001139 getColumnLimit(State))
Daniel Jasper289afc02015-04-23 09:23:17 +00001140 BreakBeforeParameter = true;
1141 } else {
1142 // For ColumnLimit = 0, we have to figure out whether there is or has to
1143 // be a line break within this call.
1144 for (const FormatToken *Tok = &Current;
1145 Tok && Tok != Current.MatchingParen; Tok = Tok->Next) {
Daniel Jaspere6fcf7d2015-06-17 13:08:06 +00001146 if (Tok->MustBreakBefore ||
Daniel Jasper289afc02015-04-23 09:23:17 +00001147 (Tok->CanBreakBefore && Tok->NewlinesBefore > 0)) {
1148 BreakBeforeParameter = true;
1149 break;
1150 }
1151 }
1152 }
1153 }
Martin Probst2c1cdae2017-05-15 11:15:29 +00001154
1155 if (Style.Language == FormatStyle::LK_JavaScript && EndsInComma)
1156 BreakBeforeParameter = true;
Daniel Jasper60553be2014-05-26 13:10:39 +00001157 }
Daniel Jasperbd73bcf2015-10-27 13:42:08 +00001158 // Generally inherit NoLineBreak from the current scope to nested scope.
1159 // However, don't do this for non-empty nested blocks, dict literals and
1160 // array literals as these follow different indentation rules.
1161 bool NoLineBreak =
1162 Current.Children.empty() &&
1163 !Current.isOneOf(TT_DictLiteral, TT_ArrayInitializerLSquare) &&
1164 (State.Stack.back().NoLineBreak ||
Daniel Jasper240527c2017-01-16 13:13:15 +00001165 State.Stack.back().NoLineBreakInOperand ||
Daniel Jasperbd73bcf2015-10-27 13:42:08 +00001166 (Current.is(TT_TemplateOpener) &&
Daniel Jasper240527c2017-01-16 13:13:15 +00001167 State.Stack.back().ContainsUnwrappedBuilder));
Daniel Jasper7d42f3f2017-01-31 11:25:01 +00001168 State.Stack.push_back(
1169 ParenState(NewIndent, LastSpace, AvoidBinPacking, NoLineBreak));
Daniel Jasper11a0ac62014-12-12 09:40:58 +00001170 State.Stack.back().NestedBlockIndent = NestedBlockIndent;
Daniel Jasper60553be2014-05-26 13:10:39 +00001171 State.Stack.back().BreakBeforeParameter = BreakBeforeParameter;
Daniel Jasper114a2bc2014-06-03 12:02:45 +00001172 State.Stack.back().HasMultipleNestedBlocks = Current.BlockParameterCount > 1;
Daniel Jasper60553be2014-05-26 13:10:39 +00001173}
1174
1175void ContinuationIndenter::moveStatePastScopeCloser(LineState &State) {
1176 const FormatToken &Current = *State.NextToken;
1177 if (!Current.closesScope())
1178 return;
1179
1180 // If we encounter a closing ), ], } or >, we can remove a level from our
1181 // stacks.
1182 if (State.Stack.size() > 1 &&
Daniel Jasperc06f6da2017-02-03 14:32:38 +00001183 (Current.isOneOf(tok::r_paren, tok::r_square, TT_TemplateString) ||
Daniel Jasper60553be2014-05-26 13:10:39 +00001184 (Current.is(tok::r_brace) && State.NextToken != State.Line->First) ||
Daniel Jaspera98b7b02014-11-25 10:05:17 +00001185 State.NextToken->is(TT_TemplateCloser)))
Daniel Jasper60553be2014-05-26 13:10:39 +00001186 State.Stack.pop_back();
Daniel Jasper335ff262014-05-28 09:11:53 +00001187
Daniel Jasper60553be2014-05-26 13:10:39 +00001188 if (Current.is(tok::r_square)) {
1189 // If this ends the array subscript expr, reset the corresponding value.
1190 const FormatToken *NextNonComment = Current.getNextNonComment();
1191 if (NextNonComment && NextNonComment->isNot(tok::l_square))
1192 State.Stack.back().StartOfArraySubscripts = 0;
1193 }
1194}
1195
1196void ContinuationIndenter::moveStateToNewBlock(LineState &State) {
Daniel Jasper11a0ac62014-12-12 09:40:58 +00001197 unsigned NestedBlockIndent = State.Stack.back().NestedBlockIndent;
Daniel Jasper50d634b2014-10-28 16:53:38 +00001198 // ObjC block sometimes follow special indentation rules.
Daniel Jaspera98b7b02014-11-25 10:05:17 +00001199 unsigned NewIndent =
Daniel Jasper11a0ac62014-12-12 09:40:58 +00001200 NestedBlockIndent + (State.NextToken->is(TT_ObjCBlockLBrace)
1201 ? Style.ObjCBlockIndentWidth
1202 : Style.IndentWidth);
Daniel Jasper7d42f3f2017-01-31 11:25:01 +00001203 State.Stack.push_back(ParenState(NewIndent, State.Stack.back().LastSpace,
1204 /*AvoidBinPacking=*/true,
1205 /*NoLineBreak=*/false));
Daniel Jasper11a0ac62014-12-12 09:40:58 +00001206 State.Stack.back().NestedBlockIndent = NestedBlockIndent;
Daniel Jasper60553be2014-05-26 13:10:39 +00001207 State.Stack.back().BreakBeforeParameter = true;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001208}
1209
Alexander Kornienko917f9e02013-09-10 12:29:48 +00001210unsigned ContinuationIndenter::addMultilineToken(const FormatToken &Current,
1211 LineState &State) {
Daniel Jasperacadc8e2016-06-08 09:45:08 +00001212 if (!Current.IsMultiline)
1213 return 0;
1214
Alexander Kornienkod7b837e2013-08-29 17:32:57 +00001215 // Break before further function parameters on all levels.
1216 for (unsigned i = 0, e = State.Stack.size(); i != e; ++i)
1217 State.Stack[i].BreakBeforeParameter = true;
1218
Alexander Kornienko39856b72013-09-10 09:38:25 +00001219 unsigned ColumnsUsed = State.Column;
Alexander Kornienko632abb92013-09-02 13:58:14 +00001220 // We can only affect layout of the first and the last line, so the penalty
1221 // for all other lines is constant, and we ignore it.
Alexander Kornienkoebb43ca2013-09-05 14:08:34 +00001222 State.Column = Current.LastLineColumnWidth;
Alexander Kornienko632abb92013-09-02 13:58:14 +00001223
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00001224 if (ColumnsUsed > getColumnLimit(State))
1225 return Style.PenaltyExcessCharacter * (ColumnsUsed - getColumnLimit(State));
Alexander Kornienkod7b837e2013-08-29 17:32:57 +00001226 return 0;
1227}
1228
Daniel Jasperde0328a2013-08-16 11:20:30 +00001229unsigned ContinuationIndenter::breakProtrudingToken(const FormatToken &Current,
1230 LineState &State,
1231 bool DryRun) {
Alexander Kornienko917f9e02013-09-10 12:29:48 +00001232 // Don't break multi-line tokens other than block comments. Instead, just
1233 // update the state.
Daniel Jaspera98b7b02014-11-25 10:05:17 +00001234 if (Current.isNot(TT_BlockComment) && Current.IsMultiline)
Alexander Kornienko917f9e02013-09-10 12:29:48 +00001235 return addMultilineToken(Current, State);
1236
Daniel Jasper616de8642014-11-23 16:46:28 +00001237 // Don't break implicit string literals or import statements.
Daniel Jaspera98b7b02014-11-25 10:05:17 +00001238 if (Current.is(TT_ImplicitStringLiteral) ||
Daniel Jasper616de8642014-11-23 16:46:28 +00001239 State.Line->Type == LT_ImportStatement)
Daniel Jasper98857842013-10-30 13:54:53 +00001240 return 0;
1241
Daniel Jasper04b6a082013-12-20 06:22:01 +00001242 if (!Current.isStringLiteral() && !Current.is(tok::comment))
Daniel Jasperf93551c2013-08-23 10:05:49 +00001243 return 0;
1244
Ahmed Charlesb8984322014-03-07 20:03:18 +00001245 std::unique_ptr<BreakableToken> Token;
Alexander Kornienko39856b72013-09-10 09:38:25 +00001246 unsigned StartColumn = State.Column - Current.ColumnWidth;
Alexander Kornienko3abbb8a2013-11-12 17:30:49 +00001247 unsigned ColumnLimit = getColumnLimit(State);
Daniel Jasperde0328a2013-08-16 11:20:30 +00001248
Daniel Jasper04b6a082013-12-20 06:22:01 +00001249 if (Current.isStringLiteral()) {
Daniel Jasper428f0b12015-01-04 09:11:17 +00001250 // FIXME: String literal breaking is currently disabled for Java and JS, as
1251 // it requires strings to be merged using "+" which we don't support.
1252 if (Style.Language == FormatStyle::LK_Java ||
Daniel Jaspere1a7b762016-02-01 11:21:02 +00001253 Style.Language == FormatStyle::LK_JavaScript ||
1254 !Style.BreakStringLiterals)
Daniel Jasper428f0b12015-01-04 09:11:17 +00001255 return 0;
1256
Alexander Kornienko384b40b2013-10-11 21:43:05 +00001257 // Don't break string literals inside preprocessor directives (except for
1258 // #define directives, as their contents are stored in separate lines and
1259 // are not affected by this check).
1260 // This way we avoid breaking code with line directives and unknown
1261 // preprocessor directives that contain long string literals.
1262 if (State.Line->Type == LT_PreprocessorDirective)
1263 return 0;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001264 // Exempts unterminated string literals from line breaking. The user will
1265 // likely want to terminate the string before any line breaking is done.
1266 if (Current.IsUnterminatedLiteral)
1267 return 0;
1268
Alexander Kornienko81e32942013-09-16 20:20:49 +00001269 StringRef Text = Current.TokenText;
1270 StringRef Prefix;
1271 StringRef Postfix;
1272 // FIXME: Handle whitespace between '_T', '(', '"..."', and ')'.
1273 // FIXME: Store Prefix and Suffix (or PrefixLength and SuffixLength to
1274 // reduce the overhead) for each FormatToken, which is a string, so that we
1275 // don't run multiple checks here on the hot path.
1276 if ((Text.endswith(Postfix = "\"") &&
Alexander Kornienkod4fa2e62017-04-11 09:55:00 +00001277 (Text.startswith(Prefix = "@\"") || Text.startswith(Prefix = "\"") ||
Daniel Jasper174b0122014-01-09 14:18:12 +00001278 Text.startswith(Prefix = "u\"") || Text.startswith(Prefix = "U\"") ||
1279 Text.startswith(Prefix = "u8\"") ||
Alexander Kornienko81e32942013-09-16 20:20:49 +00001280 Text.startswith(Prefix = "L\""))) ||
Alexander Kornienko732b6bd2014-12-14 20:47:11 +00001281 (Text.startswith(Prefix = "_T(\"") && Text.endswith(Postfix = "\")"))) {
Daniel Jasper7d42f3f2017-01-31 11:25:01 +00001282 Token.reset(new BreakableStringLiteral(Current, StartColumn, Prefix,
1283 Postfix, State.Line->InPPDirective,
1284 Encoding, Style));
Alexander Kornienko81e32942013-09-16 20:20:49 +00001285 } else {
1286 return 0;
1287 }
Daniel Jasperacadc8e2016-06-08 09:45:08 +00001288 } else if (Current.is(TT_BlockComment)) {
Krasimir Georgievc602af92017-08-24 08:55:07 +00001289 if (!Current.isTrailingComment() || !Style.ReflowComments ||
Krasimir Georgiev91834222017-01-25 13:58:58 +00001290 // If a comment token switches formatting, like
1291 // /* clang-format on */, we don't want to break it further,
1292 // but we may still want to adjust its indentation.
1293 switchesFormatting(Current))
Daniel Jasperacadc8e2016-06-08 09:45:08 +00001294 return addMultilineToken(Current, State);
Daniel Jasperde0328a2013-08-16 11:20:30 +00001295 Token.reset(new BreakableBlockComment(
Daniel Jasper7d42f3f2017-01-31 11:25:01 +00001296 Current, StartColumn, Current.OriginalColumn, !Current.Previous,
1297 State.Line->InPPDirective, Encoding, Style));
Daniel Jaspera98b7b02014-11-25 10:05:17 +00001298 } else if (Current.is(TT_LineComment) &&
Craig Topper2145bc02014-05-09 08:15:10 +00001299 (Current.Previous == nullptr ||
Daniel Jaspera98b7b02014-11-25 10:05:17 +00001300 Current.Previous->isNot(TT_ImplicitStringLiteral))) {
Daniel Jaspera0a50392015-12-01 13:28:53 +00001301 if (!Style.ReflowComments ||
Krasimir Georgiev91834222017-01-25 13:58:58 +00001302 CommentPragmasRegex.match(Current.TokenText.substr(2)) ||
1303 switchesFormatting(Current))
Alexander Kornienkoce9161a2014-01-02 15:13:14 +00001304 return 0;
Krasimir Georgiev91834222017-01-25 13:58:58 +00001305 Token.reset(new BreakableLineCommentSection(
Daniel Jasper7d42f3f2017-01-31 11:25:01 +00001306 Current, StartColumn, Current.OriginalColumn, !Current.Previous,
Krasimir Georgiev91834222017-01-25 13:58:58 +00001307 /*InPPDirective=*/false, Encoding, Style));
Alexander Kornienko3abbb8a2013-11-12 17:30:49 +00001308 // We don't insert backslashes when breaking line comments.
1309 ColumnLimit = Style.ColumnLimit;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001310 } else {
1311 return 0;
1312 }
Alexander Kornienko3abbb8a2013-11-12 17:30:49 +00001313 if (Current.UnbreakableTailLength >= ColumnLimit)
Daniel Jasperde0328a2013-08-16 11:20:30 +00001314 return 0;
1315
Alexander Kornienko3abbb8a2013-11-12 17:30:49 +00001316 unsigned RemainingSpace = ColumnLimit - Current.UnbreakableTailLength;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001317 bool BreakInserted = false;
Krasimir Georgiev91834222017-01-25 13:58:58 +00001318 // We use a conservative reflowing strategy. Reflow starts after a line is
1319 // broken or the corresponding whitespace compressed. Reflow ends as soon as a
1320 // line that doesn't get reflown with the previous line is reached.
1321 bool ReflowInProgress = false;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001322 unsigned Penalty = 0;
1323 unsigned RemainingTokenColumns = 0;
Krasimir Georgiev22d7e6b2017-07-20 22:29:39 +00001324 unsigned TailOffset = 0;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001325 for (unsigned LineIndex = 0, EndIndex = Token->getLineCount();
1326 LineIndex != EndIndex; ++LineIndex) {
Krasimir Georgiev91834222017-01-25 13:58:58 +00001327 BreakableToken::Split SplitBefore(StringRef::npos, 0);
1328 if (ReflowInProgress) {
1329 SplitBefore = Token->getSplitBefore(LineIndex, RemainingTokenColumns,
Krasimir Georgiev00c5c722017-02-02 15:32:19 +00001330 RemainingSpace, CommentPragmasRegex);
Krasimir Georgiev91834222017-01-25 13:58:58 +00001331 }
1332 ReflowInProgress = SplitBefore.first != StringRef::npos;
Krasimir Georgiev22d7e6b2017-07-20 22:29:39 +00001333 TailOffset =
Krasimir Georgiev91834222017-01-25 13:58:58 +00001334 ReflowInProgress ? (SplitBefore.first + SplitBefore.second) : 0;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001335 if (!DryRun)
Krasimir Georgiev91834222017-01-25 13:58:58 +00001336 Token->replaceWhitespaceBefore(LineIndex, RemainingTokenColumns,
1337 RemainingSpace, SplitBefore, Whitespaces);
1338 RemainingTokenColumns = Token->getLineLengthAfterSplitBefore(
1339 LineIndex, TailOffset, RemainingTokenColumns, ColumnLimit, SplitBefore);
Daniel Jasperde0328a2013-08-16 11:20:30 +00001340 while (RemainingTokenColumns > RemainingSpace) {
Krasimir Georgiev17725d82017-03-08 08:55:12 +00001341 BreakableToken::Split Split = Token->getSplit(
1342 LineIndex, TailOffset, ColumnLimit, CommentPragmasRegex);
Daniel Jasperde0328a2013-08-16 11:20:30 +00001343 if (Split.first == StringRef::npos) {
1344 // The last line's penalty is handled in addNextStateToQueue().
1345 if (LineIndex < EndIndex - 1)
1346 Penalty += Style.PenaltyExcessCharacter *
1347 (RemainingTokenColumns - RemainingSpace);
1348 break;
1349 }
1350 assert(Split.first != 0);
Alexander Kornienko875395f2013-11-12 17:50:13 +00001351
Krasimir Georgiev91834222017-01-25 13:58:58 +00001352 // Check if compressing the whitespace range will bring the line length
1353 // under the limit. If that is the case, we perform whitespace compression
1354 // instead of inserting a line break.
1355 unsigned RemainingTokenColumnsAfterCompression =
1356 Token->getLineLengthAfterCompression(RemainingTokenColumns, Split);
1357 if (RemainingTokenColumnsAfterCompression <= RemainingSpace) {
1358 RemainingTokenColumns = RemainingTokenColumnsAfterCompression;
1359 ReflowInProgress = true;
Alexander Kornienko875395f2013-11-12 17:50:13 +00001360 if (!DryRun)
Krasimir Georgiev91834222017-01-25 13:58:58 +00001361 Token->compressWhitespace(LineIndex, TailOffset, Split, Whitespaces);
Alexander Kornienko875395f2013-11-12 17:50:13 +00001362 break;
1363 }
1364
Krasimir Georgiev91834222017-01-25 13:58:58 +00001365 unsigned NewRemainingTokenColumns = Token->getLineLengthAfterSplit(
1366 LineIndex, TailOffset + Split.first + Split.second, StringRef::npos);
1367
Alexander Kornienko64a42b82014-04-15 14:52:43 +00001368 // When breaking before a tab character, it may be moved by a few columns,
1369 // but will still be expanded to the next tab stop, so we don't save any
1370 // columns.
1371 if (NewRemainingTokenColumns == RemainingTokenColumns)
1372 break;
1373
Daniel Jasperde0328a2013-08-16 11:20:30 +00001374 assert(NewRemainingTokenColumns < RemainingTokenColumns);
1375 if (!DryRun)
1376 Token->insertBreak(LineIndex, TailOffset, Split, Whitespaces);
Daniel Jasper2739af32013-08-28 10:03:58 +00001377 Penalty += Current.SplitPenalty;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001378 unsigned ColumnsUsed =
1379 Token->getLineLengthAfterSplit(LineIndex, TailOffset, Split.first);
Alexander Kornienko3abbb8a2013-11-12 17:30:49 +00001380 if (ColumnsUsed > ColumnLimit) {
1381 Penalty += Style.PenaltyExcessCharacter * (ColumnsUsed - ColumnLimit);
Daniel Jasperde0328a2013-08-16 11:20:30 +00001382 }
1383 TailOffset += Split.first + Split.second;
1384 RemainingTokenColumns = NewRemainingTokenColumns;
Krasimir Georgiev91834222017-01-25 13:58:58 +00001385 ReflowInProgress = true;
Daniel Jasperde0328a2013-08-16 11:20:30 +00001386 BreakInserted = true;
1387 }
1388 }
1389
Krasimir Georgiev3b865342017-08-09 09:42:32 +00001390 BreakableToken::Split SplitAfterLastLine =
1391 Token->getSplitAfterLastLine(TailOffset, ColumnLimit);
Krasimir Georgiev22d7e6b2017-07-20 22:29:39 +00001392 if (SplitAfterLastLine.first != StringRef::npos) {
1393 if (!DryRun)
1394 Token->replaceWhitespaceAfterLastLine(TailOffset, SplitAfterLastLine,
1395 Whitespaces);
1396 RemainingTokenColumns = Token->getLineLengthAfterSplitAfterLastLine(
1397 TailOffset, SplitAfterLastLine);
1398 }
1399
Daniel Jasperde0328a2013-08-16 11:20:30 +00001400 State.Column = RemainingTokenColumns;
1401
1402 if (BreakInserted) {
1403 // If we break the token inside a parameter list, we need to break before
1404 // the next parameter on all levels, so that the next parameter is clearly
1405 // visible. Line comments already introduce a break.
Daniel Jaspera98b7b02014-11-25 10:05:17 +00001406 if (Current.isNot(TT_LineComment)) {
Daniel Jasperde0328a2013-08-16 11:20:30 +00001407 for (unsigned i = 0, e = State.Stack.size(); i != e; ++i)
1408 State.Stack[i].BreakBeforeParameter = true;
1409 }
1410
Daniel Jasper04b6a082013-12-20 06:22:01 +00001411 Penalty += Current.isStringLiteral() ? Style.PenaltyBreakString
1412 : Style.PenaltyBreakComment;
Daniel Jasper2739af32013-08-28 10:03:58 +00001413
Daniel Jasperde0328a2013-08-16 11:20:30 +00001414 State.Stack.back().LastSpace = StartColumn;
1415 }
Krasimir Georgiev91834222017-01-25 13:58:58 +00001416
1417 Token->updateNextToken(State);
1418
Daniel Jasperde0328a2013-08-16 11:20:30 +00001419 return Penalty;
1420}
1421
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00001422unsigned ContinuationIndenter::getColumnLimit(const LineState &State) const {
Daniel Jasperde0328a2013-08-16 11:20:30 +00001423 // In preprocessor directives reserve two chars for trailing " \"
Daniel Jasper9fe0e8d2013-09-05 09:29:45 +00001424 return Style.ColumnLimit - (State.Line->InPPDirective ? 2 : 0);
Daniel Jasperde0328a2013-08-16 11:20:30 +00001425}
1426
Daniel Jasperc39b56f2013-12-16 07:23:08 +00001427bool ContinuationIndenter::nextIsMultilineString(const LineState &State) {
Daniel Jasperf438cb72013-08-23 11:57:34 +00001428 const FormatToken &Current = *State.NextToken;
Daniel Jaspera98b7b02014-11-25 10:05:17 +00001429 if (!Current.isStringLiteral() || Current.is(TT_ImplicitStringLiteral))
Daniel Jasperf438cb72013-08-23 11:57:34 +00001430 return false;
Alexander Kornienkod7b837e2013-08-29 17:32:57 +00001431 // We never consider raw string literals "multiline" for the purpose of
Daniel Jasperc39b56f2013-12-16 07:23:08 +00001432 // AlwaysBreakBeforeMultilineStrings implementation as they are special-cased
1433 // (see TokenAnnotator::mustBreakBefore().
Alexander Kornienkod7b837e2013-08-29 17:32:57 +00001434 if (Current.TokenText.startswith("R\""))
1435 return false;
Alexander Kornienko39856b72013-09-10 09:38:25 +00001436 if (Current.IsMultiline)
Alexander Kornienkod7b837e2013-08-29 17:32:57 +00001437 return true;
Daniel Jasperf438cb72013-08-23 11:57:34 +00001438 if (Current.getNextNonComment() &&
Daniel Jasper04b6a082013-12-20 06:22:01 +00001439 Current.getNextNonComment()->isStringLiteral())
Daniel Jasperf438cb72013-08-23 11:57:34 +00001440 return true; // Implicit concatenation.
Daniel Jasper6fd5d642015-01-20 12:59:20 +00001441 if (Style.ColumnLimit != 0 &&
1442 State.Column + Current.ColumnWidth + Current.UnbreakableTailLength >
1443 Style.ColumnLimit)
Daniel Jasperf438cb72013-08-23 11:57:34 +00001444 return true; // String will be split.
Alexander Kornienkod7b837e2013-08-29 17:32:57 +00001445 return false;
Daniel Jasperf438cb72013-08-23 11:57:34 +00001446}
1447
Daniel Jasperde0328a2013-08-16 11:20:30 +00001448} // namespace format
1449} // namespace clang