1 //===--- 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 15 #include "ContinuationIndenter.h" 16 #include "BreakableToken.h" 17 #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" 22 23 #define DEBUG_TYPE "format-indenter" 24 25 namespace clang { 26 namespace format { 27 28 // Returns the length of everything up to the first possible line break after 29 // the ), ], } or > matching \c Tok. 30 static unsigned getLengthToMatchingParen(const FormatToken &Tok) { 31 if (!Tok.MatchingParen) 32 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 40 static unsigned getLengthToNextOperator(const FormatToken &Tok) { 41 if (!Tok.NextOperator) 42 return 0; 43 return Tok.NextOperator->TotalLength - Tok.TotalLength; 44 } 45 46 // Returns \c true if \c Tok is the "." or "->" of a call and starts the next 47 // segment of a builder type call. 48 static bool startsSegmentOfBuilderTypeCall(const FormatToken &Tok) { 49 return Tok.isMemberAccess() && Tok.Previous && Tok.Previous->closesScope(); 50 } 51 52 // Returns \c true if \c Current starts a new parameter. 53 static bool startsNextParameter(const FormatToken &Current, 54 const FormatStyle &Style) { 55 const FormatToken &Previous = *Current.Previous; 56 if (Current.is(TT_CtorInitializerComma) && 57 Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma) 58 return true; 59 if (Style.Language == FormatStyle::LK_Proto && Current.is(TT_SelectorName)) 60 return true; 61 return Previous.is(tok::comma) && !Current.isTrailingComment() && 62 ((Previous.isNot(TT_CtorInitializerComma) || 63 Style.BreakConstructorInitializers != 64 FormatStyle::BCIS_BeforeComma) && 65 (Previous.isNot(TT_InheritanceComma) || 66 !Style.BreakBeforeInheritanceComma)); 67 } 68 69 static 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 79 ContinuationIndenter::ContinuationIndenter(const FormatStyle &Style, 80 const AdditionalKeywords &Keywords, 81 const SourceManager &SourceMgr, 82 WhitespaceManager &Whitespaces, 83 encoding::Encoding Encoding, 84 bool BinPackInconclusiveFunctions) 85 : Style(Style), Keywords(Keywords), SourceMgr(SourceMgr), 86 Whitespaces(Whitespaces), Encoding(Encoding), 87 BinPackInconclusiveFunctions(BinPackInconclusiveFunctions), 88 CommentPragmasRegex(Style.CommentPragmas) {} 89 90 LineState ContinuationIndenter::getInitialState(unsigned FirstIndent, 91 const AnnotatedLine *Line, 92 bool DryRun) { 93 LineState State; 94 State.FirstIndent = FirstIndent; 95 State.Column = FirstIndent; 96 // 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; 103 State.Line = Line; 104 State.NextToken = Line->First; 105 State.Stack.push_back(ParenState(FirstIndent, FirstIndent, 106 /*AvoidBinPacking=*/false, 107 /*NoLineBreak=*/false)); 108 State.LineContainsContinuedForLoopSection = false; 109 State.NoContinuation = false; 110 State.StartOfStringLiteral = 0; 111 State.StartOfLineLevel = 0; 112 State.LowestLevelOnLine = 0; 113 State.IgnoreStackForComparison = false; 114 115 if (Style.Language == FormatStyle::LK_TextProto) { 116 // We need this in order to deal with the bin packing of text fields at 117 // global scope. 118 State.Stack.back().AvoidBinPacking = true; 119 State.Stack.back().BreakBeforeParameter = true; 120 } 121 122 // The first token has already been indented and thus consumed. 123 moveStateToNextToken(State, DryRun, /*Newline=*/false); 124 return State; 125 } 126 127 bool ContinuationIndenter::canBreak(const LineState &State) { 128 const FormatToken &Current = *State.NextToken; 129 const FormatToken &Previous = *Current.Previous; 130 assert(&Previous == Current.Previous); 131 if (!Current.CanBreakBefore && !(State.Stack.back().BreakBeforeClosingBrace && 132 Current.closesBlockOrBlockTypeList(Style))) 133 return false; 134 // The opening "{" of a braced list has to be on the same line as the first 135 // element if it is nested in another braced init list or function call. 136 if (!Current.MustBreakBefore && Previous.is(tok::l_brace) && 137 Previous.isNot(TT_DictLiteral) && Previous.BlockKind == BK_BracedInit && 138 Previous.Previous && 139 Previous.Previous->isOneOf(tok::l_brace, tok::l_paren, tok::comma)) 140 return false; 141 // This prevents breaks like: 142 // ... 143 // SomeParameter, OtherParameter).DoSomething( 144 // ... 145 // As they hide "DoSomething" and are generally bad for readability. 146 if (Previous.opensScope() && Previous.isNot(tok::l_brace) && 147 State.LowestLevelOnLine < State.StartOfLineLevel && 148 State.LowestLevelOnLine < Current.NestingLevel) 149 return false; 150 if (Current.isMemberAccess() && State.Stack.back().ContainsUnwrappedBuilder) 151 return false; 152 153 // Don't create a 'hanging' indent if there are multiple blocks in a single 154 // statement. 155 if (Previous.is(tok::l_brace) && State.Stack.size() > 1 && 156 State.Stack[State.Stack.size() - 2].NestedBlockInlined && 157 State.Stack[State.Stack.size() - 2].HasMultipleNestedBlocks) 158 return false; 159 160 // Don't break after very short return types (e.g. "void") as that is often 161 // unexpected. 162 if (Current.is(TT_FunctionDeclarationName) && State.Column < 6) { 163 if (Style.AlwaysBreakAfterReturnType == FormatStyle::RTBS_None) 164 return false; 165 } 166 167 // If binary operators are moved to the next line (including commas for some 168 // styles of constructor initializers), that's always ok. 169 if (!Current.isOneOf(TT_BinaryOperator, tok::comma) && 170 State.Stack.back().NoLineBreakInOperand) 171 return false; 172 173 return !State.Stack.back().NoLineBreak; 174 } 175 176 bool ContinuationIndenter::mustBreak(const LineState &State) { 177 const FormatToken &Current = *State.NextToken; 178 const FormatToken &Previous = *Current.Previous; 179 if (Current.MustBreakBefore || Current.is(TT_InlineASMColon)) 180 return true; 181 if (State.Stack.back().BreakBeforeClosingBrace && 182 Current.closesBlockOrBlockTypeList(Style)) 183 return true; 184 if (Previous.is(tok::semi) && State.LineContainsContinuedForLoopSection) 185 return true; 186 if ((startsNextParameter(Current, Style) || Previous.is(tok::semi) || 187 (Previous.is(TT_TemplateCloser) && Current.is(TT_StartOfName) && 188 Style.isCpp() && 189 // FIXME: This is a temporary workaround for the case where clang-format 190 // sets BreakBeforeParameter to avoid bin packing and this creates a 191 // completely unnecessary line break after a template type that isn't 192 // line-wrapped. 193 (Previous.NestingLevel == 1 || Style.BinPackParameters)) || 194 (Style.BreakBeforeTernaryOperators && Current.is(TT_ConditionalExpr) && 195 Previous.isNot(tok::question)) || 196 (!Style.BreakBeforeTernaryOperators && 197 Previous.is(TT_ConditionalExpr))) && 198 State.Stack.back().BreakBeforeParameter && !Current.isTrailingComment() && 199 !Current.isOneOf(tok::r_paren, tok::r_brace)) 200 return true; 201 if (((Previous.is(TT_DictLiteral) && Previous.is(tok::l_brace)) || 202 (Previous.is(TT_ArrayInitializerLSquare) && 203 Previous.ParameterCount > 1) || 204 opensProtoMessageField(Previous, Style)) && 205 Style.ColumnLimit > 0 && 206 getLengthToMatchingParen(Previous) + State.Column - 1 > 207 getColumnLimit(State)) 208 return true; 209 210 const FormatToken &BreakConstructorInitializersToken = 211 Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon 212 ? Previous 213 : Current; 214 if (BreakConstructorInitializersToken.is(TT_CtorInitializerColon) && 215 (State.Column + State.Line->Last->TotalLength - Previous.TotalLength > 216 getColumnLimit(State) || 217 State.Stack.back().BreakBeforeParameter) && 218 (Style.AllowShortFunctionsOnASingleLine != FormatStyle::SFS_All || 219 Style.BreakConstructorInitializers != FormatStyle::BCIS_BeforeColon || 220 Style.ColumnLimit != 0)) 221 return true; 222 223 if (Current.is(TT_ObjCMethodExpr) && !Previous.is(TT_SelectorName) && 224 State.Line->startsWith(TT_ObjCMethodSpecifier)) 225 return true; 226 if (Current.is(TT_SelectorName) && State.Stack.back().ObjCSelectorNameFound && 227 State.Stack.back().BreakBeforeParameter) 228 return true; 229 230 unsigned NewLineColumn = getNewLineColumn(State); 231 if (Current.isMemberAccess() && Style.ColumnLimit != 0 && 232 State.Column + getLengthToNextOperator(Current) > Style.ColumnLimit && 233 (State.Column > NewLineColumn || 234 Current.NestingLevel < State.StartOfLineLevel)) 235 return true; 236 237 if (startsSegmentOfBuilderTypeCall(Current) && 238 (State.Stack.back().CallContinuation != 0 || 239 State.Stack.back().BreakBeforeParameter) && 240 // JavaScript is treated different here as there is a frequent pattern: 241 // SomeFunction(function() { 242 // ... 243 // }.bind(...)); 244 // FIXME: We should find a more generic solution to this problem. 245 !(State.Column <= NewLineColumn && 246 Style.Language == FormatStyle::LK_JavaScript)) 247 return true; 248 249 if (State.Column <= NewLineColumn) 250 return false; 251 252 if (Style.AlwaysBreakBeforeMultilineStrings && 253 (NewLineColumn == State.FirstIndent + Style.ContinuationIndentWidth || 254 Previous.is(tok::comma) || Current.NestingLevel < 2) && 255 !Previous.isOneOf(tok::kw_return, tok::lessless, tok::at) && 256 !Previous.isOneOf(TT_InlineASMColon, TT_ConditionalExpr) && 257 nextIsMultilineString(State)) 258 return true; 259 260 // Using CanBreakBefore here and below takes care of the decision whether the 261 // current style uses wrapping before or after operators for the given 262 // operator. 263 if (Previous.is(TT_BinaryOperator) && Current.CanBreakBefore) { 264 // If we need to break somewhere inside the LHS of a binary expression, we 265 // should also break after the operator. Otherwise, the formatting would 266 // hide the operator precedence, e.g. in: 267 // if (aaaaaaaaaaaaaa == 268 // bbbbbbbbbbbbbb && c) {.. 269 // For comparisons, we only apply this rule, if the LHS is a binary 270 // expression itself as otherwise, the line breaks seem superfluous. 271 // We need special cases for ">>" which we have split into two ">" while 272 // lexing in order to make template parsing easier. 273 bool IsComparison = (Previous.getPrecedence() == prec::Relational || 274 Previous.getPrecedence() == prec::Equality) && 275 Previous.Previous && 276 Previous.Previous->isNot(TT_BinaryOperator); // For >>. 277 bool LHSIsBinaryExpr = 278 Previous.Previous && Previous.Previous->EndsBinaryExpression; 279 if ((!IsComparison || LHSIsBinaryExpr) && !Current.isTrailingComment() && 280 Previous.getPrecedence() != prec::Assignment && 281 State.Stack.back().BreakBeforeParameter) 282 return true; 283 } else if (Current.is(TT_BinaryOperator) && Current.CanBreakBefore && 284 State.Stack.back().BreakBeforeParameter) { 285 return true; 286 } 287 288 // Same as above, but for the first "<<" operator. 289 if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator) && 290 State.Stack.back().BreakBeforeParameter && 291 State.Stack.back().FirstLessLess == 0) 292 return true; 293 294 if (Current.NestingLevel == 0 && !Current.isTrailingComment()) { 295 // Always break after "template <...>" and leading annotations. This is only 296 // for cases where the entire line does not fit on a single line as a 297 // different LineFormatter would be used otherwise. 298 if (Previous.ClosesTemplateDeclaration) 299 return true; 300 if (Previous.is(TT_FunctionAnnotationRParen)) 301 return true; 302 if (Previous.is(TT_LeadingJavaAnnotation) && Current.isNot(tok::l_paren) && 303 Current.isNot(TT_LeadingJavaAnnotation)) 304 return true; 305 } 306 307 // If the return type spans multiple lines, wrap before the function name. 308 if ((Current.is(TT_FunctionDeclarationName) || 309 (Current.is(tok::kw_operator) && !Previous.is(tok::coloncolon))) && 310 !Previous.is(tok::kw_template) && State.Stack.back().BreakBeforeParameter) 311 return true; 312 313 // The following could be precomputed as they do not depend on the state. 314 // However, as they should take effect only if the UnwrappedLine does not fit 315 // into the ColumnLimit, they are checked here in the ContinuationIndenter. 316 if (Style.ColumnLimit != 0 && Previous.BlockKind == BK_Block && 317 Previous.is(tok::l_brace) && !Current.isOneOf(tok::r_brace, tok::comment)) 318 return true; 319 320 if (Current.is(tok::lessless) && 321 ((Previous.is(tok::identifier) && Previous.TokenText == "endl") || 322 (Previous.Tok.isLiteral() && (Previous.TokenText.endswith("\\n\"") || 323 Previous.TokenText == "\'\\n\'")))) 324 return true; 325 326 if (Previous.is(TT_BlockComment) && Previous.IsMultiline) 327 return true; 328 329 if (State.NoContinuation) 330 return true; 331 332 return false; 333 } 334 335 unsigned ContinuationIndenter::addTokenToState(LineState &State, bool Newline, 336 bool DryRun, 337 unsigned ExtraSpaces) { 338 const FormatToken &Current = *State.NextToken; 339 340 assert(!State.Stack.empty()); 341 State.NoContinuation = false; 342 343 if ((Current.is(TT_ImplicitStringLiteral) && 344 (Current.Previous->Tok.getIdentifierInfo() == nullptr || 345 Current.Previous->Tok.getIdentifierInfo()->getPPKeywordID() == 346 tok::pp_not_keyword))) { 347 unsigned EndColumn = 348 SourceMgr.getSpellingColumnNumber(Current.WhitespaceRange.getEnd()); 349 if (Current.LastNewlineOffset != 0) { 350 // If there is a newline within this token, the final column will solely 351 // determined by the current end column. 352 State.Column = EndColumn; 353 } else { 354 unsigned StartColumn = 355 SourceMgr.getSpellingColumnNumber(Current.WhitespaceRange.getBegin()); 356 assert(EndColumn >= StartColumn); 357 State.Column += EndColumn - StartColumn; 358 } 359 moveStateToNextToken(State, DryRun, /*Newline=*/false); 360 return 0; 361 } 362 363 unsigned Penalty = 0; 364 if (Newline) 365 Penalty = addTokenOnNewLine(State, DryRun); 366 else 367 addTokenOnCurrentLine(State, DryRun, ExtraSpaces); 368 369 return moveStateToNextToken(State, DryRun, Newline) + Penalty; 370 } 371 372 void ContinuationIndenter::addTokenOnCurrentLine(LineState &State, bool DryRun, 373 unsigned ExtraSpaces) { 374 FormatToken &Current = *State.NextToken; 375 const FormatToken &Previous = *State.NextToken->Previous; 376 if (Current.is(tok::equal) && 377 (State.Line->First->is(tok::kw_for) || Current.NestingLevel == 0) && 378 State.Stack.back().VariablePos == 0) { 379 State.Stack.back().VariablePos = State.Column; 380 // Move over * and & if they are bound to the variable name. 381 const FormatToken *Tok = &Previous; 382 while (Tok && State.Stack.back().VariablePos >= Tok->ColumnWidth) { 383 State.Stack.back().VariablePos -= Tok->ColumnWidth; 384 if (Tok->SpacesRequiredBefore != 0) 385 break; 386 Tok = Tok->Previous; 387 } 388 if (Previous.PartOfMultiVariableDeclStmt) 389 State.Stack.back().LastSpace = State.Stack.back().VariablePos; 390 } 391 392 unsigned Spaces = Current.SpacesRequiredBefore + ExtraSpaces; 393 394 // Indent preprocessor directives after the hash if required. 395 int PPColumnCorrection = 0; 396 if (Style.IndentPPDirectives == FormatStyle::PPDIS_AfterHash && 397 Previous.is(tok::hash) && State.FirstIndent > 0 && 398 (State.Line->Type == LT_PreprocessorDirective || 399 State.Line->Type == LT_ImportStatement)) { 400 Spaces += State.FirstIndent; 401 402 // For preprocessor indent with tabs, State.Column will be 1 because of the 403 // hash. This causes second-level indents onward to have an extra space 404 // after the tabs. We avoid this misalignment by subtracting 1 from the 405 // column value passed to replaceWhitespace(). 406 if (Style.UseTab != FormatStyle::UT_Never) 407 PPColumnCorrection = -1; 408 } 409 410 if (!DryRun) 411 Whitespaces.replaceWhitespace(Current, /*Newlines=*/0, Spaces, 412 State.Column + Spaces + PPColumnCorrection); 413 414 // If "BreakBeforeInheritanceComma" mode, don't break within the inheritance 415 // declaration unless there is multiple inheritance. 416 if (Style.BreakBeforeInheritanceComma && Current.is(TT_InheritanceColon)) 417 State.Stack.back().NoLineBreak = true; 418 419 if (Current.is(TT_SelectorName) && 420 !State.Stack.back().ObjCSelectorNameFound) { 421 unsigned MinIndent = 422 std::max(State.FirstIndent + Style.ContinuationIndentWidth, 423 State.Stack.back().Indent); 424 unsigned FirstColonPos = State.Column + Spaces + Current.ColumnWidth; 425 if (Current.LongestObjCSelectorName == 0) 426 State.Stack.back().AlignColons = false; 427 else if (MinIndent + Current.LongestObjCSelectorName > FirstColonPos) 428 State.Stack.back().ColonPos = MinIndent + Current.LongestObjCSelectorName; 429 else 430 State.Stack.back().ColonPos = FirstColonPos; 431 } 432 433 // In "AlwaysBreak" mode, enforce wrapping directly after the parenthesis by 434 // disallowing any further line breaks if there is no line break after the 435 // opening parenthesis. Don't break if it doesn't conserve columns. 436 if (Style.AlignAfterOpenBracket == FormatStyle::BAS_AlwaysBreak && 437 Previous.isOneOf(tok::l_paren, TT_TemplateOpener, tok::l_square) && 438 State.Column > getNewLineColumn(State) && 439 (!Previous.Previous || !Previous.Previous->isOneOf( 440 tok::kw_for, tok::kw_while, tok::kw_switch)) && 441 // Don't do this for simple (no expressions) one-argument function calls 442 // as that feels like needlessly wasting whitespace, e.g.: 443 // 444 // caaaaaaaaaaaall( 445 // caaaaaaaaaaaall( 446 // caaaaaaaaaaaall( 447 // caaaaaaaaaaaaaaaaaaaaaaall(aaaaaaaaaaaaaa, aaaaaaaaa)))); 448 Current.FakeLParens.size() > 0 && 449 Current.FakeLParens.back() > prec::Unknown) 450 State.Stack.back().NoLineBreak = true; 451 if (Previous.is(TT_TemplateString) && Previous.opensScope()) 452 State.Stack.back().NoLineBreak = true; 453 454 if (Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign && 455 Previous.opensScope() && Previous.isNot(TT_ObjCMethodExpr) && 456 (Current.isNot(TT_LineComment) || Previous.BlockKind == BK_BracedInit)) 457 State.Stack.back().Indent = State.Column + Spaces; 458 if (State.Stack.back().AvoidBinPacking && startsNextParameter(Current, Style)) 459 State.Stack.back().NoLineBreak = true; 460 if (startsSegmentOfBuilderTypeCall(Current) && 461 State.Column > getNewLineColumn(State)) 462 State.Stack.back().ContainsUnwrappedBuilder = true; 463 464 if (Current.is(TT_LambdaArrow) && Style.Language == FormatStyle::LK_Java) 465 State.Stack.back().NoLineBreak = true; 466 if (Current.isMemberAccess() && Previous.is(tok::r_paren) && 467 (Previous.MatchingParen && 468 (Previous.TotalLength - Previous.MatchingParen->TotalLength > 10))) 469 // If there is a function call with long parameters, break before trailing 470 // calls. This prevents things like: 471 // EXPECT_CALL(SomeLongParameter).Times( 472 // 2); 473 // We don't want to do this for short parameters as they can just be 474 // indexes. 475 State.Stack.back().NoLineBreak = true; 476 477 // Don't allow the RHS of an operator to be split over multiple lines unless 478 // there is a line-break right after the operator. 479 // Exclude relational operators, as there, it is always more desirable to 480 // have the LHS 'left' of the RHS. 481 const FormatToken *P = Current.getPreviousNonComment(); 482 if (!Current.is(tok::comment) && P && 483 (P->isOneOf(TT_BinaryOperator, tok::comma) || 484 (P->is(TT_ConditionalExpr) && P->is(tok::colon))) && 485 !P->isOneOf(TT_OverloadedOperator, TT_CtorInitializerComma) && 486 P->getPrecedence() != prec::Assignment && 487 P->getPrecedence() != prec::Relational) { 488 bool BreakBeforeOperator = 489 P->MustBreakBefore || P->is(tok::lessless) || 490 (P->is(TT_BinaryOperator) && 491 Style.BreakBeforeBinaryOperators != FormatStyle::BOS_None) || 492 (P->is(TT_ConditionalExpr) && Style.BreakBeforeTernaryOperators); 493 // Don't do this if there are only two operands. In these cases, there is 494 // always a nice vertical separation between them and the extra line break 495 // does not help. 496 bool HasTwoOperands = 497 P->OperatorIndex == 0 && !P->NextOperator && !P->is(TT_ConditionalExpr); 498 if ((!BreakBeforeOperator && !(HasTwoOperands && Style.AlignOperands)) || 499 (!State.Stack.back().LastOperatorWrapped && BreakBeforeOperator)) 500 State.Stack.back().NoLineBreakInOperand = true; 501 } 502 503 State.Column += Spaces; 504 if (Current.isNot(tok::comment) && Previous.is(tok::l_paren) && 505 Previous.Previous && 506 (Previous.Previous->isOneOf(tok::kw_if, tok::kw_for) || 507 Previous.Previous->endsSequence(tok::kw_constexpr, tok::kw_if))) { 508 // Treat the condition inside an if as if it was a second function 509 // parameter, i.e. let nested calls have a continuation indent. 510 State.Stack.back().LastSpace = State.Column; 511 State.Stack.back().NestedBlockIndent = State.Column; 512 } else if (!Current.isOneOf(tok::comment, tok::caret) && 513 ((Previous.is(tok::comma) && 514 !Previous.is(TT_OverloadedOperator)) || 515 (Previous.is(tok::colon) && Previous.is(TT_ObjCMethodExpr)))) { 516 State.Stack.back().LastSpace = State.Column; 517 } else if (Previous.is(TT_CtorInitializerColon) && 518 Style.BreakConstructorInitializers == 519 FormatStyle::BCIS_AfterColon) { 520 State.Stack.back().Indent = State.Column; 521 State.Stack.back().LastSpace = State.Column; 522 } else if ((Previous.isOneOf(TT_BinaryOperator, TT_ConditionalExpr, 523 TT_CtorInitializerColon)) && 524 ((Previous.getPrecedence() != prec::Assignment && 525 (Previous.isNot(tok::lessless) || Previous.OperatorIndex != 0 || 526 Previous.NextOperator)) || 527 Current.StartsBinaryExpression)) { 528 // Indent relative to the RHS of the expression unless this is a simple 529 // assignment without binary expression on the RHS. Also indent relative to 530 // unary operators and the colons of constructor initializers. 531 State.Stack.back().LastSpace = State.Column; 532 } else if (Previous.is(TT_InheritanceColon)) { 533 State.Stack.back().Indent = State.Column; 534 State.Stack.back().LastSpace = State.Column; 535 } else if (Previous.opensScope()) { 536 // If a function has a trailing call, indent all parameters from the 537 // opening parenthesis. This avoids confusing indents like: 538 // OuterFunction(InnerFunctionCall( // break 539 // ParameterToInnerFunction)) // break 540 // .SecondInnerFunctionCall(); 541 bool HasTrailingCall = false; 542 if (Previous.MatchingParen) { 543 const FormatToken *Next = Previous.MatchingParen->getNextNonComment(); 544 HasTrailingCall = Next && Next->isMemberAccess(); 545 } 546 if (HasTrailingCall && State.Stack.size() > 1 && 547 State.Stack[State.Stack.size() - 2].CallContinuation == 0) 548 State.Stack.back().LastSpace = State.Column; 549 } 550 } 551 552 unsigned ContinuationIndenter::addTokenOnNewLine(LineState &State, 553 bool DryRun) { 554 FormatToken &Current = *State.NextToken; 555 const FormatToken &Previous = *State.NextToken->Previous; 556 557 // Extra penalty that needs to be added because of the way certain line 558 // breaks are chosen. 559 unsigned Penalty = 0; 560 561 const FormatToken *PreviousNonComment = Current.getPreviousNonComment(); 562 const FormatToken *NextNonComment = Previous.getNextNonComment(); 563 if (!NextNonComment) 564 NextNonComment = &Current; 565 // The first line break on any NestingLevel causes an extra penalty in order 566 // prefer similar line breaks. 567 if (!State.Stack.back().ContainsLineBreak) 568 Penalty += 15; 569 State.Stack.back().ContainsLineBreak = true; 570 571 Penalty += State.NextToken->SplitPenalty; 572 573 // Breaking before the first "<<" is generally not desirable if the LHS is 574 // short. Also always add the penalty if the LHS is split over multiple lines 575 // to avoid unnecessary line breaks that just work around this penalty. 576 if (NextNonComment->is(tok::lessless) && 577 State.Stack.back().FirstLessLess == 0 && 578 (State.Column <= Style.ColumnLimit / 3 || 579 State.Stack.back().BreakBeforeParameter)) 580 Penalty += Style.PenaltyBreakFirstLessLess; 581 582 State.Column = getNewLineColumn(State); 583 584 // Indent nested blocks relative to this column, unless in a very specific 585 // JavaScript special case where: 586 // 587 // var loooooong_name = 588 // function() { 589 // // code 590 // } 591 // 592 // is common and should be formatted like a free-standing function. The same 593 // goes for wrapping before the lambda return type arrow. 594 if (!Current.is(TT_LambdaArrow) && 595 (Style.Language != FormatStyle::LK_JavaScript || 596 Current.NestingLevel != 0 || !PreviousNonComment || 597 !PreviousNonComment->is(tok::equal) || 598 !Current.isOneOf(Keywords.kw_async, Keywords.kw_function))) 599 State.Stack.back().NestedBlockIndent = State.Column; 600 601 if (NextNonComment->isMemberAccess()) { 602 if (State.Stack.back().CallContinuation == 0) 603 State.Stack.back().CallContinuation = State.Column; 604 } else if (NextNonComment->is(TT_SelectorName)) { 605 if (!State.Stack.back().ObjCSelectorNameFound) { 606 if (NextNonComment->LongestObjCSelectorName == 0) { 607 State.Stack.back().AlignColons = false; 608 } else { 609 State.Stack.back().ColonPos = 610 (Style.IndentWrappedFunctionNames 611 ? std::max(State.Stack.back().Indent, 612 State.FirstIndent + Style.ContinuationIndentWidth) 613 : State.Stack.back().Indent) + 614 NextNonComment->LongestObjCSelectorName; 615 } 616 } else if (State.Stack.back().AlignColons && 617 State.Stack.back().ColonPos <= NextNonComment->ColumnWidth) { 618 State.Stack.back().ColonPos = State.Column + NextNonComment->ColumnWidth; 619 } 620 } else if (PreviousNonComment && PreviousNonComment->is(tok::colon) && 621 PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) { 622 // FIXME: This is hacky, find a better way. The problem is that in an ObjC 623 // method expression, the block should be aligned to the line starting it, 624 // e.g.: 625 // [aaaaaaaaaaaaaaa aaaaaaaaa: \\ break for some reason 626 // ^(int *i) { 627 // // ... 628 // }]; 629 // Thus, we set LastSpace of the next higher NestingLevel, to which we move 630 // when we consume all of the "}"'s FakeRParens at the "{". 631 if (State.Stack.size() > 1) 632 State.Stack[State.Stack.size() - 2].LastSpace = 633 std::max(State.Stack.back().LastSpace, State.Stack.back().Indent) + 634 Style.ContinuationIndentWidth; 635 } 636 637 if ((PreviousNonComment && 638 PreviousNonComment->isOneOf(tok::comma, tok::semi) && 639 !State.Stack.back().AvoidBinPacking) || 640 Previous.is(TT_BinaryOperator)) 641 State.Stack.back().BreakBeforeParameter = false; 642 if (Previous.isOneOf(TT_TemplateCloser, TT_JavaAnnotation) && 643 Current.NestingLevel == 0) 644 State.Stack.back().BreakBeforeParameter = false; 645 if (NextNonComment->is(tok::question) || 646 (PreviousNonComment && PreviousNonComment->is(tok::question))) 647 State.Stack.back().BreakBeforeParameter = true; 648 if (Current.is(TT_BinaryOperator) && Current.CanBreakBefore) 649 State.Stack.back().BreakBeforeParameter = false; 650 651 if (!DryRun) { 652 unsigned Newlines = std::max( 653 1u, std::min(Current.NewlinesBefore, Style.MaxEmptyLinesToKeep + 1)); 654 bool ContinuePPDirective = 655 State.Line->InPPDirective && State.Line->Type != LT_ImportStatement; 656 Whitespaces.replaceWhitespace(Current, Newlines, State.Column, State.Column, 657 ContinuePPDirective); 658 } 659 660 if (!Current.isTrailingComment()) 661 State.Stack.back().LastSpace = State.Column; 662 if (Current.is(tok::lessless)) 663 // If we are breaking before a "<<", we always want to indent relative to 664 // RHS. This is necessary only for "<<", as we special-case it and don't 665 // always indent relative to the RHS. 666 State.Stack.back().LastSpace += 3; // 3 -> width of "<< ". 667 668 State.StartOfLineLevel = Current.NestingLevel; 669 State.LowestLevelOnLine = Current.NestingLevel; 670 671 // Any break on this level means that the parent level has been broken 672 // and we need to avoid bin packing there. 673 bool NestedBlockSpecialCase = 674 !Style.isCpp() && Current.is(tok::r_brace) && State.Stack.size() > 1 && 675 State.Stack[State.Stack.size() - 2].NestedBlockInlined; 676 if (!NestedBlockSpecialCase) 677 for (unsigned i = 0, e = State.Stack.size() - 1; i != e; ++i) 678 State.Stack[i].BreakBeforeParameter = true; 679 680 if (PreviousNonComment && 681 !PreviousNonComment->isOneOf(tok::comma, tok::colon, tok::semi) && 682 (PreviousNonComment->isNot(TT_TemplateCloser) || 683 Current.NestingLevel != 0) && 684 !PreviousNonComment->isOneOf( 685 TT_BinaryOperator, TT_FunctionAnnotationRParen, TT_JavaAnnotation, 686 TT_LeadingJavaAnnotation) && 687 Current.isNot(TT_BinaryOperator) && !PreviousNonComment->opensScope()) 688 State.Stack.back().BreakBeforeParameter = true; 689 690 // If we break after { or the [ of an array initializer, we should also break 691 // before the corresponding } or ]. 692 if (PreviousNonComment && 693 (PreviousNonComment->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) || 694 opensProtoMessageField(*PreviousNonComment, Style))) 695 State.Stack.back().BreakBeforeClosingBrace = true; 696 697 if (State.Stack.back().AvoidBinPacking) { 698 // If we are breaking after '(', '{', '<', this is not bin packing 699 // unless AllowAllParametersOfDeclarationOnNextLine is false or this is a 700 // dict/object literal. 701 if (!Previous.isOneOf(tok::l_paren, tok::l_brace, TT_BinaryOperator) || 702 (!Style.AllowAllParametersOfDeclarationOnNextLine && 703 State.Line->MustBeDeclaration) || 704 Previous.is(TT_DictLiteral)) 705 State.Stack.back().BreakBeforeParameter = true; 706 } 707 708 return Penalty; 709 } 710 711 unsigned ContinuationIndenter::getNewLineColumn(const LineState &State) { 712 if (!State.NextToken || !State.NextToken->Previous) 713 return 0; 714 FormatToken &Current = *State.NextToken; 715 const FormatToken &Previous = *Current.Previous; 716 // If we are continuing an expression, we want to use the continuation indent. 717 unsigned ContinuationIndent = 718 std::max(State.Stack.back().LastSpace, State.Stack.back().Indent) + 719 Style.ContinuationIndentWidth; 720 const FormatToken *PreviousNonComment = Current.getPreviousNonComment(); 721 const FormatToken *NextNonComment = Previous.getNextNonComment(); 722 if (!NextNonComment) 723 NextNonComment = &Current; 724 725 // Java specific bits. 726 if (Style.Language == FormatStyle::LK_Java && 727 Current.isOneOf(Keywords.kw_implements, Keywords.kw_extends)) 728 return std::max(State.Stack.back().LastSpace, 729 State.Stack.back().Indent + Style.ContinuationIndentWidth); 730 731 if (NextNonComment->is(tok::l_brace) && NextNonComment->BlockKind == BK_Block) 732 return Current.NestingLevel == 0 ? State.FirstIndent 733 : State.Stack.back().Indent; 734 if ((Current.isOneOf(tok::r_brace, tok::r_square) || 735 (Current.is(tok::greater) && 736 (Style.Language == FormatStyle::LK_Proto || 737 Style.Language == FormatStyle::LK_TextProto))) && 738 State.Stack.size() > 1) { 739 if (Current.closesBlockOrBlockTypeList(Style)) 740 return State.Stack[State.Stack.size() - 2].NestedBlockIndent; 741 if (Current.MatchingParen && 742 Current.MatchingParen->BlockKind == BK_BracedInit) 743 return State.Stack[State.Stack.size() - 2].LastSpace; 744 return State.FirstIndent; 745 } 746 // Indent a closing parenthesis at the previous level if followed by a semi or 747 // opening brace. This allows indentations such as: 748 // foo( 749 // a, 750 // ); 751 // function foo( 752 // a, 753 // ) { 754 // code(); // 755 // } 756 if (Current.is(tok::r_paren) && State.Stack.size() > 1 && 757 (!Current.Next || Current.Next->isOneOf(tok::semi, tok::l_brace))) 758 return State.Stack[State.Stack.size() - 2].LastSpace; 759 if (NextNonComment->is(TT_TemplateString) && NextNonComment->closesScope()) 760 return State.Stack[State.Stack.size() - 2].LastSpace; 761 if (Current.is(tok::identifier) && Current.Next && 762 (Current.Next->is(TT_DictLiteral) || 763 ((Style.Language == FormatStyle::LK_Proto || 764 Style.Language == FormatStyle::LK_TextProto) && 765 Current.Next->isOneOf(TT_TemplateOpener, tok::l_brace)))) 766 return State.Stack.back().Indent; 767 if (NextNonComment->is(TT_ObjCStringLiteral) && 768 State.StartOfStringLiteral != 0) 769 return State.StartOfStringLiteral - 1; 770 if (NextNonComment->isStringLiteral() && State.StartOfStringLiteral != 0) 771 return State.StartOfStringLiteral; 772 if (NextNonComment->is(tok::lessless) && 773 State.Stack.back().FirstLessLess != 0) 774 return State.Stack.back().FirstLessLess; 775 if (NextNonComment->isMemberAccess()) { 776 if (State.Stack.back().CallContinuation == 0) 777 return ContinuationIndent; 778 return State.Stack.back().CallContinuation; 779 } 780 if (State.Stack.back().QuestionColumn != 0 && 781 ((NextNonComment->is(tok::colon) && 782 NextNonComment->is(TT_ConditionalExpr)) || 783 Previous.is(TT_ConditionalExpr))) 784 return State.Stack.back().QuestionColumn; 785 if (Previous.is(tok::comma) && State.Stack.back().VariablePos != 0) 786 return State.Stack.back().VariablePos; 787 if ((PreviousNonComment && 788 (PreviousNonComment->ClosesTemplateDeclaration || 789 PreviousNonComment->isOneOf( 790 TT_AttributeParen, TT_FunctionAnnotationRParen, TT_JavaAnnotation, 791 TT_LeadingJavaAnnotation))) || 792 (!Style.IndentWrappedFunctionNames && 793 NextNonComment->isOneOf(tok::kw_operator, TT_FunctionDeclarationName))) 794 return std::max(State.Stack.back().LastSpace, State.Stack.back().Indent); 795 if (NextNonComment->is(TT_SelectorName)) { 796 if (!State.Stack.back().ObjCSelectorNameFound) { 797 if (NextNonComment->LongestObjCSelectorName == 0) 798 return State.Stack.back().Indent; 799 return (Style.IndentWrappedFunctionNames 800 ? std::max(State.Stack.back().Indent, 801 State.FirstIndent + Style.ContinuationIndentWidth) 802 : State.Stack.back().Indent) + 803 NextNonComment->LongestObjCSelectorName - 804 NextNonComment->ColumnWidth; 805 } 806 if (!State.Stack.back().AlignColons) 807 return State.Stack.back().Indent; 808 if (State.Stack.back().ColonPos > NextNonComment->ColumnWidth) 809 return State.Stack.back().ColonPos - NextNonComment->ColumnWidth; 810 return State.Stack.back().Indent; 811 } 812 if (NextNonComment->is(tok::colon) && NextNonComment->is(TT_ObjCMethodExpr)) 813 return State.Stack.back().ColonPos; 814 if (NextNonComment->is(TT_ArraySubscriptLSquare)) { 815 if (State.Stack.back().StartOfArraySubscripts != 0) 816 return State.Stack.back().StartOfArraySubscripts; 817 return ContinuationIndent; 818 } 819 820 // This ensure that we correctly format ObjC methods calls without inputs, 821 // i.e. where the last element isn't selector like: [callee method]; 822 if (NextNonComment->is(tok::identifier) && NextNonComment->FakeRParens == 0 && 823 NextNonComment->Next && NextNonComment->Next->is(TT_ObjCMethodExpr)) 824 return State.Stack.back().Indent; 825 826 if (NextNonComment->isOneOf(TT_StartOfName, TT_PointerOrReference) || 827 Previous.isOneOf(tok::coloncolon, tok::equal, TT_JsTypeColon)) 828 return ContinuationIndent; 829 if (PreviousNonComment && PreviousNonComment->is(tok::colon) && 830 PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) 831 return ContinuationIndent; 832 if (NextNonComment->is(TT_CtorInitializerComma)) 833 return State.Stack.back().Indent; 834 if (PreviousNonComment && PreviousNonComment->is(TT_CtorInitializerColon) && 835 Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon) 836 return State.Stack.back().Indent; 837 if (NextNonComment->isOneOf(TT_CtorInitializerColon, TT_InheritanceColon, 838 TT_InheritanceComma)) 839 return State.FirstIndent + Style.ConstructorInitializerIndentWidth; 840 if (Previous.is(tok::r_paren) && !Current.isBinaryOperator() && 841 !Current.isOneOf(tok::colon, tok::comment)) 842 return ContinuationIndent; 843 if (State.Stack.back().Indent == State.FirstIndent && PreviousNonComment && 844 PreviousNonComment->isNot(tok::r_brace)) 845 // Ensure that we fall back to the continuation indent width instead of 846 // just flushing continuations left. 847 return State.Stack.back().Indent + Style.ContinuationIndentWidth; 848 return State.Stack.back().Indent; 849 } 850 851 unsigned ContinuationIndenter::moveStateToNextToken(LineState &State, 852 bool DryRun, bool Newline) { 853 assert(State.Stack.size()); 854 const FormatToken &Current = *State.NextToken; 855 856 if (Current.isOneOf(tok::comma, TT_BinaryOperator)) 857 State.Stack.back().NoLineBreakInOperand = false; 858 if (Current.is(TT_InheritanceColon)) 859 State.Stack.back().AvoidBinPacking = true; 860 if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator)) { 861 if (State.Stack.back().FirstLessLess == 0) 862 State.Stack.back().FirstLessLess = State.Column; 863 else 864 State.Stack.back().LastOperatorWrapped = Newline; 865 } 866 if (Current.is(TT_BinaryOperator) && Current.isNot(tok::lessless)) 867 State.Stack.back().LastOperatorWrapped = Newline; 868 if (Current.is(TT_ConditionalExpr) && Current.Previous && 869 !Current.Previous->is(TT_ConditionalExpr)) 870 State.Stack.back().LastOperatorWrapped = Newline; 871 if (Current.is(TT_ArraySubscriptLSquare) && 872 State.Stack.back().StartOfArraySubscripts == 0) 873 State.Stack.back().StartOfArraySubscripts = State.Column; 874 if (Style.BreakBeforeTernaryOperators && Current.is(tok::question)) 875 State.Stack.back().QuestionColumn = State.Column; 876 if (!Style.BreakBeforeTernaryOperators && Current.isNot(tok::colon)) { 877 const FormatToken *Previous = Current.Previous; 878 while (Previous && Previous->isTrailingComment()) 879 Previous = Previous->Previous; 880 if (Previous && Previous->is(tok::question)) 881 State.Stack.back().QuestionColumn = State.Column; 882 } 883 if (!Current.opensScope() && !Current.closesScope() && 884 !Current.is(TT_PointerOrReference)) 885 State.LowestLevelOnLine = 886 std::min(State.LowestLevelOnLine, Current.NestingLevel); 887 if (Current.isMemberAccess()) 888 State.Stack.back().StartOfFunctionCall = 889 !Current.NextOperator ? 0 : State.Column; 890 if (Current.is(TT_SelectorName)) { 891 State.Stack.back().ObjCSelectorNameFound = true; 892 if (Style.IndentWrappedFunctionNames) { 893 State.Stack.back().Indent = 894 State.FirstIndent + Style.ContinuationIndentWidth; 895 } 896 } 897 if (Current.is(TT_CtorInitializerColon) && 898 Style.BreakConstructorInitializers != FormatStyle::BCIS_AfterColon) { 899 // Indent 2 from the column, so: 900 // SomeClass::SomeClass() 901 // : First(...), ... 902 // Next(...) 903 // ^ line up here. 904 State.Stack.back().Indent = 905 State.Column + 906 (Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma 907 ? 0 908 : 2); 909 State.Stack.back().NestedBlockIndent = State.Stack.back().Indent; 910 if (Style.ConstructorInitializerAllOnOneLineOrOnePerLine) 911 State.Stack.back().AvoidBinPacking = true; 912 State.Stack.back().BreakBeforeParameter = false; 913 } 914 if (Current.is(TT_CtorInitializerColon) && 915 Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon) { 916 State.Stack.back().Indent = 917 State.FirstIndent + Style.ConstructorInitializerIndentWidth; 918 State.Stack.back().NestedBlockIndent = State.Stack.back().Indent; 919 if (Style.ConstructorInitializerAllOnOneLineOrOnePerLine) 920 State.Stack.back().AvoidBinPacking = true; 921 } 922 if (Current.is(TT_InheritanceColon)) 923 State.Stack.back().Indent = 924 State.FirstIndent + Style.ContinuationIndentWidth; 925 if (Current.isOneOf(TT_BinaryOperator, TT_ConditionalExpr) && Newline) 926 State.Stack.back().NestedBlockIndent = 927 State.Column + Current.ColumnWidth + 1; 928 if (Current.isOneOf(TT_LambdaLSquare, TT_LambdaArrow)) 929 State.Stack.back().LastSpace = State.Column; 930 931 // Insert scopes created by fake parenthesis. 932 const FormatToken *Previous = Current.getPreviousNonComment(); 933 934 // Add special behavior to support a format commonly used for JavaScript 935 // closures: 936 // SomeFunction(function() { 937 // foo(); 938 // bar(); 939 // }, a, b, c); 940 if (Current.isNot(tok::comment) && Previous && 941 Previous->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) && 942 !Previous->is(TT_DictLiteral) && State.Stack.size() > 1) { 943 if (State.Stack[State.Stack.size() - 2].NestedBlockInlined && Newline) 944 for (unsigned i = 0, e = State.Stack.size() - 1; i != e; ++i) 945 State.Stack[i].NoLineBreak = true; 946 State.Stack[State.Stack.size() - 2].NestedBlockInlined = false; 947 } 948 if (Previous && 949 (Previous->isOneOf(tok::l_paren, tok::comma, tok::colon) || 950 Previous->isOneOf(TT_BinaryOperator, TT_ConditionalExpr)) && 951 !Previous->isOneOf(TT_DictLiteral, TT_ObjCMethodExpr)) { 952 State.Stack.back().NestedBlockInlined = 953 !Newline && 954 (Previous->isNot(tok::l_paren) || Previous->ParameterCount > 1); 955 } 956 957 moveStatePastFakeLParens(State, Newline); 958 moveStatePastScopeCloser(State); 959 bool CanBreakProtrudingToken = !State.Stack.back().NoLineBreak && 960 !State.Stack.back().NoLineBreakInOperand; 961 moveStatePastScopeOpener(State, Newline); 962 moveStatePastFakeRParens(State); 963 964 if (Current.is(TT_ObjCStringLiteral) && State.StartOfStringLiteral == 0) 965 State.StartOfStringLiteral = State.Column + 1; 966 else if (Current.isStringLiteral() && State.StartOfStringLiteral == 0) 967 State.StartOfStringLiteral = State.Column; 968 else if (!Current.isOneOf(tok::comment, tok::identifier, tok::hash) && 969 !Current.isStringLiteral()) 970 State.StartOfStringLiteral = 0; 971 972 State.Column += Current.ColumnWidth; 973 State.NextToken = State.NextToken->Next; 974 unsigned Penalty = 0; 975 if (CanBreakProtrudingToken) 976 Penalty = breakProtrudingToken(Current, State, DryRun); 977 if (State.Column > getColumnLimit(State)) { 978 unsigned ExcessCharacters = State.Column - getColumnLimit(State); 979 Penalty += Style.PenaltyExcessCharacter * ExcessCharacters; 980 } 981 982 if (Current.Role) 983 Current.Role->formatFromToken(State, this, DryRun); 984 // If the previous has a special role, let it consume tokens as appropriate. 985 // It is necessary to start at the previous token for the only implemented 986 // role (comma separated list). That way, the decision whether or not to break 987 // after the "{" is already done and both options are tried and evaluated. 988 // FIXME: This is ugly, find a better way. 989 if (Previous && Previous->Role) 990 Penalty += Previous->Role->formatAfterToken(State, this, DryRun); 991 992 return Penalty; 993 } 994 995 void ContinuationIndenter::moveStatePastFakeLParens(LineState &State, 996 bool Newline) { 997 const FormatToken &Current = *State.NextToken; 998 const FormatToken *Previous = Current.getPreviousNonComment(); 999 1000 // Don't add extra indentation for the first fake parenthesis after 1001 // 'return', assignments or opening <({[. The indentation for these cases 1002 // is special cased. 1003 bool SkipFirstExtraIndent = 1004 (Previous && (Previous->opensScope() || 1005 Previous->isOneOf(tok::semi, tok::kw_return) || 1006 (Previous->getPrecedence() == prec::Assignment && 1007 Style.AlignOperands) || 1008 Previous->is(TT_ObjCMethodExpr))); 1009 for (SmallVectorImpl<prec::Level>::const_reverse_iterator 1010 I = Current.FakeLParens.rbegin(), 1011 E = Current.FakeLParens.rend(); 1012 I != E; ++I) { 1013 ParenState NewParenState = State.Stack.back(); 1014 NewParenState.ContainsLineBreak = false; 1015 NewParenState.LastOperatorWrapped = true; 1016 NewParenState.NoLineBreak = 1017 NewParenState.NoLineBreak || State.Stack.back().NoLineBreakInOperand; 1018 1019 // Don't propagate AvoidBinPacking into subexpressions of arg/param lists. 1020 if (*I > prec::Comma) 1021 NewParenState.AvoidBinPacking = false; 1022 1023 // Indent from 'LastSpace' unless these are fake parentheses encapsulating 1024 // a builder type call after 'return' or, if the alignment after opening 1025 // brackets is disabled. 1026 if (!Current.isTrailingComment() && 1027 (Style.AlignOperands || *I < prec::Assignment) && 1028 (!Previous || Previous->isNot(tok::kw_return) || 1029 (Style.Language != FormatStyle::LK_Java && *I > 0)) && 1030 (Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign || 1031 *I != prec::Comma || Current.NestingLevel == 0)) 1032 NewParenState.Indent = 1033 std::max(std::max(State.Column, NewParenState.Indent), 1034 State.Stack.back().LastSpace); 1035 1036 // Do not indent relative to the fake parentheses inserted for "." or "->". 1037 // This is a special case to make the following to statements consistent: 1038 // OuterFunction(InnerFunctionCall( // break 1039 // ParameterToInnerFunction)); 1040 // OuterFunction(SomeObject.InnerFunctionCall( // break 1041 // ParameterToInnerFunction)); 1042 if (*I > prec::Unknown) 1043 NewParenState.LastSpace = std::max(NewParenState.LastSpace, State.Column); 1044 if (*I != prec::Conditional && !Current.is(TT_UnaryOperator) && 1045 Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign) 1046 NewParenState.StartOfFunctionCall = State.Column; 1047 1048 // Always indent conditional expressions. Never indent expression where 1049 // the 'operator' is ',', ';' or an assignment (i.e. *I <= 1050 // prec::Assignment) as those have different indentation rules. Indent 1051 // other expression, unless the indentation needs to be skipped. 1052 if (*I == prec::Conditional || 1053 (!SkipFirstExtraIndent && *I > prec::Assignment && 1054 !Current.isTrailingComment())) 1055 NewParenState.Indent += Style.ContinuationIndentWidth; 1056 if ((Previous && !Previous->opensScope()) || *I != prec::Comma) 1057 NewParenState.BreakBeforeParameter = false; 1058 State.Stack.push_back(NewParenState); 1059 SkipFirstExtraIndent = false; 1060 } 1061 } 1062 1063 void ContinuationIndenter::moveStatePastFakeRParens(LineState &State) { 1064 for (unsigned i = 0, e = State.NextToken->FakeRParens; i != e; ++i) { 1065 unsigned VariablePos = State.Stack.back().VariablePos; 1066 if (State.Stack.size() == 1) { 1067 // Do not pop the last element. 1068 break; 1069 } 1070 State.Stack.pop_back(); 1071 State.Stack.back().VariablePos = VariablePos; 1072 } 1073 } 1074 1075 void ContinuationIndenter::moveStatePastScopeOpener(LineState &State, 1076 bool Newline) { 1077 const FormatToken &Current = *State.NextToken; 1078 if (!Current.opensScope()) 1079 return; 1080 1081 if (Current.MatchingParen && Current.BlockKind == BK_Block) { 1082 moveStateToNewBlock(State); 1083 return; 1084 } 1085 1086 unsigned NewIndent; 1087 unsigned LastSpace = State.Stack.back().LastSpace; 1088 bool AvoidBinPacking; 1089 bool BreakBeforeParameter = false; 1090 unsigned NestedBlockIndent = std::max(State.Stack.back().StartOfFunctionCall, 1091 State.Stack.back().NestedBlockIndent); 1092 if (Current.isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) || 1093 opensProtoMessageField(Current, Style)) { 1094 if (Current.opensBlockOrBlockTypeList(Style)) { 1095 NewIndent = Style.IndentWidth + 1096 std::min(State.Column, State.Stack.back().NestedBlockIndent); 1097 } else { 1098 NewIndent = State.Stack.back().LastSpace + Style.ContinuationIndentWidth; 1099 } 1100 const FormatToken *NextNoComment = Current.getNextNonComment(); 1101 bool EndsInComma = Current.MatchingParen && 1102 Current.MatchingParen->Previous && 1103 Current.MatchingParen->Previous->is(tok::comma); 1104 AvoidBinPacking = EndsInComma || Current.is(TT_DictLiteral) || 1105 Style.Language == FormatStyle::LK_Proto || 1106 Style.Language == FormatStyle::LK_TextProto || 1107 !Style.BinPackArguments || 1108 (NextNoComment && 1109 NextNoComment->isOneOf(TT_DesignatedInitializerPeriod, 1110 TT_DesignatedInitializerLSquare)); 1111 BreakBeforeParameter = EndsInComma; 1112 if (Current.ParameterCount > 1) 1113 NestedBlockIndent = std::max(NestedBlockIndent, State.Column + 1); 1114 } else { 1115 NewIndent = Style.ContinuationIndentWidth + 1116 std::max(State.Stack.back().LastSpace, 1117 State.Stack.back().StartOfFunctionCall); 1118 1119 // Ensure that different different brackets force relative alignment, e.g.: 1120 // void SomeFunction(vector< // break 1121 // int> v); 1122 // FIXME: We likely want to do this for more combinations of brackets. 1123 // Verify that it is wanted for ObjC, too. 1124 if (Current.is(tok::less) && Current.ParentBracket == tok::l_paren) { 1125 NewIndent = std::max(NewIndent, State.Stack.back().Indent); 1126 LastSpace = std::max(LastSpace, State.Stack.back().Indent); 1127 } 1128 1129 bool EndsInComma = 1130 Current.MatchingParen && 1131 Current.MatchingParen->getPreviousNonComment() && 1132 Current.MatchingParen->getPreviousNonComment()->is(tok::comma); 1133 1134 AvoidBinPacking = 1135 (Style.Language == FormatStyle::LK_JavaScript && EndsInComma) || 1136 (State.Line->MustBeDeclaration && !Style.BinPackParameters) || 1137 (!State.Line->MustBeDeclaration && !Style.BinPackArguments) || 1138 (Style.ExperimentalAutoDetectBinPacking && 1139 (Current.PackingKind == PPK_OnePerLine || 1140 (!BinPackInconclusiveFunctions && 1141 Current.PackingKind == PPK_Inconclusive))); 1142 1143 if (Current.is(TT_ObjCMethodExpr) && Current.MatchingParen) { 1144 if (Style.ColumnLimit) { 1145 // If this '[' opens an ObjC call, determine whether all parameters fit 1146 // into one line and put one per line if they don't. 1147 if (getLengthToMatchingParen(Current) + State.Column > 1148 getColumnLimit(State)) 1149 BreakBeforeParameter = true; 1150 } else { 1151 // For ColumnLimit = 0, we have to figure out whether there is or has to 1152 // be a line break within this call. 1153 for (const FormatToken *Tok = &Current; 1154 Tok && Tok != Current.MatchingParen; Tok = Tok->Next) { 1155 if (Tok->MustBreakBefore || 1156 (Tok->CanBreakBefore && Tok->NewlinesBefore > 0)) { 1157 BreakBeforeParameter = true; 1158 break; 1159 } 1160 } 1161 } 1162 } 1163 1164 if (Style.Language == FormatStyle::LK_JavaScript && EndsInComma) 1165 BreakBeforeParameter = true; 1166 } 1167 // Generally inherit NoLineBreak from the current scope to nested scope. 1168 // However, don't do this for non-empty nested blocks, dict literals and 1169 // array literals as these follow different indentation rules. 1170 bool NoLineBreak = 1171 Current.Children.empty() && 1172 !Current.isOneOf(TT_DictLiteral, TT_ArrayInitializerLSquare) && 1173 (State.Stack.back().NoLineBreak || 1174 State.Stack.back().NoLineBreakInOperand || 1175 (Current.is(TT_TemplateOpener) && 1176 State.Stack.back().ContainsUnwrappedBuilder)); 1177 State.Stack.push_back( 1178 ParenState(NewIndent, LastSpace, AvoidBinPacking, NoLineBreak)); 1179 State.Stack.back().NestedBlockIndent = NestedBlockIndent; 1180 State.Stack.back().BreakBeforeParameter = BreakBeforeParameter; 1181 State.Stack.back().HasMultipleNestedBlocks = Current.BlockParameterCount > 1; 1182 } 1183 1184 void ContinuationIndenter::moveStatePastScopeCloser(LineState &State) { 1185 const FormatToken &Current = *State.NextToken; 1186 if (!Current.closesScope()) 1187 return; 1188 1189 // If we encounter a closing ), ], } or >, we can remove a level from our 1190 // stacks. 1191 if (State.Stack.size() > 1 && 1192 (Current.isOneOf(tok::r_paren, tok::r_square, TT_TemplateString) || 1193 (Current.is(tok::r_brace) && State.NextToken != State.Line->First) || 1194 State.NextToken->is(TT_TemplateCloser))) 1195 State.Stack.pop_back(); 1196 1197 if (Current.is(tok::r_square)) { 1198 // If this ends the array subscript expr, reset the corresponding value. 1199 const FormatToken *NextNonComment = Current.getNextNonComment(); 1200 if (NextNonComment && NextNonComment->isNot(tok::l_square)) 1201 State.Stack.back().StartOfArraySubscripts = 0; 1202 } 1203 } 1204 1205 void ContinuationIndenter::moveStateToNewBlock(LineState &State) { 1206 unsigned NestedBlockIndent = State.Stack.back().NestedBlockIndent; 1207 // ObjC block sometimes follow special indentation rules. 1208 unsigned NewIndent = 1209 NestedBlockIndent + (State.NextToken->is(TT_ObjCBlockLBrace) 1210 ? Style.ObjCBlockIndentWidth 1211 : Style.IndentWidth); 1212 State.Stack.push_back(ParenState(NewIndent, State.Stack.back().LastSpace, 1213 /*AvoidBinPacking=*/true, 1214 /*NoLineBreak=*/false)); 1215 State.Stack.back().NestedBlockIndent = NestedBlockIndent; 1216 State.Stack.back().BreakBeforeParameter = true; 1217 } 1218 1219 unsigned ContinuationIndenter::addMultilineToken(const FormatToken &Current, 1220 LineState &State) { 1221 if (!Current.IsMultiline) 1222 return 0; 1223 1224 // Break before further function parameters on all levels. 1225 for (unsigned i = 0, e = State.Stack.size(); i != e; ++i) 1226 State.Stack[i].BreakBeforeParameter = true; 1227 1228 unsigned ColumnsUsed = State.Column; 1229 // We can only affect layout of the first and the last line, so the penalty 1230 // for all other lines is constant, and we ignore it. 1231 State.Column = Current.LastLineColumnWidth; 1232 1233 if (ColumnsUsed > getColumnLimit(State)) 1234 return Style.PenaltyExcessCharacter * (ColumnsUsed - getColumnLimit(State)); 1235 return 0; 1236 } 1237 1238 unsigned ContinuationIndenter::breakProtrudingToken(const FormatToken &Current, 1239 LineState &State, 1240 bool DryRun) { 1241 // Don't break multi-line tokens other than block comments. Instead, just 1242 // update the state. 1243 if (Current.isNot(TT_BlockComment) && Current.IsMultiline) 1244 return addMultilineToken(Current, State); 1245 1246 // Don't break implicit string literals or import statements. 1247 if (Current.is(TT_ImplicitStringLiteral) || 1248 State.Line->Type == LT_ImportStatement) 1249 return 0; 1250 1251 if (!Current.isStringLiteral() && !Current.is(tok::comment)) 1252 return 0; 1253 1254 std::unique_ptr<BreakableToken> Token; 1255 unsigned StartColumn = State.Column - Current.ColumnWidth; 1256 unsigned ColumnLimit = getColumnLimit(State); 1257 1258 if (Current.isStringLiteral()) { 1259 // FIXME: String literal breaking is currently disabled for Java and JS, as 1260 // it requires strings to be merged using "+" which we don't support. 1261 if (Style.Language == FormatStyle::LK_Java || 1262 Style.Language == FormatStyle::LK_JavaScript || 1263 !Style.BreakStringLiterals) 1264 return 0; 1265 1266 // Don't break string literals inside preprocessor directives (except for 1267 // #define directives, as their contents are stored in separate lines and 1268 // are not affected by this check). 1269 // This way we avoid breaking code with line directives and unknown 1270 // preprocessor directives that contain long string literals. 1271 if (State.Line->Type == LT_PreprocessorDirective) 1272 return 0; 1273 // Exempts unterminated string literals from line breaking. The user will 1274 // likely want to terminate the string before any line breaking is done. 1275 if (Current.IsUnterminatedLiteral) 1276 return 0; 1277 1278 StringRef Text = Current.TokenText; 1279 StringRef Prefix; 1280 StringRef Postfix; 1281 // FIXME: Handle whitespace between '_T', '(', '"..."', and ')'. 1282 // FIXME: Store Prefix and Suffix (or PrefixLength and SuffixLength to 1283 // reduce the overhead) for each FormatToken, which is a string, so that we 1284 // don't run multiple checks here on the hot path. 1285 if ((Text.endswith(Postfix = "\"") && 1286 (Text.startswith(Prefix = "@\"") || Text.startswith(Prefix = "\"") || 1287 Text.startswith(Prefix = "u\"") || Text.startswith(Prefix = "U\"") || 1288 Text.startswith(Prefix = "u8\"") || 1289 Text.startswith(Prefix = "L\""))) || 1290 (Text.startswith(Prefix = "_T(\"") && Text.endswith(Postfix = "\")"))) { 1291 Token.reset(new BreakableStringLiteral(Current, StartColumn, Prefix, 1292 Postfix, State.Line->InPPDirective, 1293 Encoding, Style)); 1294 } else { 1295 return 0; 1296 } 1297 } else if (Current.is(TT_BlockComment)) { 1298 if (!Style.ReflowComments || 1299 // If a comment token switches formatting, like 1300 // /* clang-format on */, we don't want to break it further, 1301 // but we may still want to adjust its indentation. 1302 switchesFormatting(Current)) 1303 return addMultilineToken(Current, State); 1304 Token.reset(new BreakableBlockComment( 1305 Current, StartColumn, Current.OriginalColumn, !Current.Previous, 1306 State.Line->InPPDirective, Encoding, Style)); 1307 } else if (Current.is(TT_LineComment) && 1308 (Current.Previous == nullptr || 1309 Current.Previous->isNot(TT_ImplicitStringLiteral))) { 1310 if (!Style.ReflowComments || 1311 CommentPragmasRegex.match(Current.TokenText.substr(2)) || 1312 switchesFormatting(Current)) 1313 return 0; 1314 Token.reset(new BreakableLineCommentSection( 1315 Current, StartColumn, Current.OriginalColumn, !Current.Previous, 1316 /*InPPDirective=*/false, Encoding, Style)); 1317 // We don't insert backslashes when breaking line comments. 1318 ColumnLimit = Style.ColumnLimit; 1319 } else { 1320 return 0; 1321 } 1322 if (Current.UnbreakableTailLength >= ColumnLimit) 1323 return 0; 1324 1325 unsigned RemainingSpace = ColumnLimit - Current.UnbreakableTailLength; 1326 bool BreakInserted = false; 1327 // We use a conservative reflowing strategy. Reflow starts after a line is 1328 // broken or the corresponding whitespace compressed. Reflow ends as soon as a 1329 // line that doesn't get reflown with the previous line is reached. 1330 bool ReflowInProgress = false; 1331 unsigned Penalty = 0; 1332 unsigned RemainingTokenColumns = 0; 1333 unsigned TailOffset = 0; 1334 for (unsigned LineIndex = 0, EndIndex = Token->getLineCount(); 1335 LineIndex != EndIndex; ++LineIndex) { 1336 BreakableToken::Split SplitBefore(StringRef::npos, 0); 1337 if (ReflowInProgress) { 1338 SplitBefore = Token->getSplitBefore(LineIndex, RemainingTokenColumns, 1339 RemainingSpace, CommentPragmasRegex); 1340 } 1341 ReflowInProgress = SplitBefore.first != StringRef::npos; 1342 TailOffset = 1343 ReflowInProgress ? (SplitBefore.first + SplitBefore.second) : 0; 1344 BreakInserted = BreakInserted || Token->introducesBreakBefore(LineIndex); 1345 if (!DryRun) 1346 Token->replaceWhitespaceBefore(LineIndex, RemainingTokenColumns, 1347 RemainingSpace, SplitBefore, Whitespaces); 1348 RemainingTokenColumns = Token->getLineLengthAfterSplitBefore( 1349 LineIndex, TailOffset, RemainingTokenColumns, ColumnLimit, SplitBefore); 1350 while (RemainingTokenColumns > RemainingSpace) { 1351 BreakableToken::Split Split = Token->getSplit( 1352 LineIndex, TailOffset, ColumnLimit, CommentPragmasRegex); 1353 if (Split.first == StringRef::npos) { 1354 // The last line's penalty is handled in addNextStateToQueue(). 1355 if (LineIndex < EndIndex - 1) 1356 Penalty += Style.PenaltyExcessCharacter * 1357 (RemainingTokenColumns - RemainingSpace); 1358 break; 1359 } 1360 assert(Split.first != 0); 1361 1362 // Check if compressing the whitespace range will bring the line length 1363 // under the limit. If that is the case, we perform whitespace compression 1364 // instead of inserting a line break. 1365 unsigned RemainingTokenColumnsAfterCompression = 1366 Token->getLineLengthAfterCompression(RemainingTokenColumns, Split); 1367 if (RemainingTokenColumnsAfterCompression <= RemainingSpace) { 1368 RemainingTokenColumns = RemainingTokenColumnsAfterCompression; 1369 ReflowInProgress = true; 1370 if (!DryRun) 1371 Token->compressWhitespace(LineIndex, TailOffset, Split, Whitespaces); 1372 break; 1373 } 1374 1375 unsigned NewRemainingTokenColumns = Token->getLineLengthAfterSplit( 1376 LineIndex, TailOffset + Split.first + Split.second, StringRef::npos); 1377 1378 // When breaking before a tab character, it may be moved by a few columns, 1379 // but will still be expanded to the next tab stop, so we don't save any 1380 // columns. 1381 if (NewRemainingTokenColumns == RemainingTokenColumns) 1382 break; 1383 1384 assert(NewRemainingTokenColumns < RemainingTokenColumns); 1385 if (!DryRun) 1386 Token->insertBreak(LineIndex, TailOffset, Split, Whitespaces); 1387 Penalty += Current.SplitPenalty; 1388 unsigned ColumnsUsed = 1389 Token->getLineLengthAfterSplit(LineIndex, TailOffset, Split.first); 1390 if (ColumnsUsed > ColumnLimit) { 1391 Penalty += Style.PenaltyExcessCharacter * (ColumnsUsed - ColumnLimit); 1392 } 1393 TailOffset += Split.first + Split.second; 1394 RemainingTokenColumns = NewRemainingTokenColumns; 1395 ReflowInProgress = true; 1396 BreakInserted = true; 1397 } 1398 } 1399 1400 BreakableToken::Split SplitAfterLastLine = 1401 Token->getSplitAfterLastLine(TailOffset, ColumnLimit); 1402 if (SplitAfterLastLine.first != StringRef::npos) { 1403 if (!DryRun) 1404 Token->replaceWhitespaceAfterLastLine(TailOffset, SplitAfterLastLine, 1405 Whitespaces); 1406 RemainingTokenColumns = Token->getLineLengthAfterSplitAfterLastLine( 1407 TailOffset, SplitAfterLastLine); 1408 } 1409 1410 State.Column = RemainingTokenColumns; 1411 1412 if (BreakInserted) { 1413 // If we break the token inside a parameter list, we need to break before 1414 // the next parameter on all levels, so that the next parameter is clearly 1415 // visible. Line comments already introduce a break. 1416 if (Current.isNot(TT_LineComment)) { 1417 for (unsigned i = 0, e = State.Stack.size(); i != e; ++i) 1418 State.Stack[i].BreakBeforeParameter = true; 1419 } 1420 1421 if (Current.is(TT_BlockComment)) 1422 State.NoContinuation = true; 1423 1424 Penalty += Current.isStringLiteral() ? Style.PenaltyBreakString 1425 : Style.PenaltyBreakComment; 1426 1427 State.Stack.back().LastSpace = StartColumn; 1428 } 1429 1430 Token->updateNextToken(State); 1431 1432 return Penalty; 1433 } 1434 1435 unsigned ContinuationIndenter::getColumnLimit(const LineState &State) const { 1436 // In preprocessor directives reserve two chars for trailing " \" 1437 return Style.ColumnLimit - (State.Line->InPPDirective ? 2 : 0); 1438 } 1439 1440 bool ContinuationIndenter::nextIsMultilineString(const LineState &State) { 1441 const FormatToken &Current = *State.NextToken; 1442 if (!Current.isStringLiteral() || Current.is(TT_ImplicitStringLiteral)) 1443 return false; 1444 // We never consider raw string literals "multiline" for the purpose of 1445 // AlwaysBreakBeforeMultilineStrings implementation as they are special-cased 1446 // (see TokenAnnotator::mustBreakBefore(). 1447 if (Current.TokenText.startswith("R\"")) 1448 return false; 1449 if (Current.IsMultiline) 1450 return true; 1451 if (Current.getNextNonComment() && 1452 Current.getNextNonComment()->isStringLiteral()) 1453 return true; // Implicit concatenation. 1454 if (Style.ColumnLimit != 0 && 1455 State.Column + Current.ColumnWidth + Current.UnbreakableTailLength > 1456 Style.ColumnLimit) 1457 return true; // String will be split. 1458 return false; 1459 } 1460 1461 } // namespace format 1462 } // namespace clang 1463