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 "BreakableToken.h" 16 #include "ContinuationIndenter.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 #include <string> 23 24 #define DEBUG_TYPE "format-formatter" 25 26 namespace clang { 27 namespace format { 28 29 // Returns the length of everything up to the first possible line break after 30 // the ), ], } or > matching \c Tok. 31 static unsigned getLengthToMatchingParen(const FormatToken &Tok) { 32 if (!Tok.MatchingParen) 33 return 0; 34 FormatToken *End = Tok.MatchingParen; 35 while (End->Next && !End->Next->CanBreakBefore) { 36 End = End->Next; 37 } 38 return End->TotalLength - Tok.TotalLength + 1; 39 } 40 41 // Returns \c true if \c Tok is the "." or "->" of a call and starts the next 42 // segment of a builder type call. 43 static bool startsSegmentOfBuilderTypeCall(const FormatToken &Tok) { 44 return Tok.isMemberAccess() && Tok.Previous && Tok.Previous->closesScope(); 45 } 46 47 // Returns \c true if \c Current starts a new parameter. 48 static bool startsNextParameter(const FormatToken &Current, 49 const FormatStyle &Style) { 50 const FormatToken &Previous = *Current.Previous; 51 if (Current.is(TT_CtorInitializerComma) && 52 Style.BreakConstructorInitializersBeforeComma) 53 return true; 54 return Previous.is(tok::comma) && !Current.isTrailingComment() && 55 (Previous.isNot(TT_CtorInitializerComma) || 56 !Style.BreakConstructorInitializersBeforeComma); 57 } 58 59 ContinuationIndenter::ContinuationIndenter(const FormatStyle &Style, 60 const AdditionalKeywords &Keywords, 61 SourceManager &SourceMgr, 62 WhitespaceManager &Whitespaces, 63 encoding::Encoding Encoding, 64 bool BinPackInconclusiveFunctions) 65 : Style(Style), Keywords(Keywords), SourceMgr(SourceMgr), 66 Whitespaces(Whitespaces), Encoding(Encoding), 67 BinPackInconclusiveFunctions(BinPackInconclusiveFunctions), 68 CommentPragmasRegex(Style.CommentPragmas) {} 69 70 LineState ContinuationIndenter::getInitialState(unsigned FirstIndent, 71 const AnnotatedLine *Line, 72 bool DryRun) { 73 LineState State; 74 State.FirstIndent = FirstIndent; 75 State.Column = FirstIndent; 76 State.Line = Line; 77 State.NextToken = Line->First; 78 State.Stack.push_back(ParenState(FirstIndent, Line->Level, FirstIndent, 79 /*AvoidBinPacking=*/false, 80 /*NoLineBreak=*/false)); 81 State.LineContainsContinuedForLoopSection = false; 82 State.StartOfStringLiteral = 0; 83 State.StartOfLineLevel = 0; 84 State.LowestLevelOnLine = 0; 85 State.IgnoreStackForComparison = false; 86 87 // The first token has already been indented and thus consumed. 88 moveStateToNextToken(State, DryRun, /*Newline=*/false); 89 return State; 90 } 91 92 bool ContinuationIndenter::canBreak(const LineState &State) { 93 const FormatToken &Current = *State.NextToken; 94 const FormatToken &Previous = *Current.Previous; 95 assert(&Previous == Current.Previous); 96 if (!Current.CanBreakBefore && !(State.Stack.back().BreakBeforeClosingBrace && 97 Current.closesBlockTypeList(Style))) 98 return false; 99 // The opening "{" of a braced list has to be on the same line as the first 100 // element if it is nested in another braced init list or function call. 101 if (!Current.MustBreakBefore && Previous.is(tok::l_brace) && 102 Previous.isNot(TT_DictLiteral) && Previous.BlockKind == BK_BracedInit && 103 Previous.Previous && 104 Previous.Previous->isOneOf(tok::l_brace, tok::l_paren, tok::comma)) 105 return false; 106 // This prevents breaks like: 107 // ... 108 // SomeParameter, OtherParameter).DoSomething( 109 // ... 110 // As they hide "DoSomething" and are generally bad for readability. 111 if (Previous.opensScope() && Previous.isNot(tok::l_brace) && 112 State.LowestLevelOnLine < State.StartOfLineLevel && 113 State.LowestLevelOnLine < Current.NestingLevel) 114 return false; 115 if (Current.isMemberAccess() && State.Stack.back().ContainsUnwrappedBuilder) 116 return false; 117 118 // Don't create a 'hanging' indent if there are multiple blocks in a single 119 // statement. 120 if (Previous.is(tok::l_brace) && State.Stack.size() > 1 && 121 State.Stack[State.Stack.size() - 2].NestedBlockInlined && 122 State.Stack[State.Stack.size() - 2].HasMultipleNestedBlocks) 123 return false; 124 125 // Don't break after very short return types (e.g. "void") as that is often 126 // unexpected. 127 if (Current.is(TT_FunctionDeclarationName) && 128 !Style.AlwaysBreakAfterDefinitionReturnType && State.Column < 6) 129 return false; 130 131 return !State.Stack.back().NoLineBreak; 132 } 133 134 bool ContinuationIndenter::mustBreak(const LineState &State) { 135 const FormatToken &Current = *State.NextToken; 136 const FormatToken &Previous = *Current.Previous; 137 if (Current.MustBreakBefore || Current.is(TT_InlineASMColon)) 138 return true; 139 if (State.Stack.back().BreakBeforeClosingBrace && 140 Current.closesBlockTypeList(Style)) 141 return true; 142 if (Previous.is(tok::semi) && State.LineContainsContinuedForLoopSection) 143 return true; 144 if ((startsNextParameter(Current, Style) || Previous.is(tok::semi) || 145 (Style.BreakBeforeTernaryOperators && 146 (Current.is(tok::question) || 147 (Current.is(TT_ConditionalExpr) && Previous.isNot(tok::question)))) || 148 (!Style.BreakBeforeTernaryOperators && 149 (Previous.is(tok::question) || Previous.is(TT_ConditionalExpr)))) && 150 State.Stack.back().BreakBeforeParameter && !Current.isTrailingComment() && 151 !Current.isOneOf(tok::r_paren, tok::r_brace)) 152 return true; 153 if (Style.AlwaysBreakBeforeMultilineStrings && 154 State.Column > State.Stack.back().Indent && // Breaking saves columns. 155 !Previous.isOneOf(tok::kw_return, tok::lessless, tok::at) && 156 !Previous.isOneOf(TT_InlineASMColon, TT_ConditionalExpr) && 157 nextIsMultilineString(State)) 158 return true; 159 if (((Previous.is(TT_DictLiteral) && Previous.is(tok::l_brace)) || 160 Previous.is(TT_ArrayInitializerLSquare)) && 161 Style.ColumnLimit > 0 && 162 getLengthToMatchingParen(Previous) + State.Column > getColumnLimit(State)) 163 return true; 164 if (Current.is(TT_CtorInitializerColon) && 165 ((Style.AllowShortFunctionsOnASingleLine != FormatStyle::SFS_All) || 166 Style.BreakConstructorInitializersBeforeComma || Style.ColumnLimit != 0)) 167 return true; 168 169 if (State.Column < getNewLineColumn(State)) 170 return false; 171 if (Style.BreakBeforeBinaryOperators == FormatStyle::BOS_None) { 172 // If we need to break somewhere inside the LHS of a binary expression, we 173 // should also break after the operator. Otherwise, the formatting would 174 // hide the operator precedence, e.g. in: 175 // if (aaaaaaaaaaaaaa == 176 // bbbbbbbbbbbbbb && c) {.. 177 // For comparisons, we only apply this rule, if the LHS is a binary 178 // expression itself as otherwise, the line breaks seem superfluous. 179 // We need special cases for ">>" which we have split into two ">" while 180 // lexing in order to make template parsing easier. 181 bool IsComparison = (Previous.getPrecedence() == prec::Relational || 182 Previous.getPrecedence() == prec::Equality) && 183 Previous.Previous && 184 Previous.Previous->isNot(TT_BinaryOperator); // For >>. 185 bool LHSIsBinaryExpr = 186 Previous.Previous && Previous.Previous->EndsBinaryExpression; 187 if (Previous.is(TT_BinaryOperator) && (!IsComparison || LHSIsBinaryExpr) && 188 Current.isNot(TT_BinaryOperator) && // For >>. 189 !Current.isTrailingComment() && !Previous.is(tok::lessless) && 190 Previous.getPrecedence() != prec::Assignment && 191 State.Stack.back().BreakBeforeParameter) 192 return true; 193 } else { 194 if (Current.is(TT_BinaryOperator) && Previous.EndsBinaryExpression && 195 State.Stack.back().BreakBeforeParameter) 196 return true; 197 } 198 199 200 // Same as above, but for the first "<<" operator. 201 if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator) && 202 State.Stack.back().BreakBeforeParameter && 203 State.Stack.back().FirstLessLess == 0) 204 return true; 205 206 if (Current.is(TT_SelectorName) && State.Stack.back().ObjCSelectorNameFound && 207 State.Stack.back().BreakBeforeParameter) 208 return true; 209 if (Current.NestingLevel == 0 && !Current.isTrailingComment()) { 210 if (Previous.ClosesTemplateDeclaration) 211 return true; 212 if (Previous.is(TT_LeadingJavaAnnotation) && Current.isNot(tok::l_paren)) 213 return true; 214 } 215 216 // If the return type spans multiple lines, wrap before the function name. 217 if (Current.isOneOf(TT_FunctionDeclarationName ,tok::kw_operator) && 218 State.Stack.back().BreakBeforeParameter) 219 return true; 220 221 if (startsSegmentOfBuilderTypeCall(Current) && 222 (State.Stack.back().CallContinuation != 0 || 223 (State.Stack.back().BreakBeforeParameter && 224 State.Stack.back().ContainsUnwrappedBuilder))) 225 return true; 226 227 // The following could be precomputed as they do not depend on the state. 228 // However, as they should take effect only if the UnwrappedLine does not fit 229 // into the ColumnLimit, they are checked here in the ContinuationIndenter. 230 if (Style.ColumnLimit != 0 && Previous.BlockKind == BK_Block && 231 Previous.is(tok::l_brace) && !Current.isOneOf(tok::r_brace, tok::comment)) 232 return true; 233 234 return false; 235 } 236 237 unsigned ContinuationIndenter::addTokenToState(LineState &State, bool Newline, 238 bool DryRun, 239 unsigned ExtraSpaces) { 240 const FormatToken &Current = *State.NextToken; 241 242 assert(!State.Stack.empty()); 243 if ((Current.is(TT_ImplicitStringLiteral) && 244 (Current.Previous->Tok.getIdentifierInfo() == nullptr || 245 Current.Previous->Tok.getIdentifierInfo()->getPPKeywordID() == 246 tok::pp_not_keyword))) { 247 // FIXME: Is this correct? 248 int WhitespaceLength = SourceMgr.getSpellingColumnNumber( 249 State.NextToken->WhitespaceRange.getEnd()) - 250 SourceMgr.getSpellingColumnNumber( 251 State.NextToken->WhitespaceRange.getBegin()); 252 State.Column += WhitespaceLength; 253 moveStateToNextToken(State, DryRun, /*Newline=*/false); 254 return 0; 255 } 256 257 unsigned Penalty = 0; 258 if (Newline) 259 Penalty = addTokenOnNewLine(State, DryRun); 260 else 261 addTokenOnCurrentLine(State, DryRun, ExtraSpaces); 262 263 return moveStateToNextToken(State, DryRun, Newline) + Penalty; 264 } 265 266 void ContinuationIndenter::addTokenOnCurrentLine(LineState &State, bool DryRun, 267 unsigned ExtraSpaces) { 268 FormatToken &Current = *State.NextToken; 269 const FormatToken &Previous = *State.NextToken->Previous; 270 if (Current.is(tok::equal) && 271 (State.Line->First->is(tok::kw_for) || Current.NestingLevel == 0) && 272 State.Stack.back().VariablePos == 0) { 273 State.Stack.back().VariablePos = State.Column; 274 // Move over * and & if they are bound to the variable name. 275 const FormatToken *Tok = &Previous; 276 while (Tok && State.Stack.back().VariablePos >= Tok->ColumnWidth) { 277 State.Stack.back().VariablePos -= Tok->ColumnWidth; 278 if (Tok->SpacesRequiredBefore != 0) 279 break; 280 Tok = Tok->Previous; 281 } 282 if (Previous.PartOfMultiVariableDeclStmt) 283 State.Stack.back().LastSpace = State.Stack.back().VariablePos; 284 } 285 286 unsigned Spaces = Current.SpacesRequiredBefore + ExtraSpaces; 287 288 if (!DryRun) 289 Whitespaces.replaceWhitespace(Current, /*Newlines=*/0, /*IndentLevel=*/0, 290 Spaces, State.Column + Spaces); 291 292 if (Current.is(TT_SelectorName) && 293 !State.Stack.back().ObjCSelectorNameFound) { 294 if (Current.LongestObjCSelectorName == 0) 295 State.Stack.back().AlignColons = false; 296 else if (State.Stack.back().Indent + Current.LongestObjCSelectorName > 297 State.Column + Spaces + Current.ColumnWidth) 298 State.Stack.back().ColonPos = 299 State.Stack.back().Indent + Current.LongestObjCSelectorName; 300 else 301 State.Stack.back().ColonPos = State.Column + Spaces + Current.ColumnWidth; 302 } 303 304 if (Style.AlignAfterOpenBracket && Previous.opensScope() && 305 Previous.isNot(TT_ObjCMethodExpr) && 306 (Current.isNot(TT_LineComment) || Previous.BlockKind == BK_BracedInit)) 307 State.Stack.back().Indent = State.Column + Spaces; 308 if (State.Stack.back().AvoidBinPacking && startsNextParameter(Current, Style)) 309 State.Stack.back().NoLineBreak = true; 310 if (startsSegmentOfBuilderTypeCall(Current)) 311 State.Stack.back().ContainsUnwrappedBuilder = true; 312 313 if (Current.isMemberAccess() && Previous.is(tok::r_paren) && 314 (Previous.MatchingParen && 315 (Previous.TotalLength - Previous.MatchingParen->TotalLength > 10))) { 316 // If there is a function call with long parameters, break before trailing 317 // calls. This prevents things like: 318 // EXPECT_CALL(SomeLongParameter).Times( 319 // 2); 320 // We don't want to do this for short parameters as they can just be 321 // indexes. 322 State.Stack.back().NoLineBreak = true; 323 } 324 325 State.Column += Spaces; 326 if (Current.isNot(tok::comment) && Previous.is(tok::l_paren) && 327 Previous.Previous && Previous.Previous->isOneOf(tok::kw_if, tok::kw_for)) 328 // Treat the condition inside an if as if it was a second function 329 // parameter, i.e. let nested calls have a continuation indent. 330 State.Stack.back().LastSpace = State.Column; 331 else if (!Current.isOneOf(tok::comment, tok::caret) && 332 (Previous.is(tok::comma) || 333 (Previous.is(tok::colon) && Previous.is(TT_ObjCMethodExpr)))) 334 State.Stack.back().LastSpace = State.Column; 335 else if ((Previous.isOneOf(TT_BinaryOperator, TT_ConditionalExpr, 336 TT_CtorInitializerColon)) && 337 ((Previous.getPrecedence() != prec::Assignment && 338 (Previous.isNot(tok::lessless) || Previous.OperatorIndex != 0 || 339 !Previous.LastOperator)) || 340 Current.StartsBinaryExpression)) 341 // Always indent relative to the RHS of the expression unless this is a 342 // simple assignment without binary expression on the RHS. Also indent 343 // relative to unary operators and the colons of constructor initializers. 344 State.Stack.back().LastSpace = State.Column; 345 else if (Previous.is(TT_InheritanceColon)) { 346 State.Stack.back().Indent = State.Column; 347 State.Stack.back().LastSpace = State.Column; 348 } else if (Previous.opensScope()) { 349 // If a function has a trailing call, indent all parameters from the 350 // opening parenthesis. This avoids confusing indents like: 351 // OuterFunction(InnerFunctionCall( // break 352 // ParameterToInnerFunction)) // break 353 // .SecondInnerFunctionCall(); 354 bool HasTrailingCall = false; 355 if (Previous.MatchingParen) { 356 const FormatToken *Next = Previous.MatchingParen->getNextNonComment(); 357 HasTrailingCall = Next && Next->isMemberAccess(); 358 } 359 if (HasTrailingCall && 360 State.Stack[State.Stack.size() - 2].CallContinuation == 0) 361 State.Stack.back().LastSpace = State.Column; 362 } 363 } 364 365 unsigned ContinuationIndenter::addTokenOnNewLine(LineState &State, 366 bool DryRun) { 367 FormatToken &Current = *State.NextToken; 368 const FormatToken &Previous = *State.NextToken->Previous; 369 370 // Extra penalty that needs to be added because of the way certain line 371 // breaks are chosen. 372 unsigned Penalty = 0; 373 374 const FormatToken *PreviousNonComment = Current.getPreviousNonComment(); 375 const FormatToken *NextNonComment = Previous.getNextNonComment(); 376 if (!NextNonComment) 377 NextNonComment = &Current; 378 // The first line break on any NestingLevel causes an extra penalty in order 379 // prefer similar line breaks. 380 if (!State.Stack.back().ContainsLineBreak) 381 Penalty += 15; 382 State.Stack.back().ContainsLineBreak = true; 383 384 Penalty += State.NextToken->SplitPenalty; 385 386 // Breaking before the first "<<" is generally not desirable if the LHS is 387 // short. Also always add the penalty if the LHS is split over mutliple lines 388 // to avoid unnecessary line breaks that just work around this penalty. 389 if (NextNonComment->is(tok::lessless) && 390 State.Stack.back().FirstLessLess == 0 && 391 (State.Column <= Style.ColumnLimit / 3 || 392 State.Stack.back().BreakBeforeParameter)) 393 Penalty += Style.PenaltyBreakFirstLessLess; 394 395 State.Column = getNewLineColumn(State); 396 if (NextNonComment->isMemberAccess()) { 397 if (State.Stack.back().CallContinuation == 0) 398 State.Stack.back().CallContinuation = State.Column; 399 } else if (NextNonComment->is(TT_SelectorName)) { 400 if (!State.Stack.back().ObjCSelectorNameFound) { 401 if (NextNonComment->LongestObjCSelectorName == 0) { 402 State.Stack.back().AlignColons = false; 403 } else { 404 State.Stack.back().ColonPos = 405 State.Stack.back().Indent + NextNonComment->LongestObjCSelectorName; 406 } 407 } else if (State.Stack.back().AlignColons && 408 State.Stack.back().ColonPos <= NextNonComment->ColumnWidth) { 409 State.Stack.back().ColonPos = State.Column + NextNonComment->ColumnWidth; 410 } 411 } else if (PreviousNonComment && PreviousNonComment->is(tok::colon) && 412 PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) { 413 // FIXME: This is hacky, find a better way. The problem is that in an ObjC 414 // method expression, the block should be aligned to the line starting it, 415 // e.g.: 416 // [aaaaaaaaaaaaaaa aaaaaaaaa: \\ break for some reason 417 // ^(int *i) { 418 // // ... 419 // }]; 420 // Thus, we set LastSpace of the next higher NestingLevel, to which we move 421 // when we consume all of the "}"'s FakeRParens at the "{". 422 if (State.Stack.size() > 1) 423 State.Stack[State.Stack.size() - 2].LastSpace = 424 std::max(State.Stack.back().LastSpace, State.Stack.back().Indent) + 425 Style.ContinuationIndentWidth; 426 } 427 428 if ((Previous.isOneOf(tok::comma, tok::semi) && 429 !State.Stack.back().AvoidBinPacking) || 430 Previous.is(TT_BinaryOperator)) 431 State.Stack.back().BreakBeforeParameter = false; 432 if (Previous.isOneOf(TT_TemplateCloser, TT_JavaAnnotation) && 433 Current.NestingLevel == 0) 434 State.Stack.back().BreakBeforeParameter = false; 435 if (NextNonComment->is(tok::question) || 436 (PreviousNonComment && PreviousNonComment->is(tok::question))) 437 State.Stack.back().BreakBeforeParameter = true; 438 439 if (!DryRun) { 440 unsigned Newlines = std::max( 441 1u, std::min(Current.NewlinesBefore, Style.MaxEmptyLinesToKeep + 1)); 442 Whitespaces.replaceWhitespace(Current, Newlines, 443 State.Stack.back().IndentLevel, State.Column, 444 State.Column, State.Line->InPPDirective); 445 } 446 447 if (!Current.isTrailingComment()) 448 State.Stack.back().LastSpace = State.Column; 449 State.StartOfLineLevel = Current.NestingLevel; 450 State.LowestLevelOnLine = Current.NestingLevel; 451 452 // Any break on this level means that the parent level has been broken 453 // and we need to avoid bin packing there. 454 bool NestedBlockSpecialCase = 455 Current.is(tok::r_brace) && State.Stack.size() > 1 && 456 State.Stack[State.Stack.size() - 2].NestedBlockInlined; 457 if (!NestedBlockSpecialCase) { 458 for (unsigned i = 0, e = State.Stack.size() - 1; i != e; ++i) { 459 State.Stack[i].BreakBeforeParameter = true; 460 } 461 } 462 463 if (PreviousNonComment && 464 !PreviousNonComment->isOneOf(tok::comma, tok::semi) && 465 (PreviousNonComment->isNot(TT_TemplateCloser) || 466 Current.NestingLevel != 0) && 467 !PreviousNonComment->isOneOf(TT_BinaryOperator, TT_JavaAnnotation, 468 TT_LeadingJavaAnnotation) && 469 Current.isNot(TT_BinaryOperator) && !PreviousNonComment->opensScope()) 470 State.Stack.back().BreakBeforeParameter = true; 471 472 // If we break after { or the [ of an array initializer, we should also break 473 // before the corresponding } or ]. 474 if (PreviousNonComment && 475 (PreviousNonComment->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare))) 476 State.Stack.back().BreakBeforeClosingBrace = true; 477 478 if (State.Stack.back().AvoidBinPacking) { 479 // If we are breaking after '(', '{', '<', this is not bin packing 480 // unless AllowAllParametersOfDeclarationOnNextLine is false or this is a 481 // dict/object literal. 482 if (!Previous.isOneOf(tok::l_paren, tok::l_brace, TT_BinaryOperator) || 483 (!Style.AllowAllParametersOfDeclarationOnNextLine && 484 State.Line->MustBeDeclaration) || 485 Previous.is(TT_DictLiteral)) 486 State.Stack.back().BreakBeforeParameter = true; 487 } 488 489 return Penalty; 490 } 491 492 unsigned ContinuationIndenter::getNewLineColumn(const LineState &State) { 493 if (!State.NextToken || !State.NextToken->Previous) 494 return 0; 495 FormatToken &Current = *State.NextToken; 496 const FormatToken &Previous = *Current.Previous; 497 // If we are continuing an expression, we want to use the continuation indent. 498 unsigned ContinuationIndent = 499 std::max(State.Stack.back().LastSpace, State.Stack.back().Indent) + 500 Style.ContinuationIndentWidth; 501 const FormatToken *PreviousNonComment = Current.getPreviousNonComment(); 502 const FormatToken *NextNonComment = Previous.getNextNonComment(); 503 if (!NextNonComment) 504 NextNonComment = &Current; 505 506 // Java specific bits. 507 if (Style.Language == FormatStyle::LK_Java && 508 Current.isOneOf(Keywords.kw_implements, Keywords.kw_extends)) 509 return std::max(State.Stack.back().LastSpace, 510 State.Stack.back().Indent + Style.ContinuationIndentWidth); 511 512 if (NextNonComment->is(tok::l_brace) && NextNonComment->BlockKind == BK_Block) 513 return Current.NestingLevel == 0 ? State.FirstIndent 514 : State.Stack.back().Indent; 515 if (Current.isOneOf(tok::r_brace, tok::r_square)) { 516 if (State.Stack.size() > 1 && 517 State.Stack[State.Stack.size() - 2].NestedBlockInlined) 518 return State.FirstIndent; 519 if (Current.closesBlockTypeList(Style) || 520 (Current.MatchingParen && 521 Current.MatchingParen->BlockKind == BK_BracedInit)) 522 return State.Stack[State.Stack.size() - 2].LastSpace; 523 else 524 return State.FirstIndent; 525 } 526 if (Current.is(tok::identifier) && Current.Next && 527 Current.Next->is(TT_DictLiteral)) 528 return State.Stack.back().Indent; 529 if (NextNonComment->isStringLiteral() && State.StartOfStringLiteral != 0) 530 return State.StartOfStringLiteral; 531 if (NextNonComment->is(tok::lessless) && 532 State.Stack.back().FirstLessLess != 0) 533 return State.Stack.back().FirstLessLess; 534 if (NextNonComment->isMemberAccess()) { 535 if (State.Stack.back().CallContinuation == 0) { 536 return ContinuationIndent; 537 } else { 538 return State.Stack.back().CallContinuation; 539 } 540 } 541 if (State.Stack.back().QuestionColumn != 0 && 542 ((NextNonComment->is(tok::colon) && 543 NextNonComment->is(TT_ConditionalExpr)) || 544 Previous.is(TT_ConditionalExpr))) 545 return State.Stack.back().QuestionColumn; 546 if (Previous.is(tok::comma) && State.Stack.back().VariablePos != 0) 547 return State.Stack.back().VariablePos; 548 if ((PreviousNonComment && 549 (PreviousNonComment->ClosesTemplateDeclaration || 550 PreviousNonComment->isOneOf(TT_AttributeParen, TT_JavaAnnotation, 551 TT_LeadingJavaAnnotation))) || 552 (!Style.IndentWrappedFunctionNames && 553 NextNonComment->isOneOf(tok::kw_operator, TT_FunctionDeclarationName))) 554 return std::max(State.Stack.back().LastSpace, State.Stack.back().Indent); 555 if (NextNonComment->is(TT_SelectorName)) { 556 if (!State.Stack.back().ObjCSelectorNameFound) { 557 if (NextNonComment->LongestObjCSelectorName == 0) { 558 return State.Stack.back().Indent; 559 } else { 560 return State.Stack.back().Indent + 561 NextNonComment->LongestObjCSelectorName - 562 NextNonComment->ColumnWidth; 563 } 564 } else if (!State.Stack.back().AlignColons) { 565 return State.Stack.back().Indent; 566 } else if (State.Stack.back().ColonPos > NextNonComment->ColumnWidth) { 567 return State.Stack.back().ColonPos - NextNonComment->ColumnWidth; 568 } else { 569 return State.Stack.back().Indent; 570 } 571 } 572 if (NextNonComment->is(TT_ArraySubscriptLSquare)) { 573 if (State.Stack.back().StartOfArraySubscripts != 0) 574 return State.Stack.back().StartOfArraySubscripts; 575 else 576 return ContinuationIndent; 577 } 578 if (NextNonComment->is(TT_StartOfName) || 579 Previous.isOneOf(tok::coloncolon, tok::equal)) { 580 return ContinuationIndent; 581 } 582 if (PreviousNonComment && PreviousNonComment->is(tok::colon) && 583 PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) 584 return ContinuationIndent; 585 if (NextNonComment->is(TT_CtorInitializerColon)) 586 return State.FirstIndent + Style.ConstructorInitializerIndentWidth; 587 if (NextNonComment->is(TT_CtorInitializerComma)) 588 return State.Stack.back().Indent; 589 if (Previous.is(tok::r_paren) && !Current.isBinaryOperator() && 590 !Current.isOneOf(tok::colon, tok::comment)) 591 return ContinuationIndent; 592 if (State.Stack.back().Indent == State.FirstIndent && PreviousNonComment && 593 PreviousNonComment->isNot(tok::r_brace)) 594 // Ensure that we fall back to the continuation indent width instead of 595 // just flushing continuations left. 596 return State.Stack.back().Indent + Style.ContinuationIndentWidth; 597 return State.Stack.back().Indent; 598 } 599 600 unsigned ContinuationIndenter::moveStateToNextToken(LineState &State, 601 bool DryRun, bool Newline) { 602 assert(State.Stack.size()); 603 const FormatToken &Current = *State.NextToken; 604 605 if (Current.is(TT_InheritanceColon)) 606 State.Stack.back().AvoidBinPacking = true; 607 if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator)) { 608 if (State.Stack.back().FirstLessLess == 0) 609 State.Stack.back().FirstLessLess = State.Column; 610 else 611 State.Stack.back().LastOperatorWrapped = Newline; 612 } 613 if ((Current.is(TT_BinaryOperator) && Current.isNot(tok::lessless)) || 614 Current.is(TT_ConditionalExpr)) 615 State.Stack.back().LastOperatorWrapped = Newline; 616 if (Current.is(TT_ArraySubscriptLSquare) && 617 State.Stack.back().StartOfArraySubscripts == 0) 618 State.Stack.back().StartOfArraySubscripts = State.Column; 619 if ((Current.is(tok::question) && Style.BreakBeforeTernaryOperators) || 620 (Current.getPreviousNonComment() && Current.isNot(tok::colon) && 621 Current.getPreviousNonComment()->is(tok::question) && 622 !Style.BreakBeforeTernaryOperators)) 623 State.Stack.back().QuestionColumn = State.Column; 624 if (!Current.opensScope() && !Current.closesScope()) 625 State.LowestLevelOnLine = 626 std::min(State.LowestLevelOnLine, Current.NestingLevel); 627 if (Current.isMemberAccess()) 628 State.Stack.back().StartOfFunctionCall = 629 Current.LastOperator ? 0 : State.Column + Current.ColumnWidth; 630 if (Current.is(TT_SelectorName)) 631 State.Stack.back().ObjCSelectorNameFound = true; 632 if (Current.is(TT_CtorInitializerColon)) { 633 // Indent 2 from the column, so: 634 // SomeClass::SomeClass() 635 // : First(...), ... 636 // Next(...) 637 // ^ line up here. 638 State.Stack.back().Indent = 639 State.Column + (Style.BreakConstructorInitializersBeforeComma ? 0 : 2); 640 if (Style.ConstructorInitializerAllOnOneLineOrOnePerLine) 641 State.Stack.back().AvoidBinPacking = true; 642 State.Stack.back().BreakBeforeParameter = false; 643 } 644 645 // In ObjC method declaration we align on the ":" of parameters, but we need 646 // to ensure that we indent parameters on subsequent lines by at least our 647 // continuation indent width. 648 if (Current.is(TT_ObjCMethodSpecifier)) 649 State.Stack.back().Indent += Style.ContinuationIndentWidth; 650 651 // Insert scopes created by fake parenthesis. 652 const FormatToken *Previous = Current.getPreviousNonComment(); 653 654 // Add special behavior to support a format commonly used for JavaScript 655 // closures: 656 // SomeFunction(function() { 657 // foo(); 658 // bar(); 659 // }, a, b, c); 660 if (Current.isNot(tok::comment) && Previous && Previous->is(tok::l_brace) && 661 State.Stack.size() > 1) { 662 if (State.Stack[State.Stack.size() - 2].NestedBlockInlined && Newline) { 663 for (unsigned i = 0, e = State.Stack.size() - 1; i != e; ++i) { 664 State.Stack[i].NoLineBreak = true; 665 } 666 } 667 State.Stack[State.Stack.size() - 2].NestedBlockInlined = false; 668 } 669 if (Previous && (Previous->isOneOf(tok::l_paren, tok::comma, tok::colon) || 670 Previous->isOneOf(TT_BinaryOperator, TT_ConditionalExpr)) && 671 !Previous->isOneOf(TT_DictLiteral, TT_ObjCMethodExpr)) { 672 State.Stack.back().NestedBlockInlined = 673 !Newline && 674 (Previous->isNot(tok::l_paren) || Previous->ParameterCount > 1) && !Newline; 675 } 676 677 moveStatePastFakeLParens(State, Newline); 678 moveStatePastScopeOpener(State, Newline); 679 moveStatePastScopeCloser(State); 680 moveStatePastFakeRParens(State); 681 682 if (Current.isStringLiteral() && State.StartOfStringLiteral == 0) { 683 State.StartOfStringLiteral = State.Column; 684 } else if (!Current.isOneOf(tok::comment, tok::identifier, tok::hash) && 685 !Current.isStringLiteral()) { 686 State.StartOfStringLiteral = 0; 687 } 688 689 State.Column += Current.ColumnWidth; 690 State.NextToken = State.NextToken->Next; 691 unsigned Penalty = breakProtrudingToken(Current, State, DryRun); 692 if (State.Column > getColumnLimit(State)) { 693 unsigned ExcessCharacters = State.Column - getColumnLimit(State); 694 Penalty += Style.PenaltyExcessCharacter * ExcessCharacters; 695 } 696 697 if (Current.Role) 698 Current.Role->formatFromToken(State, this, DryRun); 699 // If the previous has a special role, let it consume tokens as appropriate. 700 // It is necessary to start at the previous token for the only implemented 701 // role (comma separated list). That way, the decision whether or not to break 702 // after the "{" is already done and both options are tried and evaluated. 703 // FIXME: This is ugly, find a better way. 704 if (Previous && Previous->Role) 705 Penalty += Previous->Role->formatAfterToken(State, this, DryRun); 706 707 return Penalty; 708 } 709 710 void ContinuationIndenter::moveStatePastFakeLParens(LineState &State, 711 bool Newline) { 712 const FormatToken &Current = *State.NextToken; 713 const FormatToken *Previous = Current.getPreviousNonComment(); 714 715 // Don't add extra indentation for the first fake parenthesis after 716 // 'return', assignments or opening <({[. The indentation for these cases 717 // is special cased. 718 bool SkipFirstExtraIndent = 719 (Previous && (Previous->opensScope() || Previous->is(tok::kw_return) || 720 (Previous->getPrecedence() == prec::Assignment && 721 Style.AlignOperands) || 722 Previous->is(TT_ObjCMethodExpr))); 723 for (SmallVectorImpl<prec::Level>::const_reverse_iterator 724 I = Current.FakeLParens.rbegin(), 725 E = Current.FakeLParens.rend(); 726 I != E; ++I) { 727 ParenState NewParenState = State.Stack.back(); 728 NewParenState.ContainsLineBreak = false; 729 730 // Indent from 'LastSpace' unless these are fake parentheses encapsulating 731 // a builder type call after 'return' or, if the alignment after opening 732 // brackets is disabled. 733 if (!Current.isTrailingComment() && 734 (Style.AlignOperands || *I < prec::Assignment) && 735 (!Previous || Previous->isNot(tok::kw_return) || 736 (Style.Language != FormatStyle::LK_Java && *I > 0)) && 737 (Style.AlignAfterOpenBracket || *I != prec::Comma || 738 Current.NestingLevel == 0)) 739 NewParenState.Indent = 740 std::max(std::max(State.Column, NewParenState.Indent), 741 State.Stack.back().LastSpace); 742 743 // Don't allow the RHS of an operator to be split over multiple lines unless 744 // there is a line-break right after the operator. 745 // Exclude relational operators, as there, it is always more desirable to 746 // have the LHS 'left' of the RHS. 747 if (Previous && Previous->getPrecedence() > prec::Assignment && 748 Previous->isOneOf(TT_BinaryOperator, TT_ConditionalExpr) && 749 Previous->getPrecedence() != prec::Relational) { 750 bool BreakBeforeOperator = 751 Previous->is(tok::lessless) || 752 (Previous->is(TT_BinaryOperator) && 753 Style.BreakBeforeBinaryOperators != FormatStyle::BOS_None) || 754 (Previous->is(TT_ConditionalExpr) && 755 Style.BreakBeforeTernaryOperators); 756 if ((!Newline && !BreakBeforeOperator) || 757 (!State.Stack.back().LastOperatorWrapped && BreakBeforeOperator)) 758 NewParenState.NoLineBreak = true; 759 } 760 761 // Do not indent relative to the fake parentheses inserted for "." or "->". 762 // This is a special case to make the following to statements consistent: 763 // OuterFunction(InnerFunctionCall( // break 764 // ParameterToInnerFunction)); 765 // OuterFunction(SomeObject.InnerFunctionCall( // break 766 // ParameterToInnerFunction)); 767 if (*I > prec::Unknown) 768 NewParenState.LastSpace = std::max(NewParenState.LastSpace, State.Column); 769 if (*I != prec::Conditional) 770 NewParenState.StartOfFunctionCall = State.Column; 771 772 // Always indent conditional expressions. Never indent expression where 773 // the 'operator' is ',', ';' or an assignment (i.e. *I <= 774 // prec::Assignment) as those have different indentation rules. Indent 775 // other expression, unless the indentation needs to be skipped. 776 if (*I == prec::Conditional || 777 (!SkipFirstExtraIndent && *I > prec::Assignment && 778 !Current.isTrailingComment())) 779 NewParenState.Indent += Style.ContinuationIndentWidth; 780 if ((Previous && !Previous->opensScope()) || *I > prec::Comma) 781 NewParenState.BreakBeforeParameter = false; 782 State.Stack.push_back(NewParenState); 783 SkipFirstExtraIndent = false; 784 } 785 } 786 787 // Remove the fake r_parens after 'Tok'. 788 static void consumeRParens(LineState& State, const FormatToken &Tok) { 789 for (unsigned i = 0, e = Tok.FakeRParens; i != e; ++i) { 790 unsigned VariablePos = State.Stack.back().VariablePos; 791 assert(State.Stack.size() > 1); 792 if (State.Stack.size() == 1) { 793 // Do not pop the last element. 794 break; 795 } 796 State.Stack.pop_back(); 797 State.Stack.back().VariablePos = VariablePos; 798 } 799 } 800 801 // Returns whether 'Tok' opens or closes a scope requiring special handling 802 // of the subsequent fake r_parens. 803 // 804 // For example, if this is an l_brace starting a nested block, we pretend (wrt. 805 // to indentation) that we already consumed the corresponding r_brace. Thus, we 806 // remove all ParenStates caused by fake parentheses that end at the r_brace. 807 // The net effect of this is that we don't indent relative to the l_brace, if 808 // the nested block is the last parameter of a function. This formats: 809 // 810 // SomeFunction(a, [] { 811 // f(); // break 812 // }); 813 // 814 // instead of: 815 // SomeFunction(a, [] { 816 // f(); // break 817 // }); 818 static bool fakeRParenSpecialCase(const LineState &State) { 819 const FormatToken &Tok = *State.NextToken; 820 if (!Tok.MatchingParen) 821 return false; 822 const FormatToken *Left = &Tok; 823 if (Tok.isOneOf(tok::r_brace, tok::r_square)) 824 Left = Tok.MatchingParen; 825 return !State.Stack.back().HasMultipleNestedBlocks && 826 Left->isOneOf(tok::l_brace, tok::l_square) && 827 (Left->BlockKind == BK_Block || 828 Left->isOneOf(TT_ArrayInitializerLSquare, TT_DictLiteral)); 829 } 830 831 void ContinuationIndenter::moveStatePastFakeRParens(LineState &State) { 832 // Don't remove FakeRParens attached to r_braces that surround nested blocks 833 // as they will have been removed early (see above). 834 if (fakeRParenSpecialCase(State)) 835 return; 836 837 consumeRParens(State, *State.NextToken); 838 } 839 840 void ContinuationIndenter::moveStatePastScopeOpener(LineState &State, 841 bool Newline) { 842 const FormatToken &Current = *State.NextToken; 843 if (!Current.opensScope()) 844 return; 845 846 if (Current.MatchingParen && Current.BlockKind == BK_Block) { 847 moveStateToNewBlock(State); 848 return; 849 } 850 851 unsigned NewIndent; 852 unsigned NewIndentLevel = State.Stack.back().IndentLevel; 853 bool AvoidBinPacking; 854 bool BreakBeforeParameter = false; 855 if (Current.isOneOf(tok::l_brace, TT_ArrayInitializerLSquare)) { 856 if (fakeRParenSpecialCase(State)) 857 consumeRParens(State, *Current.MatchingParen); 858 859 NewIndent = State.Stack.back().LastSpace; 860 if (Current.opensBlockTypeList(Style)) { 861 NewIndent += Style.IndentWidth; 862 NewIndent = std::min(State.Column + 2, NewIndent); 863 ++NewIndentLevel; 864 } else { 865 NewIndent += Style.ContinuationIndentWidth; 866 NewIndent = std::min(State.Column + 1, NewIndent); 867 } 868 const FormatToken *NextNoComment = Current.getNextNonComment(); 869 AvoidBinPacking = 870 Current.isOneOf(TT_ArrayInitializerLSquare, TT_DictLiteral) || 871 Style.Language == FormatStyle::LK_Proto || !Style.BinPackParameters || 872 (NextNoComment && NextNoComment->is(TT_DesignatedInitializerPeriod)); 873 } else { 874 NewIndent = Style.ContinuationIndentWidth + 875 std::max(State.Stack.back().LastSpace, 876 State.Stack.back().StartOfFunctionCall); 877 AvoidBinPacking = 878 (State.Line->MustBeDeclaration && !Style.BinPackParameters) || 879 (!State.Line->MustBeDeclaration && !Style.BinPackArguments) || 880 (Style.ExperimentalAutoDetectBinPacking && 881 (Current.PackingKind == PPK_OnePerLine || 882 (!BinPackInconclusiveFunctions && 883 Current.PackingKind == PPK_Inconclusive))); 884 // If this '[' opens an ObjC call, determine whether all parameters fit 885 // into one line and put one per line if they don't. 886 if (Current.is(TT_ObjCMethodExpr) && Style.ColumnLimit != 0 && 887 getLengthToMatchingParen(Current) + State.Column > 888 getColumnLimit(State)) 889 BreakBeforeParameter = true; 890 } 891 bool NoLineBreak = State.Stack.back().NoLineBreak || 892 (Current.is(TT_TemplateOpener) && 893 State.Stack.back().ContainsUnwrappedBuilder); 894 State.Stack.push_back(ParenState(NewIndent, NewIndentLevel, 895 State.Stack.back().LastSpace, 896 AvoidBinPacking, NoLineBreak)); 897 State.Stack.back().BreakBeforeParameter = BreakBeforeParameter; 898 State.Stack.back().HasMultipleNestedBlocks = Current.BlockParameterCount > 1; 899 } 900 901 void ContinuationIndenter::moveStatePastScopeCloser(LineState &State) { 902 const FormatToken &Current = *State.NextToken; 903 if (!Current.closesScope()) 904 return; 905 906 // If we encounter a closing ), ], } or >, we can remove a level from our 907 // stacks. 908 if (State.Stack.size() > 1 && 909 (Current.isOneOf(tok::r_paren, tok::r_square) || 910 (Current.is(tok::r_brace) && State.NextToken != State.Line->First) || 911 State.NextToken->is(TT_TemplateCloser))) 912 State.Stack.pop_back(); 913 914 if (Current.is(tok::r_square)) { 915 // If this ends the array subscript expr, reset the corresponding value. 916 const FormatToken *NextNonComment = Current.getNextNonComment(); 917 if (NextNonComment && NextNonComment->isNot(tok::l_square)) 918 State.Stack.back().StartOfArraySubscripts = 0; 919 } 920 } 921 922 void ContinuationIndenter::moveStateToNewBlock(LineState &State) { 923 // If we have already found more than one lambda introducers on this level, we 924 // opt out of this because similarity between the lambdas is more important. 925 if (fakeRParenSpecialCase(State)) 926 consumeRParens(State, *State.NextToken->MatchingParen); 927 928 // ObjC block sometimes follow special indentation rules. 929 unsigned NewIndent = 930 State.Stack.back().LastSpace + (State.NextToken->is(TT_ObjCBlockLBrace) 931 ? Style.ObjCBlockIndentWidth 932 : Style.IndentWidth); 933 State.Stack.push_back(ParenState( 934 NewIndent, /*NewIndentLevel=*/State.Stack.back().IndentLevel + 1, 935 State.Stack.back().LastSpace, /*AvoidBinPacking=*/true, 936 State.Stack.back().NoLineBreak)); 937 State.Stack.back().BreakBeforeParameter = true; 938 } 939 940 unsigned ContinuationIndenter::addMultilineToken(const FormatToken &Current, 941 LineState &State) { 942 // Break before further function parameters on all levels. 943 for (unsigned i = 0, e = State.Stack.size(); i != e; ++i) 944 State.Stack[i].BreakBeforeParameter = true; 945 946 unsigned ColumnsUsed = State.Column; 947 // We can only affect layout of the first and the last line, so the penalty 948 // for all other lines is constant, and we ignore it. 949 State.Column = Current.LastLineColumnWidth; 950 951 if (ColumnsUsed > getColumnLimit(State)) 952 return Style.PenaltyExcessCharacter * (ColumnsUsed - getColumnLimit(State)); 953 return 0; 954 } 955 956 static bool getRawStringLiteralPrefixPostfix(StringRef Text, StringRef &Prefix, 957 StringRef &Postfix) { 958 if (Text.startswith(Prefix = "R\"") || Text.startswith(Prefix = "uR\"") || 959 Text.startswith(Prefix = "UR\"") || Text.startswith(Prefix = "u8R\"") || 960 Text.startswith(Prefix = "LR\"")) { 961 size_t ParenPos = Text.find('('); 962 if (ParenPos != StringRef::npos) { 963 StringRef Delimiter = 964 Text.substr(Prefix.size(), ParenPos - Prefix.size()); 965 Prefix = Text.substr(0, ParenPos + 1); 966 Postfix = Text.substr(Text.size() - 2 - Delimiter.size()); 967 return Postfix.front() == ')' && Postfix.back() == '"' && 968 Postfix.substr(1).startswith(Delimiter); 969 } 970 } 971 return false; 972 } 973 974 unsigned ContinuationIndenter::breakProtrudingToken(const FormatToken &Current, 975 LineState &State, 976 bool DryRun) { 977 // Don't break multi-line tokens other than block comments. Instead, just 978 // update the state. 979 if (Current.isNot(TT_BlockComment) && Current.IsMultiline) 980 return addMultilineToken(Current, State); 981 982 // Don't break implicit string literals or import statements. 983 if (Current.is(TT_ImplicitStringLiteral) || 984 State.Line->Type == LT_ImportStatement) 985 return 0; 986 987 if (!Current.isStringLiteral() && !Current.is(tok::comment)) 988 return 0; 989 990 std::unique_ptr<BreakableToken> Token; 991 unsigned StartColumn = State.Column - Current.ColumnWidth; 992 unsigned ColumnLimit = getColumnLimit(State); 993 994 if (Current.isStringLiteral()) { 995 // Don't break string literals inside preprocessor directives (except for 996 // #define directives, as their contents are stored in separate lines and 997 // are not affected by this check). 998 // This way we avoid breaking code with line directives and unknown 999 // preprocessor directives that contain long string literals. 1000 if (State.Line->Type == LT_PreprocessorDirective) 1001 return 0; 1002 // Exempts unterminated string literals from line breaking. The user will 1003 // likely want to terminate the string before any line breaking is done. 1004 if (Current.IsUnterminatedLiteral) 1005 return 0; 1006 1007 StringRef Text = Current.TokenText; 1008 StringRef Prefix; 1009 StringRef Postfix; 1010 bool IsNSStringLiteral = false; 1011 // FIXME: Handle whitespace between '_T', '(', '"..."', and ')'. 1012 // FIXME: Store Prefix and Suffix (or PrefixLength and SuffixLength to 1013 // reduce the overhead) for each FormatToken, which is a string, so that we 1014 // don't run multiple checks here on the hot path. 1015 if (Text.startswith("\"") && Current.Previous && 1016 Current.Previous->is(tok::at)) { 1017 IsNSStringLiteral = true; 1018 Prefix = "@\""; 1019 } 1020 if ((Text.endswith(Postfix = "\"") && 1021 (IsNSStringLiteral || Text.startswith(Prefix = "\"") || 1022 Text.startswith(Prefix = "u\"") || Text.startswith(Prefix = "U\"") || 1023 Text.startswith(Prefix = "u8\"") || 1024 Text.startswith(Prefix = "L\""))) || 1025 (Text.startswith(Prefix = "_T(\"") && Text.endswith(Postfix = "\")")) || 1026 getRawStringLiteralPrefixPostfix(Text, Prefix, Postfix)) { 1027 Token.reset(new BreakableStringLiteral( 1028 Current, State.Line->Level, StartColumn, Prefix, Postfix, 1029 State.Line->InPPDirective, Encoding, Style)); 1030 } else { 1031 return 0; 1032 } 1033 } else if (Current.is(TT_BlockComment) && Current.isTrailingComment()) { 1034 if (CommentPragmasRegex.match(Current.TokenText.substr(2))) 1035 return 0; 1036 Token.reset(new BreakableBlockComment( 1037 Current, State.Line->Level, StartColumn, Current.OriginalColumn, 1038 !Current.Previous, State.Line->InPPDirective, Encoding, Style)); 1039 } else if (Current.is(TT_LineComment) && 1040 (Current.Previous == nullptr || 1041 Current.Previous->isNot(TT_ImplicitStringLiteral))) { 1042 if (CommentPragmasRegex.match(Current.TokenText.substr(2))) 1043 return 0; 1044 Token.reset(new BreakableLineComment(Current, State.Line->Level, 1045 StartColumn, /*InPPDirective=*/false, 1046 Encoding, Style)); 1047 // We don't insert backslashes when breaking line comments. 1048 ColumnLimit = Style.ColumnLimit; 1049 } else { 1050 return 0; 1051 } 1052 if (Current.UnbreakableTailLength >= ColumnLimit) 1053 return 0; 1054 1055 unsigned RemainingSpace = ColumnLimit - Current.UnbreakableTailLength; 1056 bool BreakInserted = false; 1057 unsigned Penalty = 0; 1058 unsigned RemainingTokenColumns = 0; 1059 for (unsigned LineIndex = 0, EndIndex = Token->getLineCount(); 1060 LineIndex != EndIndex; ++LineIndex) { 1061 if (!DryRun) 1062 Token->replaceWhitespaceBefore(LineIndex, Whitespaces); 1063 unsigned TailOffset = 0; 1064 RemainingTokenColumns = 1065 Token->getLineLengthAfterSplit(LineIndex, TailOffset, StringRef::npos); 1066 while (RemainingTokenColumns > RemainingSpace) { 1067 BreakableToken::Split Split = 1068 Token->getSplit(LineIndex, TailOffset, ColumnLimit); 1069 if (Split.first == StringRef::npos) { 1070 // The last line's penalty is handled in addNextStateToQueue(). 1071 if (LineIndex < EndIndex - 1) 1072 Penalty += Style.PenaltyExcessCharacter * 1073 (RemainingTokenColumns - RemainingSpace); 1074 break; 1075 } 1076 assert(Split.first != 0); 1077 unsigned NewRemainingTokenColumns = Token->getLineLengthAfterSplit( 1078 LineIndex, TailOffset + Split.first + Split.second, StringRef::npos); 1079 1080 // We can remove extra whitespace instead of breaking the line. 1081 if (RemainingTokenColumns + 1 - Split.second <= RemainingSpace) { 1082 RemainingTokenColumns = 0; 1083 if (!DryRun) 1084 Token->replaceWhitespace(LineIndex, TailOffset, Split, Whitespaces); 1085 break; 1086 } 1087 1088 // When breaking before a tab character, it may be moved by a few columns, 1089 // but will still be expanded to the next tab stop, so we don't save any 1090 // columns. 1091 if (NewRemainingTokenColumns == RemainingTokenColumns) 1092 break; 1093 1094 assert(NewRemainingTokenColumns < RemainingTokenColumns); 1095 if (!DryRun) 1096 Token->insertBreak(LineIndex, TailOffset, Split, Whitespaces); 1097 Penalty += Current.SplitPenalty; 1098 unsigned ColumnsUsed = 1099 Token->getLineLengthAfterSplit(LineIndex, TailOffset, Split.first); 1100 if (ColumnsUsed > ColumnLimit) { 1101 Penalty += Style.PenaltyExcessCharacter * (ColumnsUsed - ColumnLimit); 1102 } 1103 TailOffset += Split.first + Split.second; 1104 RemainingTokenColumns = NewRemainingTokenColumns; 1105 BreakInserted = true; 1106 } 1107 } 1108 1109 State.Column = RemainingTokenColumns; 1110 1111 if (BreakInserted) { 1112 // If we break the token inside a parameter list, we need to break before 1113 // the next parameter on all levels, so that the next parameter is clearly 1114 // visible. Line comments already introduce a break. 1115 if (Current.isNot(TT_LineComment)) { 1116 for (unsigned i = 0, e = State.Stack.size(); i != e; ++i) 1117 State.Stack[i].BreakBeforeParameter = true; 1118 } 1119 1120 Penalty += Current.isStringLiteral() ? Style.PenaltyBreakString 1121 : Style.PenaltyBreakComment; 1122 1123 State.Stack.back().LastSpace = StartColumn; 1124 } 1125 return Penalty; 1126 } 1127 1128 unsigned ContinuationIndenter::getColumnLimit(const LineState &State) const { 1129 // In preprocessor directives reserve two chars for trailing " \" 1130 return Style.ColumnLimit - (State.Line->InPPDirective ? 2 : 0); 1131 } 1132 1133 bool ContinuationIndenter::nextIsMultilineString(const LineState &State) { 1134 const FormatToken &Current = *State.NextToken; 1135 if (!Current.isStringLiteral() || Current.is(TT_ImplicitStringLiteral)) 1136 return false; 1137 // We never consider raw string literals "multiline" for the purpose of 1138 // AlwaysBreakBeforeMultilineStrings implementation as they are special-cased 1139 // (see TokenAnnotator::mustBreakBefore(). 1140 if (Current.TokenText.startswith("R\"")) 1141 return false; 1142 if (Current.IsMultiline) 1143 return true; 1144 if (Current.getNextNonComment() && 1145 Current.getNextNonComment()->isStringLiteral()) 1146 return true; // Implicit concatenation. 1147 if (State.Column + Current.ColumnWidth + Current.UnbreakableTailLength > 1148 Style.ColumnLimit) 1149 return true; // String will be split. 1150 return false; 1151 } 1152 1153 } // namespace format 1154 } // namespace clang 1155