1de0328aeSDaniel Jasper //===--- ContinuationIndenter.cpp - Format C++ code -----------------------===// 2de0328aeSDaniel Jasper // 32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information. 52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6de0328aeSDaniel Jasper // 7de0328aeSDaniel Jasper //===----------------------------------------------------------------------===// 8de0328aeSDaniel Jasper /// 9de0328aeSDaniel Jasper /// \file 109fc8faf9SAdrian Prantl /// This file implements the continuation indenter. 11de0328aeSDaniel Jasper /// 12de0328aeSDaniel Jasper //===----------------------------------------------------------------------===// 13de0328aeSDaniel Jasper 14de0328aeSDaniel Jasper #include "ContinuationIndenter.h" 1589628f64SManuel Klimek #include "BreakableToken.h" 169ad83fe7SKrasimir Georgiev #include "FormatInternal.h" 17f3671a68SKrasimir Georgiev #include "FormatToken.h" 18de0328aeSDaniel Jasper #include "WhitespaceManager.h" 19de0328aeSDaniel Jasper #include "clang/Basic/OperatorPrecedence.h" 20de0328aeSDaniel Jasper #include "clang/Basic/SourceManager.h" 21de0328aeSDaniel Jasper #include "clang/Format/Format.h" 22befb9dc3SKrasimir Georgiev #include "llvm/ADT/StringSet.h" 23de0328aeSDaniel Jasper #include "llvm/Support/Debug.h" 24de0328aeSDaniel Jasper 2591834227SKrasimir Georgiev #define DEBUG_TYPE "format-indenter" 2610346667SChandler Carruth 27de0328aeSDaniel Jasper namespace clang { 28de0328aeSDaniel Jasper namespace format { 29de0328aeSDaniel Jasper 30416348efSBen Hamilton // Returns true if a TT_SelectorName should be indented when wrapped, 31416348efSBen Hamilton // false otherwise. 32416348efSBen Hamilton static bool shouldIndentWrappedSelectorName(const FormatStyle &Style, 33416348efSBen Hamilton LineType LineType) { 34416348efSBen Hamilton return Style.IndentWrappedFunctionNames || LineType == LT_ObjCMethodDecl; 35416348efSBen Hamilton } 36416348efSBen Hamilton 37de0328aeSDaniel Jasper // Returns the length of everything up to the first possible line break after 38de0328aeSDaniel Jasper // the ), ], } or > matching \c Tok. 398b4bc76aSKrasimir Georgiev static unsigned getLengthToMatchingParen(const FormatToken &Tok, 408b4bc76aSKrasimir Georgiev const std::vector<ParenState> &Stack) { 418b4bc76aSKrasimir Georgiev // Normally whether or not a break before T is possible is calculated and 428b4bc76aSKrasimir Georgiev // stored in T.CanBreakBefore. Braces, array initializers and text proto 438b4bc76aSKrasimir Georgiev // messages like `key: < ... >` are an exception: a break is possible 448b4bc76aSKrasimir Georgiev // before a closing brace R if a break was inserted after the corresponding 458b4bc76aSKrasimir Georgiev // opening brace. The information about whether or not a break is needed 468b4bc76aSKrasimir Georgiev // before a closing brace R is stored in the ParenState field 478b4bc76aSKrasimir Georgiev // S.BreakBeforeClosingBrace where S is the state that R closes. 488b4bc76aSKrasimir Georgiev // 498b4bc76aSKrasimir Georgiev // In order to decide whether there can be a break before encountered right 508b4bc76aSKrasimir Georgiev // braces, this implementation iterates over the sequence of tokens and over 518b4bc76aSKrasimir Georgiev // the paren stack in lockstep, keeping track of the stack level which visited 528b4bc76aSKrasimir Georgiev // right braces correspond to in MatchingStackIndex. 538b4bc76aSKrasimir Georgiev // 548b4bc76aSKrasimir Georgiev // For example, consider: 558b4bc76aSKrasimir Georgiev // L. <- line number 568b4bc76aSKrasimir Georgiev // 1. { 578b4bc76aSKrasimir Georgiev // 2. {1}, 588b4bc76aSKrasimir Georgiev // 3. {2}, 598b4bc76aSKrasimir Georgiev // 4. {{3}}} 608b4bc76aSKrasimir Georgiev // ^ where we call this method with this token. 618b4bc76aSKrasimir Georgiev // The paren stack at this point contains 3 brace levels: 628b4bc76aSKrasimir Georgiev // 0. { at line 1, BreakBeforeClosingBrace: true 638b4bc76aSKrasimir Georgiev // 1. first { at line 4, BreakBeforeClosingBrace: false 648b4bc76aSKrasimir Georgiev // 2. second { at line 4, BreakBeforeClosingBrace: false, 658b4bc76aSKrasimir Georgiev // where there might be fake parens levels in-between these levels. 668b4bc76aSKrasimir Georgiev // The algorithm will start at the first } on line 4, which is the matching 678b4bc76aSKrasimir Georgiev // brace of the initial left brace and at level 2 of the stack. Then, 688b4bc76aSKrasimir Georgiev // examining BreakBeforeClosingBrace: false at level 2, it will continue to 698b4bc76aSKrasimir Georgiev // the second } on line 4, and will traverse the stack downwards until it 708b4bc76aSKrasimir Georgiev // finds the matching { on level 1. Then, examining BreakBeforeClosingBrace: 718b4bc76aSKrasimir Georgiev // false at level 1, it will continue to the third } on line 4 and will 728b4bc76aSKrasimir Georgiev // traverse the stack downwards until it finds the matching { on level 0. 738b4bc76aSKrasimir Georgiev // Then, examining BreakBeforeClosingBrace: true at level 0, the algorithm 748b4bc76aSKrasimir Georgiev // will stop and will use the second } on line 4 to determine the length to 758b4bc76aSKrasimir Georgiev // return, as in this example the range will include the tokens: {3}} 768b4bc76aSKrasimir Georgiev // 778b4bc76aSKrasimir Georgiev // The algorithm will only traverse the stack if it encounters braces, array 788b4bc76aSKrasimir Georgiev // initializer squares or text proto angle brackets. 792145bc02SCraig Topper if (!Tok.MatchingParen) 80de0328aeSDaniel Jasper return 0; 81de0328aeSDaniel Jasper FormatToken *End = Tok.MatchingParen; 828b4bc76aSKrasimir Georgiev // Maintains a stack level corresponding to the current End token. 838b4bc76aSKrasimir Georgiev int MatchingStackIndex = Stack.size() - 1; 848b4bc76aSKrasimir Georgiev // Traverses the stack downwards, looking for the level to which LBrace 858b4bc76aSKrasimir Georgiev // corresponds. Returns either a pointer to the matching level or nullptr if 868b4bc76aSKrasimir Georgiev // LParen is not found in the initial portion of the stack up to 878b4bc76aSKrasimir Georgiev // MatchingStackIndex. 888b4bc76aSKrasimir Georgiev auto FindParenState = [&](const FormatToken *LBrace) -> const ParenState * { 898b4bc76aSKrasimir Georgiev while (MatchingStackIndex >= 0 && Stack[MatchingStackIndex].Tok != LBrace) 908b4bc76aSKrasimir Georgiev --MatchingStackIndex; 918b4bc76aSKrasimir Georgiev return MatchingStackIndex >= 0 ? &Stack[MatchingStackIndex] : nullptr; 928b4bc76aSKrasimir Georgiev }; 938b4bc76aSKrasimir Georgiev for (; End->Next; End = End->Next) { 94964293e4SKrasimir Georgiev if (End->Next->CanBreakBefore) 958b4bc76aSKrasimir Georgiev break; 96964293e4SKrasimir Georgiev if (!End->Next->closesScope()) 97964293e4SKrasimir Georgiev continue; 98f163bdc0SKrasimir Georgiev if (End->Next->MatchingParen && 99f163bdc0SKrasimir Georgiev End->Next->MatchingParen->isOneOf( 100f163bdc0SKrasimir Georgiev tok::l_brace, TT_ArrayInitializerLSquare, tok::less)) { 1018b4bc76aSKrasimir Georgiev const ParenState *State = FindParenState(End->Next->MatchingParen); 1028b4bc76aSKrasimir Georgiev if (State && State->BreakBeforeClosingBrace) 1038b4bc76aSKrasimir Georgiev break; 1048b4bc76aSKrasimir Georgiev } 105de0328aeSDaniel Jasper } 106de0328aeSDaniel Jasper return End->TotalLength - Tok.TotalLength + 1; 107de0328aeSDaniel Jasper } 108de0328aeSDaniel Jasper 109801cdb27SDaniel Jasper static unsigned getLengthToNextOperator(const FormatToken &Tok) { 110801cdb27SDaniel Jasper if (!Tok.NextOperator) 111801cdb27SDaniel Jasper return 0; 112801cdb27SDaniel Jasper return Tok.NextOperator->TotalLength - Tok.TotalLength; 113801cdb27SDaniel Jasper } 114801cdb27SDaniel Jasper 1154c6e0059SDaniel Jasper // Returns \c true if \c Tok is the "." or "->" of a call and starts the next 1164c6e0059SDaniel Jasper // segment of a builder type call. 1174c6e0059SDaniel Jasper static bool startsSegmentOfBuilderTypeCall(const FormatToken &Tok) { 1184c6e0059SDaniel Jasper return Tok.isMemberAccess() && Tok.Previous && Tok.Previous->closesScope(); 1194c6e0059SDaniel Jasper } 1204c6e0059SDaniel Jasper 121ec01cd6eSDaniel Jasper // Returns \c true if \c Current starts a new parameter. 122ec01cd6eSDaniel Jasper static bool startsNextParameter(const FormatToken &Current, 123ec01cd6eSDaniel Jasper const FormatStyle &Style) { 124ec01cd6eSDaniel Jasper const FormatToken &Previous = *Current.Previous; 125a98b7b01SDaniel Jasper if (Current.is(TT_CtorInitializerComma) && 126a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma) 127ec01cd6eSDaniel Jasper return true; 128ff747be4SKrasimir Georgiev if (Style.Language == FormatStyle::LK_Proto && Current.is(TT_SelectorName)) 129ff747be4SKrasimir Georgiev return true; 130ec01cd6eSDaniel Jasper return Previous.is(tok::comma) && !Current.isTrailingComment() && 1310ef8ee19SAndi-Bogdan Postelnicu ((Previous.isNot(TT_CtorInitializerComma) || 132a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers != 133a6b6d51bSFrancois Ferrand FormatStyle::BCIS_BeforeComma) && 1340ef8ee19SAndi-Bogdan Postelnicu (Previous.isNot(TT_InheritanceComma) || 1356bb103f9SFrancois Ferrand Style.BreakInheritanceList != FormatStyle::BILS_BeforeComma)); 136ec01cd6eSDaniel Jasper } 137ec01cd6eSDaniel Jasper 13826b144ccSKrasimir Georgiev static bool opensProtoMessageField(const FormatToken &LessTok, 13926b144ccSKrasimir Georgiev const FormatStyle &Style) { 14026b144ccSKrasimir Georgiev if (LessTok.isNot(tok::less)) 14126b144ccSKrasimir Georgiev return false; 14226b144ccSKrasimir Georgiev return Style.Language == FormatStyle::LK_TextProto || 14326b144ccSKrasimir Georgiev (Style.Language == FormatStyle::LK_Proto && 14426b144ccSKrasimir Georgiev (LessTok.NestingLevel > 0 || 14526b144ccSKrasimir Georgiev (LessTok.Previous && LessTok.Previous->is(tok::equal)))); 14626b144ccSKrasimir Georgiev } 14726b144ccSKrasimir Georgiev 1489ad83fe7SKrasimir Georgiev // Returns the delimiter of a raw string literal, or None if TokenText is not 1499ad83fe7SKrasimir Georgiev // the text of a raw string literal. The delimiter could be the empty string. 1509ad83fe7SKrasimir Georgiev // For example, the delimiter of R"deli(cont)deli" is deli. 1519ad83fe7SKrasimir Georgiev static llvm::Optional<StringRef> getRawStringDelimiter(StringRef TokenText) { 1529ad83fe7SKrasimir Georgiev if (TokenText.size() < 5 // The smallest raw string possible is 'R"()"'. 1539ad83fe7SKrasimir Georgiev || !TokenText.startswith("R\"") || !TokenText.endswith("\"")) 1549ad83fe7SKrasimir Georgiev return None; 1559ad83fe7SKrasimir Georgiev 1569ad83fe7SKrasimir Georgiev // A raw string starts with 'R"<delimiter>(' and delimiter is ascii and has 1579ad83fe7SKrasimir Georgiev // size at most 16 by the standard, so the first '(' must be among the first 1589ad83fe7SKrasimir Georgiev // 19 bytes. 1599ad83fe7SKrasimir Georgiev size_t LParenPos = TokenText.substr(0, 19).find_first_of('('); 1609ad83fe7SKrasimir Georgiev if (LParenPos == StringRef::npos) 1619ad83fe7SKrasimir Georgiev return None; 1629ad83fe7SKrasimir Georgiev StringRef Delimiter = TokenText.substr(2, LParenPos - 2); 1639ad83fe7SKrasimir Georgiev 1649ad83fe7SKrasimir Georgiev // Check that the string ends in ')Delimiter"'. 1659ad83fe7SKrasimir Georgiev size_t RParenPos = TokenText.size() - Delimiter.size() - 2; 1669ad83fe7SKrasimir Georgiev if (TokenText[RParenPos] != ')') 1679ad83fe7SKrasimir Georgiev return None; 1689ad83fe7SKrasimir Georgiev if (!TokenText.substr(RParenPos + 1).startswith(Delimiter)) 1699ad83fe7SKrasimir Georgiev return None; 1709ad83fe7SKrasimir Georgiev return Delimiter; 1719ad83fe7SKrasimir Georgiev } 1729ad83fe7SKrasimir Georgiev 173412ed095SKrasimir Georgiev // Returns the canonical delimiter for \p Language, or the empty string if no 174412ed095SKrasimir Georgiev // canonical delimiter is specified. 175412ed095SKrasimir Georgiev static StringRef 176412ed095SKrasimir Georgiev getCanonicalRawStringDelimiter(const FormatStyle &Style, 177412ed095SKrasimir Georgiev FormatStyle::LanguageKind Language) { 178412ed095SKrasimir Georgiev for (const auto &Format : Style.RawStringFormats) { 179412ed095SKrasimir Georgiev if (Format.Language == Language) 180412ed095SKrasimir Georgiev return StringRef(Format.CanonicalDelimiter); 181412ed095SKrasimir Georgiev } 182412ed095SKrasimir Georgiev return ""; 183412ed095SKrasimir Georgiev } 184412ed095SKrasimir Georgiev 1859ad83fe7SKrasimir Georgiev RawStringFormatStyleManager::RawStringFormatStyleManager( 1869ad83fe7SKrasimir Georgiev const FormatStyle &CodeStyle) { 1879ad83fe7SKrasimir Georgiev for (const auto &RawStringFormat : CodeStyle.RawStringFormats) { 1884527f13aSKrasimir Georgiev llvm::Optional<FormatStyle> LanguageStyle = 1894527f13aSKrasimir Georgiev CodeStyle.GetLanguageStyle(RawStringFormat.Language); 1904527f13aSKrasimir Georgiev if (!LanguageStyle) { 1914527f13aSKrasimir Georgiev FormatStyle PredefinedStyle; 1929ad83fe7SKrasimir Georgiev if (!getPredefinedStyle(RawStringFormat.BasedOnStyle, 1934527f13aSKrasimir Georgiev RawStringFormat.Language, &PredefinedStyle)) { 1944527f13aSKrasimir Georgiev PredefinedStyle = getLLVMStyle(); 1954527f13aSKrasimir Georgiev PredefinedStyle.Language = RawStringFormat.Language; 1969ad83fe7SKrasimir Georgiev } 1974527f13aSKrasimir Georgiev LanguageStyle = PredefinedStyle; 1984527f13aSKrasimir Georgiev } 1994527f13aSKrasimir Georgiev LanguageStyle->ColumnLimit = CodeStyle.ColumnLimit; 2002537e220SKrasimir Georgiev for (StringRef Delimiter : RawStringFormat.Delimiters) { 2014527f13aSKrasimir Georgiev DelimiterStyle.insert({Delimiter, *LanguageStyle}); 2024527f13aSKrasimir Georgiev } 2032537e220SKrasimir Georgiev for (StringRef EnclosingFunction : RawStringFormat.EnclosingFunctions) { 2042537e220SKrasimir Georgiev EnclosingFunctionStyle.insert({EnclosingFunction, *LanguageStyle}); 2052537e220SKrasimir Georgiev } 2069ad83fe7SKrasimir Georgiev } 2079ad83fe7SKrasimir Georgiev } 2089ad83fe7SKrasimir Georgiev 2099ad83fe7SKrasimir Georgiev llvm::Optional<FormatStyle> 2102537e220SKrasimir Georgiev RawStringFormatStyleManager::getDelimiterStyle(StringRef Delimiter) const { 2119ad83fe7SKrasimir Georgiev auto It = DelimiterStyle.find(Delimiter); 2129ad83fe7SKrasimir Georgiev if (It == DelimiterStyle.end()) 2139ad83fe7SKrasimir Georgiev return None; 2149ad83fe7SKrasimir Georgiev return It->second; 2159ad83fe7SKrasimir Georgiev } 2169ad83fe7SKrasimir Georgiev 2172537e220SKrasimir Georgiev llvm::Optional<FormatStyle> 2182537e220SKrasimir Georgiev RawStringFormatStyleManager::getEnclosingFunctionStyle( 2192537e220SKrasimir Georgiev StringRef EnclosingFunction) const { 2202537e220SKrasimir Georgiev auto It = EnclosingFunctionStyle.find(EnclosingFunction); 2212537e220SKrasimir Georgiev if (It == EnclosingFunctionStyle.end()) 2222537e220SKrasimir Georgiev return None; 2232537e220SKrasimir Georgiev return It->second; 2242537e220SKrasimir Georgiev } 2252537e220SKrasimir Georgiev 226de0328aeSDaniel Jasper ContinuationIndenter::ContinuationIndenter(const FormatStyle &Style, 227d0ec0d62SDaniel Jasper const AdditionalKeywords &Keywords, 228635423e6SEric Liu const SourceManager &SourceMgr, 229de0328aeSDaniel Jasper WhitespaceManager &Whitespaces, 230de0328aeSDaniel Jasper encoding::Encoding Encoding, 231de0328aeSDaniel Jasper bool BinPackInconclusiveFunctions) 232d0ec0d62SDaniel Jasper : Style(Style), Keywords(Keywords), SourceMgr(SourceMgr), 233d0ec0d62SDaniel Jasper Whitespaces(Whitespaces), Encoding(Encoding), 234ce9161a5SAlexander Kornienko BinPackInconclusiveFunctions(BinPackInconclusiveFunctions), 2359ad83fe7SKrasimir Georgiev CommentPragmasRegex(Style.CommentPragmas), RawStringFormats(Style) {} 236de0328aeSDaniel Jasper 2379fe0e8daSDaniel Jasper LineState ContinuationIndenter::getInitialState(unsigned FirstIndent, 2389ad83fe7SKrasimir Georgiev unsigned FirstStartColumn, 2391c5d9df8SDaniel Jasper const AnnotatedLine *Line, 2401c5d9df8SDaniel Jasper bool DryRun) { 241de0328aeSDaniel Jasper LineState State; 2429fe0e8daSDaniel Jasper State.FirstIndent = FirstIndent; 2439ad83fe7SKrasimir Georgiev if (FirstStartColumn && Line->First->NewlinesBefore == 0) 2449ad83fe7SKrasimir Georgiev State.Column = FirstStartColumn; 2459ad83fe7SKrasimir Georgiev else 246de0328aeSDaniel Jasper State.Column = FirstIndent; 247ad47c907SKrasimir Georgiev // With preprocessor directive indentation, the line starts on column 0 248ad47c907SKrasimir Georgiev // since it's indented after the hash, but FirstIndent is set to the 249ad47c907SKrasimir Georgiev // preprocessor indent. 250ad47c907SKrasimir Georgiev if (Style.IndentPPDirectives == FormatStyle::PPDIS_AfterHash && 251ad47c907SKrasimir Georgiev (Line->Type == LT_PreprocessorDirective || 252ad47c907SKrasimir Georgiev Line->Type == LT_ImportStatement)) 253ad47c907SKrasimir Georgiev State.Column = 0; 2549fe0e8daSDaniel Jasper State.Line = Line; 2559fe0e8daSDaniel Jasper State.NextToken = Line->First; 2568b4bc76aSKrasimir Georgiev State.Stack.push_back(ParenState(/*Tok=*/nullptr, FirstIndent, FirstIndent, 257de0328aeSDaniel Jasper /*AvoidBinPacking=*/false, 258de0328aeSDaniel Jasper /*NoLineBreak=*/false)); 259de0328aeSDaniel Jasper State.LineContainsContinuedForLoopSection = false; 26035599fdfSKrasimir Georgiev State.NoContinuation = false; 261de0328aeSDaniel Jasper State.StartOfStringLiteral = 0; 26205cd5868SDaniel Jasper State.StartOfLineLevel = 0; 26305cd5868SDaniel Jasper State.LowestLevelOnLine = 0; 264de0328aeSDaniel Jasper State.IgnoreStackForComparison = false; 265de0328aeSDaniel Jasper 26626b144ccSKrasimir Georgiev if (Style.Language == FormatStyle::LK_TextProto) { 26726b144ccSKrasimir Georgiev // We need this in order to deal with the bin packing of text fields at 26826b144ccSKrasimir Georgiev // global scope. 26926b144ccSKrasimir Georgiev State.Stack.back().AvoidBinPacking = true; 27026b144ccSKrasimir Georgiev State.Stack.back().BreakBeforeParameter = true; 271c8b461b3SKrasimir Georgiev State.Stack.back().AlignColons = false; 27226b144ccSKrasimir Georgiev } 27326b144ccSKrasimir Georgiev 274de0328aeSDaniel Jasper // The first token has already been indented and thus consumed. 2751c5d9df8SDaniel Jasper moveStateToNextToken(State, DryRun, /*Newline=*/false); 276de0328aeSDaniel Jasper return State; 277de0328aeSDaniel Jasper } 278de0328aeSDaniel Jasper 279de0328aeSDaniel Jasper bool ContinuationIndenter::canBreak(const LineState &State) { 280de0328aeSDaniel Jasper const FormatToken &Current = *State.NextToken; 281de0328aeSDaniel Jasper const FormatToken &Previous = *Current.Previous; 282de0328aeSDaniel Jasper assert(&Previous == Current.Previous); 28389628f64SManuel Klimek if (!Current.CanBreakBefore && !(State.Stack.back().BreakBeforeClosingBrace && 284bd73bcf8SDaniel Jasper Current.closesBlockOrBlockTypeList(Style))) 285de0328aeSDaniel Jasper return false; 286de0328aeSDaniel Jasper // The opening "{" of a braced list has to be on the same line as the first 287de0328aeSDaniel Jasper // element if it is nested in another braced init list or function call. 288de0328aeSDaniel Jasper if (!Current.MustBreakBefore && Previous.is(tok::l_brace) && 289f5acd11dSBruno Ricci Previous.isNot(TT_DictLiteral) && Previous.is(BK_BracedInit) && 290b05a81deSDaniel Jasper Previous.Previous && 291de0328aeSDaniel Jasper Previous.Previous->isOneOf(tok::l_brace, tok::l_paren, tok::comma)) 292de0328aeSDaniel Jasper return false; 293de0328aeSDaniel Jasper // This prevents breaks like: 294de0328aeSDaniel Jasper // ... 295de0328aeSDaniel Jasper // SomeParameter, OtherParameter).DoSomething( 296de0328aeSDaniel Jasper // ... 297de0328aeSDaniel Jasper // As they hide "DoSomething" and are generally bad for readability. 2988f59ae53SDaniel Jasper if (Previous.opensScope() && Previous.isNot(tok::l_brace) && 299fcfac10cSDaniel Jasper State.LowestLevelOnLine < State.StartOfLineLevel && 300fcfac10cSDaniel Jasper State.LowestLevelOnLine < Current.NestingLevel) 301de0328aeSDaniel Jasper return false; 3024c6e0059SDaniel Jasper if (Current.isMemberAccess() && State.Stack.back().ContainsUnwrappedBuilder) 3034c6e0059SDaniel Jasper return false; 304114a2bc9SDaniel Jasper 305114a2bc9SDaniel Jasper // Don't create a 'hanging' indent if there are multiple blocks in a single 306114a2bc9SDaniel Jasper // statement. 3074b444495SDaniel Jasper if (Previous.is(tok::l_brace) && State.Stack.size() > 1 && 3084b444495SDaniel Jasper State.Stack[State.Stack.size() - 2].NestedBlockInlined && 309114a2bc9SDaniel Jasper State.Stack[State.Stack.size() - 2].HasMultipleNestedBlocks) 310114a2bc9SDaniel Jasper return false; 311114a2bc9SDaniel Jasper 312e068ac77SDaniel Jasper // Don't break after very short return types (e.g. "void") as that is often 313e068ac77SDaniel Jasper // unexpected. 314448592eeSZachary Turner if (Current.is(TT_FunctionDeclarationName) && State.Column < 6) { 315448592eeSZachary Turner if (Style.AlwaysBreakAfterReturnType == FormatStyle::RTBS_None) 316e068ac77SDaniel Jasper return false; 317448592eeSZachary Turner } 318e068ac77SDaniel Jasper 319240527caSDaniel Jasper // If binary operators are moved to the next line (including commas for some 320240527caSDaniel Jasper // styles of constructor initializers), that's always ok. 321240527caSDaniel Jasper if (!Current.isOneOf(TT_BinaryOperator, tok::comma) && 322240527caSDaniel Jasper State.Stack.back().NoLineBreakInOperand) 323240527caSDaniel Jasper return false; 324240527caSDaniel Jasper 325e7c4b590SJacek Olesiak if (Previous.is(tok::l_square) && Previous.is(TT_ObjCMethodExpr)) 326e7c4b590SJacek Olesiak return false; 327e7c4b590SJacek Olesiak 328de0328aeSDaniel Jasper return !State.Stack.back().NoLineBreak; 329de0328aeSDaniel Jasper } 330de0328aeSDaniel Jasper 331de0328aeSDaniel Jasper bool ContinuationIndenter::mustBreak(const LineState &State) { 332de0328aeSDaniel Jasper const FormatToken &Current = *State.NextToken; 333de0328aeSDaniel Jasper const FormatToken &Previous = *Current.Previous; 334fa0118e6SWawha if (Style.BraceWrapping.BeforeLambdaBody && Current.CanBreakBefore && 3355b8ffb41Smydeveloperday Current.is(TT_LambdaLBrace) && Previous.isNot(TT_LineComment)) { 336fa0118e6SWawha auto LambdaBodyLength = getLengthToMatchingParen(Current, State.Stack); 337fa0118e6SWawha return (LambdaBodyLength > getColumnLimit(State)); 338fa0118e6SWawha } 339a98b7b01SDaniel Jasper if (Current.MustBreakBefore || Current.is(TT_InlineASMColon)) 340de0328aeSDaniel Jasper return true; 3411db6c386SDaniel Jasper if (State.Stack.back().BreakBeforeClosingBrace && 342bd73bcf8SDaniel Jasper Current.closesBlockOrBlockTypeList(Style)) 343de0328aeSDaniel Jasper return true; 344de0328aeSDaniel Jasper if (Previous.is(tok::semi) && State.LineContainsContinuedForLoopSection) 345de0328aeSDaniel Jasper return true; 346fb7f5c08SJacek Olesiak if (Style.Language == FormatStyle::LK_ObjC && 34728ecd7f0SKanglei Fang Style.ObjCBreakBeforeNestedBlockParam && 348fb7f5c08SJacek Olesiak Current.ObjCSelectorNameParts > 1 && 349fb7f5c08SJacek Olesiak Current.startsSequence(TT_SelectorName, tok::colon, tok::caret)) { 350fb7f5c08SJacek Olesiak return true; 351fb7f5c08SJacek Olesiak } 352d1b412aeSJonathan Coe // Avoid producing inconsistent states by requiring breaks where they are not 353d1b412aeSJonathan Coe // permitted for C# generic type constraints. 354d1b412aeSJonathan Coe if (State.Stack.back().IsCSharpGenericTypeConstraint && 355d1b412aeSJonathan Coe Previous.isNot(TT_CSharpGenericTypeConstraintComma)) 356d1b412aeSJonathan Coe return false; 357ec01cd6eSDaniel Jasper if ((startsNextParameter(Current, Style) || Previous.is(tok::semi) || 35800693b08SDaniel Jasper (Previous.is(TT_TemplateCloser) && Current.is(TT_StartOfName) && 3591dbc2105SDaniel Jasper Style.isCpp() && 36006ca0fc6SDaniel Jasper // FIXME: This is a temporary workaround for the case where clang-format 36106ca0fc6SDaniel Jasper // sets BreakBeforeParameter to avoid bin packing and this creates a 36206ca0fc6SDaniel Jasper // completely unnecessary line break after a template type that isn't 36306ca0fc6SDaniel Jasper // line-wrapped. 36406ca0fc6SDaniel Jasper (Previous.NestingLevel == 1 || Style.BinPackParameters)) || 3653e0dcc27SDaniel Jasper (Style.BreakBeforeTernaryOperators && Current.is(TT_ConditionalExpr) && 3663e0dcc27SDaniel Jasper Previous.isNot(tok::question)) || 367165b29e2SDaniel Jasper (!Style.BreakBeforeTernaryOperators && 3683e0dcc27SDaniel Jasper Previous.is(TT_ConditionalExpr))) && 369de0328aeSDaniel Jasper State.Stack.back().BreakBeforeParameter && !Current.isTrailingComment() && 370de0328aeSDaniel Jasper !Current.isOneOf(tok::r_paren, tok::r_brace)) 371de0328aeSDaniel Jasper return true; 3724db94094SFrancois Ferrand if (State.Stack.back().IsChainedConditional && 3734db94094SFrancois Ferrand ((Style.BreakBeforeTernaryOperators && Current.is(TT_ConditionalExpr) && 3744db94094SFrancois Ferrand Current.is(tok::colon)) || 3754db94094SFrancois Ferrand (!Style.BreakBeforeTernaryOperators && Previous.is(TT_ConditionalExpr) && 3764db94094SFrancois Ferrand Previous.is(tok::colon)))) 3774db94094SFrancois Ferrand return true; 378a98b7b01SDaniel Jasper if (((Previous.is(TT_DictLiteral) && Previous.is(tok::l_brace)) || 379ccff4d1eSDaniel Jasper (Previous.is(TT_ArrayInitializerLSquare) && 38026b144ccSKrasimir Georgiev Previous.ParameterCount > 1) || 38126b144ccSKrasimir Georgiev opensProtoMessageField(Previous, Style)) && 382db8804b3SDaniel Jasper Style.ColumnLimit > 0 && 3838b4bc76aSKrasimir Georgiev getLengthToMatchingParen(Previous, State.Stack) + State.Column - 1 > 384199d0c9eSDaniel Jasper getColumnLimit(State)) 385d489dd34SDaniel Jasper return true; 386a6b6d51bSFrancois Ferrand 387a6b6d51bSFrancois Ferrand const FormatToken &BreakConstructorInitializersToken = 388a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon 389a6b6d51bSFrancois Ferrand ? Previous 390a6b6d51bSFrancois Ferrand : Current; 391a6b6d51bSFrancois Ferrand if (BreakConstructorInitializersToken.is(TT_CtorInitializerColon) && 392a6b6d51bSFrancois Ferrand (State.Column + State.Line->Last->TotalLength - Previous.TotalLength > 3937b259cdaSDaniel Jasper getColumnLimit(State) || 3947b259cdaSDaniel Jasper State.Stack.back().BreakBeforeParameter) && 395a6b6d51bSFrancois Ferrand (Style.AllowShortFunctionsOnASingleLine != FormatStyle::SFS_All || 396a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers != FormatStyle::BCIS_BeforeColon || 397a6b6d51bSFrancois Ferrand Style.ColumnLimit != 0)) 3985d2587daSDaniel Jasper return true; 399a6b6d51bSFrancois Ferrand 400fd36f0b5SDaniel Jasper if (Current.is(TT_ObjCMethodExpr) && !Previous.is(TT_SelectorName) && 401fd36f0b5SDaniel Jasper State.Line->startsWith(TT_ObjCMethodSpecifier)) 402fd36f0b5SDaniel Jasper return true; 403da5cca4fSKrasimir Georgiev if (Current.is(TT_SelectorName) && !Previous.is(tok::at) && 404da5cca4fSKrasimir Georgiev State.Stack.back().ObjCSelectorNameFound && 405244022a3SSamuel Giddins State.Stack.back().BreakBeforeParameter && 406244022a3SSamuel Giddins (Style.ObjCBreakBeforeNestedBlockParam || 407244022a3SSamuel Giddins !Current.startsSequence(TT_SelectorName, tok::colon, tok::caret))) 4082746a308SDaniel Jasper return true; 409de0328aeSDaniel Jasper 4102aaedd3aSDaniel Jasper unsigned NewLineColumn = getNewLineColumn(State); 411a3cd2164SDaniel Jasper if (Current.isMemberAccess() && Style.ColumnLimit != 0 && 41228024560SDaniel Jasper State.Column + getLengthToNextOperator(Current) > Style.ColumnLimit && 41328024560SDaniel Jasper (State.Column > NewLineColumn || 41428024560SDaniel Jasper Current.NestingLevel < State.StartOfLineLevel)) 41528024560SDaniel Jasper return true; 41628024560SDaniel Jasper 417e61f9f99SDaniel Jasper if (startsSegmentOfBuilderTypeCall(Current) && 418e61f9f99SDaniel Jasper (State.Stack.back().CallContinuation != 0 || 41951c868e9SDaniel Jasper State.Stack.back().BreakBeforeParameter) && 42051c868e9SDaniel Jasper // JavaScript is treated different here as there is a frequent pattern: 42151c868e9SDaniel Jasper // SomeFunction(function() { 42251c868e9SDaniel Jasper // ... 42351c868e9SDaniel Jasper // }.bind(...)); 42451c868e9SDaniel Jasper // FIXME: We should find a more generic solution to this problem. 425142e79b8Smydeveloperday !(State.Column <= NewLineColumn && Style.isJavaScript()) && 4265bcf99b4SPaul Hoad !(Previous.closesScopeAfterBlock() && State.Column <= NewLineColumn)) 427e61f9f99SDaniel Jasper return true; 428e61f9f99SDaniel Jasper 42958e6fe5bSFrancois Ferrand // If the template declaration spans multiple lines, force wrap before the 43058e6fe5bSFrancois Ferrand // function/class declaration 43158e6fe5bSFrancois Ferrand if (Previous.ClosesTemplateDeclaration && 4320fb19de0SKrasimir Georgiev State.Stack.back().BreakBeforeParameter && Current.CanBreakBefore) 43358e6fe5bSFrancois Ferrand return true; 43458e6fe5bSFrancois Ferrand 435072ae7c1Smydeveloperday if (!State.Line->First->is(tok::kw_enum) && State.Column <= NewLineColumn) 4365d2587daSDaniel Jasper return false; 437173504e4SDaniel Jasper 4382aaedd3aSDaniel Jasper if (Style.AlwaysBreakBeforeMultilineStrings && 4391bf729ceSDaniel Jasper (NewLineColumn == State.FirstIndent + Style.ContinuationIndentWidth || 4409fb676a2SDaniel Jasper Previous.is(tok::comma) || Current.NestingLevel < 2) && 441cbb726d0SPaul Hoad !Previous.isOneOf(tok::kw_return, tok::lessless, tok::at, 442cbb726d0SPaul Hoad Keywords.kw_dollar) && 4432aaedd3aSDaniel Jasper !Previous.isOneOf(TT_InlineASMColon, TT_ConditionalExpr) && 4442aaedd3aSDaniel Jasper nextIsMultilineString(State)) 4452aaedd3aSDaniel Jasper return true; 4462aaedd3aSDaniel Jasper 447173504e4SDaniel Jasper // Using CanBreakBefore here and below takes care of the decision whether the 448173504e4SDaniel Jasper // current style uses wrapping before or after operators for the given 449173504e4SDaniel Jasper // operator. 450173504e4SDaniel Jasper if (Previous.is(TT_BinaryOperator) && Current.CanBreakBefore) { 451de0328aeSDaniel Jasper // If we need to break somewhere inside the LHS of a binary expression, we 452de0328aeSDaniel Jasper // should also break after the operator. Otherwise, the formatting would 453de0328aeSDaniel Jasper // hide the operator precedence, e.g. in: 454de0328aeSDaniel Jasper // if (aaaaaaaaaaaaaa == 455de0328aeSDaniel Jasper // bbbbbbbbbbbbbb && c) {.. 456de0328aeSDaniel Jasper // For comparisons, we only apply this rule, if the LHS is a binary 457de0328aeSDaniel Jasper // expression itself as otherwise, the line breaks seem superfluous. 458de0328aeSDaniel Jasper // We need special cases for ">>" which we have split into two ">" while 459de0328aeSDaniel Jasper // lexing in order to make template parsing easier. 460de0328aeSDaniel Jasper bool IsComparison = (Previous.getPrecedence() == prec::Relational || 461c70f1d63SRichard Smith Previous.getPrecedence() == prec::Equality || 462c70f1d63SRichard Smith Previous.getPrecedence() == prec::Spaceship) && 463de0328aeSDaniel Jasper Previous.Previous && 464a98b7b01SDaniel Jasper Previous.Previous->isNot(TT_BinaryOperator); // For >>. 465de0328aeSDaniel Jasper bool LHSIsBinaryExpr = 466562ecd44SDaniel Jasper Previous.Previous && Previous.Previous->EndsBinaryExpression; 467173504e4SDaniel Jasper if ((!IsComparison || LHSIsBinaryExpr) && !Current.isTrailingComment() && 468de0328aeSDaniel Jasper Previous.getPrecedence() != prec::Assignment && 469de0328aeSDaniel Jasper State.Stack.back().BreakBeforeParameter) 470de0328aeSDaniel Jasper return true; 471173504e4SDaniel Jasper } else if (Current.is(TT_BinaryOperator) && Current.CanBreakBefore && 472173504e4SDaniel Jasper State.Stack.back().BreakBeforeParameter) { 4733219e43cSDaniel Jasper return true; 474de0328aeSDaniel Jasper } 475de0328aeSDaniel Jasper 476de0328aeSDaniel Jasper // Same as above, but for the first "<<" operator. 477a98b7b01SDaniel Jasper if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator) && 47886b2dfdcSAlexander Kornienko State.Stack.back().BreakBeforeParameter && 479de0328aeSDaniel Jasper State.Stack.back().FirstLessLess == 0) 480de0328aeSDaniel Jasper return true; 481de0328aeSDaniel Jasper 482211e1329SDaniel Jasper if (Current.NestingLevel == 0 && !Current.isTrailingComment()) { 483f090f031SDaniel Jasper // Always break after "template <...>" and leading annotations. This is only 484f090f031SDaniel Jasper // for cases where the entire line does not fit on a single line as a 485f090f031SDaniel Jasper // different LineFormatter would be used otherwise. 486211e1329SDaniel Jasper if (Previous.ClosesTemplateDeclaration) 48758e6fe5bSFrancois Ferrand return Style.AlwaysBreakTemplateDeclarations != FormatStyle::BTDS_No; 488*47a9eb21SArmen Khachkinaev if (Previous.is(TT_FunctionAnnotationRParen) && 489*47a9eb21SArmen Khachkinaev State.Line->Type != LT_PreprocessorDirective) 490f090f031SDaniel Jasper return true; 491beb03938SNico Weber if (Previous.is(TT_LeadingJavaAnnotation) && Current.isNot(tok::l_paren) && 492beb03938SNico Weber Current.isNot(TT_LeadingJavaAnnotation)) 493211e1329SDaniel Jasper return true; 494211e1329SDaniel Jasper } 495de0328aeSDaniel Jasper 496142e79b8Smydeveloperday if (Style.isJavaScript() && Previous.is(tok::r_paren) && 497142e79b8Smydeveloperday Previous.is(TT_JavaAnnotation)) { 498befb9dc3SKrasimir Georgiev // Break after the closing parenthesis of TypeScript decorators before 499befb9dc3SKrasimir Georgiev // functions, getters and setters. 500befb9dc3SKrasimir Georgiev static const llvm::StringSet<> BreakBeforeDecoratedTokens = {"get", "set", 501befb9dc3SKrasimir Georgiev "function"}; 5028d1c8545SBjörn Schäpers if (BreakBeforeDecoratedTokens.contains(Current.TokenText)) 503f3671a68SKrasimir Georgiev return true; 504f3671a68SKrasimir Georgiev } 505f3671a68SKrasimir Georgiev 5064355e7f0SDaniel Jasper // If the return type spans multiple lines, wrap before the function name. 507a2f963bbSPaul Hoad if (((Current.is(TT_FunctionDeclarationName) && 508a2f963bbSPaul Hoad // Don't break before a C# function when no break after return type 509a2f963bbSPaul Hoad (!Style.isCSharp() || 5104f4f2783SKrasimir Georgiev Style.AlwaysBreakAfterReturnType != FormatStyle::RTBS_None) && 5114f4f2783SKrasimir Georgiev // Don't always break between a JavaScript `function` and the function 5124f4f2783SKrasimir Georgiev // name. 513142e79b8Smydeveloperday !Style.isJavaScript()) || 514525579d4SBirunthan Mohanathas (Current.is(tok::kw_operator) && !Previous.is(tok::coloncolon))) && 5150c9772e8SDaniel Jasper !Previous.is(tok::kw_template) && State.Stack.back().BreakBeforeParameter) 516de0328aeSDaniel Jasper return true; 5174355e7f0SDaniel Jasper 5189697281eSDaniel Jasper // The following could be precomputed as they do not depend on the state. 5199697281eSDaniel Jasper // However, as they should take effect only if the UnwrappedLine does not fit 5209697281eSDaniel Jasper // into the ColumnLimit, they are checked here in the ContinuationIndenter. 521f5acd11dSBruno Ricci if (Style.ColumnLimit != 0 && Previous.is(BK_Block) && 52235995679SDaniel Jasper Previous.is(tok::l_brace) && !Current.isOneOf(tok::r_brace, tok::comment)) 5239697281eSDaniel Jasper return true; 5249697281eSDaniel Jasper 5250a589416SDaniel Jasper if (Current.is(tok::lessless) && 5260a589416SDaniel Jasper ((Previous.is(tok::identifier) && Previous.TokenText == "endl") || 5270a589416SDaniel Jasper (Previous.Tok.isLiteral() && (Previous.TokenText.endswith("\\n\"") || 5280a589416SDaniel Jasper Previous.TokenText == "\'\\n\'")))) 52969963129SDaniel Jasper return true; 53069963129SDaniel Jasper 53135599fdfSKrasimir Georgiev if (Previous.is(TT_BlockComment) && Previous.IsMultiline) 53235599fdfSKrasimir Georgiev return true; 53335599fdfSKrasimir Georgiev 53435599fdfSKrasimir Georgiev if (State.NoContinuation) 53535599fdfSKrasimir Georgiev return true; 53635599fdfSKrasimir Georgiev 537de0328aeSDaniel Jasper return false; 538de0328aeSDaniel Jasper } 539de0328aeSDaniel Jasper 540de0328aeSDaniel Jasper unsigned ContinuationIndenter::addTokenToState(LineState &State, bool Newline, 5418de9ed05SDaniel Jasper bool DryRun, 5428de9ed05SDaniel Jasper unsigned ExtraSpaces) { 543de0328aeSDaniel Jasper const FormatToken &Current = *State.NextToken; 544de0328aeSDaniel Jasper 545819788daSManuel Klimek assert(!State.Stack.empty()); 54635599fdfSKrasimir Georgiev State.NoContinuation = false; 54735599fdfSKrasimir Georgiev 548a98b7b01SDaniel Jasper if ((Current.is(TT_ImplicitStringLiteral) && 5492145bc02SCraig Topper (Current.Previous->Tok.getIdentifierInfo() == nullptr || 5509885784dSDaniel Jasper Current.Previous->Tok.getIdentifierInfo()->getPPKeywordID() == 5519885784dSDaniel Jasper tok::pp_not_keyword))) { 55230526e79SDaniel Jasper unsigned EndColumn = 55330526e79SDaniel Jasper SourceMgr.getSpellingColumnNumber(Current.WhitespaceRange.getEnd()); 55430526e79SDaniel Jasper if (Current.LastNewlineOffset != 0) { 55530526e79SDaniel Jasper // If there is a newline within this token, the final column will solely 55630526e79SDaniel Jasper // determined by the current end column. 55730526e79SDaniel Jasper State.Column = EndColumn; 55830526e79SDaniel Jasper } else { 55930526e79SDaniel Jasper unsigned StartColumn = 56030526e79SDaniel Jasper SourceMgr.getSpellingColumnNumber(Current.WhitespaceRange.getBegin()); 56130526e79SDaniel Jasper assert(EndColumn >= StartColumn); 56230526e79SDaniel Jasper State.Column += EndColumn - StartColumn; 56330526e79SDaniel Jasper } 564240dfda3SDaniel Jasper moveStateToNextToken(State, DryRun, /*Newline=*/false); 565de0328aeSDaniel Jasper return 0; 566de0328aeSDaniel Jasper } 567de0328aeSDaniel Jasper 5681f80396dSAlexander Kornienko unsigned Penalty = 0; 5691f80396dSAlexander Kornienko if (Newline) 5701f80396dSAlexander Kornienko Penalty = addTokenOnNewLine(State, DryRun); 5711f80396dSAlexander Kornienko else 57248437ce5SDaniel Jasper addTokenOnCurrentLine(State, DryRun, ExtraSpaces); 5731f80396dSAlexander Kornienko 5741f80396dSAlexander Kornienko return moveStateToNextToken(State, DryRun, Newline) + Penalty; 5751f80396dSAlexander Kornienko } 5761f80396dSAlexander Kornienko 57748437ce5SDaniel Jasper void ContinuationIndenter::addTokenOnCurrentLine(LineState &State, bool DryRun, 5781f80396dSAlexander Kornienko unsigned ExtraSpaces) { 57971814b44SManuel Klimek FormatToken &Current = *State.NextToken; 5801f80396dSAlexander Kornienko const FormatToken &Previous = *State.NextToken->Previous; 5811f80396dSAlexander Kornienko if (Current.is(tok::equal) && 58205cd5868SDaniel Jasper (State.Line->First->is(tok::kw_for) || Current.NestingLevel == 0) && 5831f80396dSAlexander Kornienko State.Stack.back().VariablePos == 0) { 5841f80396dSAlexander Kornienko State.Stack.back().VariablePos = State.Column; 5851f80396dSAlexander Kornienko // Move over * and & if they are bound to the variable name. 5861f80396dSAlexander Kornienko const FormatToken *Tok = &Previous; 5871f80396dSAlexander Kornienko while (Tok && State.Stack.back().VariablePos >= Tok->ColumnWidth) { 5881f80396dSAlexander Kornienko State.Stack.back().VariablePos -= Tok->ColumnWidth; 5891f80396dSAlexander Kornienko if (Tok->SpacesRequiredBefore != 0) 5901f80396dSAlexander Kornienko break; 5911f80396dSAlexander Kornienko Tok = Tok->Previous; 5921f80396dSAlexander Kornienko } 5931f80396dSAlexander Kornienko if (Previous.PartOfMultiVariableDeclStmt) 5941f80396dSAlexander Kornienko State.Stack.back().LastSpace = State.Stack.back().VariablePos; 5951f80396dSAlexander Kornienko } 5961f80396dSAlexander Kornienko 5971f80396dSAlexander Kornienko unsigned Spaces = Current.SpacesRequiredBefore + ExtraSpaces; 5981f80396dSAlexander Kornienko 599ad47c907SKrasimir Georgiev // Indent preprocessor directives after the hash if required. 600ad47c907SKrasimir Georgiev int PPColumnCorrection = 0; 601ad47c907SKrasimir Georgiev if (Style.IndentPPDirectives == FormatStyle::PPDIS_AfterHash && 602ad47c907SKrasimir Georgiev Previous.is(tok::hash) && State.FirstIndent > 0 && 603ad47c907SKrasimir Georgiev (State.Line->Type == LT_PreprocessorDirective || 604ad47c907SKrasimir Georgiev State.Line->Type == LT_ImportStatement)) { 605ad47c907SKrasimir Georgiev Spaces += State.FirstIndent; 606ad47c907SKrasimir Georgiev 607ad47c907SKrasimir Georgiev // For preprocessor indent with tabs, State.Column will be 1 because of the 608ad47c907SKrasimir Georgiev // hash. This causes second-level indents onward to have an extra space 609ad47c907SKrasimir Georgiev // after the tabs. We avoid this misalignment by subtracting 1 from the 610ad47c907SKrasimir Georgiev // column value passed to replaceWhitespace(). 611ad47c907SKrasimir Georgiev if (Style.UseTab != FormatStyle::UT_Never) 612ad47c907SKrasimir Georgiev PPColumnCorrection = -1; 613ad47c907SKrasimir Georgiev } 614ad47c907SKrasimir Georgiev 6151f80396dSAlexander Kornienko if (!DryRun) 6167d42f3f7SDaniel Jasper Whitespaces.replaceWhitespace(Current, /*Newlines=*/0, Spaces, 617ad47c907SKrasimir Georgiev State.Column + Spaces + PPColumnCorrection); 6181f80396dSAlexander Kornienko 6190ef8ee19SAndi-Bogdan Postelnicu // If "BreakBeforeInheritanceComma" mode, don't break within the inheritance 6200ef8ee19SAndi-Bogdan Postelnicu // declaration unless there is multiple inheritance. 6216bb103f9SFrancois Ferrand if (Style.BreakInheritanceList == FormatStyle::BILS_BeforeComma && 6226bb103f9SFrancois Ferrand Current.is(TT_InheritanceColon)) 6236bb103f9SFrancois Ferrand State.Stack.back().NoLineBreak = true; 6246bb103f9SFrancois Ferrand if (Style.BreakInheritanceList == FormatStyle::BILS_AfterColon && 6256bb103f9SFrancois Ferrand Previous.is(TT_InheritanceColon)) 6260ef8ee19SAndi-Bogdan Postelnicu State.Stack.back().NoLineBreak = true; 6270ef8ee19SAndi-Bogdan Postelnicu 628a98b7b01SDaniel Jasper if (Current.is(TT_SelectorName) && 629b88b25feSDaniel Jasper !State.Stack.back().ObjCSelectorNameFound) { 63006a26957SDaniel Jasper unsigned MinIndent = 63106a26957SDaniel Jasper std::max(State.FirstIndent + Style.ContinuationIndentWidth, 63206a26957SDaniel Jasper State.Stack.back().Indent); 63306a26957SDaniel Jasper unsigned FirstColonPos = State.Column + Spaces + Current.ColumnWidth; 634b88b25feSDaniel Jasper if (Current.LongestObjCSelectorName == 0) 635b88b25feSDaniel Jasper State.Stack.back().AlignColons = false; 63606a26957SDaniel Jasper else if (MinIndent + Current.LongestObjCSelectorName > FirstColonPos) 63706a26957SDaniel Jasper State.Stack.back().ColonPos = MinIndent + Current.LongestObjCSelectorName; 6381f80396dSAlexander Kornienko else 63906a26957SDaniel Jasper State.Stack.back().ColonPos = FirstColonPos; 6401f80396dSAlexander Kornienko } 6411f80396dSAlexander Kornienko 6426501f7e8SDaniel Jasper // In "AlwaysBreak" mode, enforce wrapping directly after the parenthesis by 6436501f7e8SDaniel Jasper // disallowing any further line breaks if there is no line break after the 6446501f7e8SDaniel Jasper // opening parenthesis. Don't break if it doesn't conserve columns. 6456501f7e8SDaniel Jasper if (Style.AlignAfterOpenBracket == FormatStyle::BAS_AlwaysBreak && 646d40ddb9dSOwen Pan (Previous.isOneOf(tok::l_paren, TT_TemplateOpener, tok::l_square) || 647f5acd11dSBruno Ricci (Previous.is(tok::l_brace) && Previous.isNot(BK_Block) && 648d40ddb9dSOwen Pan Style.Cpp11BracedListStyle)) && 649b618a985SDaniel Jasper State.Column > getNewLineColumn(State) && 65089628f64SManuel Klimek (!Previous.Previous || !Previous.Previous->isOneOf( 65189628f64SManuel Klimek tok::kw_for, tok::kw_while, tok::kw_switch)) && 652710f8493SDaniel Jasper // Don't do this for simple (no expressions) one-argument function calls 653710f8493SDaniel Jasper // as that feels like needlessly wasting whitespace, e.g.: 654710f8493SDaniel Jasper // 655710f8493SDaniel Jasper // caaaaaaaaaaaall( 656710f8493SDaniel Jasper // caaaaaaaaaaaall( 657710f8493SDaniel Jasper // caaaaaaaaaaaall( 658710f8493SDaniel Jasper // caaaaaaaaaaaaaaaaaaaaaaall(aaaaaaaaaaaaaa, aaaaaaaaa)))); 659710f8493SDaniel Jasper Current.FakeLParens.size() > 0 && 660710f8493SDaniel Jasper Current.FakeLParens.back() > prec::Unknown) 6616501f7e8SDaniel Jasper State.Stack.back().NoLineBreak = true; 66298e0b12dSDaniel Jasper if (Previous.is(TT_TemplateString) && Previous.opensScope()) 66398e0b12dSDaniel Jasper State.Stack.back().NoLineBreak = true; 6646501f7e8SDaniel Jasper 6656501f7e8SDaniel Jasper if (Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign && 66678e2a3c6SJonathan Coe !State.Stack.back().IsCSharpGenericTypeConstraint && 6676501f7e8SDaniel Jasper Previous.opensScope() && Previous.isNot(TT_ObjCMethodExpr) && 668f5acd11dSBruno Ricci (Current.isNot(TT_LineComment) || Previous.is(BK_BracedInit))) { 6691f80396dSAlexander Kornienko State.Stack.back().Indent = State.Column + Spaces; 670e8111502Smydeveloperday State.Stack.back().IsAligned = true; 671e8111502Smydeveloperday } 672ec01cd6eSDaniel Jasper if (State.Stack.back().AvoidBinPacking && startsNextParameter(Current, Style)) 6731f80396dSAlexander Kornienko State.Stack.back().NoLineBreak = true; 674775954beSDaniel Jasper if (startsSegmentOfBuilderTypeCall(Current) && 675775954beSDaniel Jasper State.Column > getNewLineColumn(State)) 6761f80396dSAlexander Kornienko State.Stack.back().ContainsUnwrappedBuilder = true; 6771f80396dSAlexander Kornienko 6786f2b88a3SDaniel Jasper if (Current.is(TT_LambdaArrow) && Style.Language == FormatStyle::LK_Java) 67905cd9292SDaniel Jasper State.Stack.back().NoLineBreak = true; 680d6f17d83SDaniel Jasper if (Current.isMemberAccess() && Previous.is(tok::r_paren) && 681d6f17d83SDaniel Jasper (Previous.MatchingParen && 68298e0b12dSDaniel Jasper (Previous.TotalLength - Previous.MatchingParen->TotalLength > 10))) 683d6f17d83SDaniel Jasper // If there is a function call with long parameters, break before trailing 684d6f17d83SDaniel Jasper // calls. This prevents things like: 685d6f17d83SDaniel Jasper // EXPECT_CALL(SomeLongParameter).Times( 686d6f17d83SDaniel Jasper // 2); 687d6f17d83SDaniel Jasper // We don't want to do this for short parameters as they can just be 688d6f17d83SDaniel Jasper // indexes. 689d6f17d83SDaniel Jasper State.Stack.back().NoLineBreak = true; 690d6f17d83SDaniel Jasper 691240527caSDaniel Jasper // Don't allow the RHS of an operator to be split over multiple lines unless 692240527caSDaniel Jasper // there is a line-break right after the operator. 693240527caSDaniel Jasper // Exclude relational operators, as there, it is always more desirable to 694240527caSDaniel Jasper // have the LHS 'left' of the RHS. 695240527caSDaniel Jasper const FormatToken *P = Current.getPreviousNonComment(); 696240527caSDaniel Jasper if (!Current.is(tok::comment) && P && 697240527caSDaniel Jasper (P->isOneOf(TT_BinaryOperator, tok::comma) || 698240527caSDaniel Jasper (P->is(TT_ConditionalExpr) && P->is(tok::colon))) && 699240527caSDaniel Jasper !P->isOneOf(TT_OverloadedOperator, TT_CtorInitializerComma) && 700240527caSDaniel Jasper P->getPrecedence() != prec::Assignment && 701c70f1d63SRichard Smith P->getPrecedence() != prec::Relational && 702c70f1d63SRichard Smith P->getPrecedence() != prec::Spaceship) { 703240527caSDaniel Jasper bool BreakBeforeOperator = 704b127039cSDaniel Jasper P->MustBreakBefore || P->is(tok::lessless) || 705240527caSDaniel Jasper (P->is(TT_BinaryOperator) && 706240527caSDaniel Jasper Style.BreakBeforeBinaryOperators != FormatStyle::BOS_None) || 707240527caSDaniel Jasper (P->is(TT_ConditionalExpr) && Style.BreakBeforeTernaryOperators); 70821f7dea5SDaniel Jasper // Don't do this if there are only two operands. In these cases, there is 70921f7dea5SDaniel Jasper // always a nice vertical separation between them and the extra line break 71021f7dea5SDaniel Jasper // does not help. 71121f7dea5SDaniel Jasper bool HasTwoOperands = 71221f7dea5SDaniel Jasper P->OperatorIndex == 0 && !P->NextOperator && !P->is(TT_ConditionalExpr); 7130ee04e6eSFrancois Ferrand if ((!BreakBeforeOperator && 7140ee04e6eSFrancois Ferrand !(HasTwoOperands && 7150ee04e6eSFrancois Ferrand Style.AlignOperands != FormatStyle::OAS_DontAlign)) || 716240527caSDaniel Jasper (!State.Stack.back().LastOperatorWrapped && BreakBeforeOperator)) 717240527caSDaniel Jasper State.Stack.back().NoLineBreakInOperand = true; 718240527caSDaniel Jasper } 719240527caSDaniel Jasper 7201f80396dSAlexander Kornienko State.Column += Spaces; 7218acf822bSDaniel Jasper if (Current.isNot(tok::comment) && Previous.is(tok::l_paren) && 72211a0ac66SDaniel Jasper Previous.Previous && 7231361a4c2SNico Weber (Previous.Previous->is(tok::kw_for) || Previous.Previous->isIf())) { 7241f80396dSAlexander Kornienko // Treat the condition inside an if as if it was a second function 7256633ab8aSDaniel Jasper // parameter, i.e. let nested calls have a continuation indent. 7268acf822bSDaniel Jasper State.Stack.back().LastSpace = State.Column; 72711a0ac66SDaniel Jasper State.Stack.back().NestedBlockIndent = State.Column; 72811a0ac66SDaniel Jasper } else if (!Current.isOneOf(tok::comment, tok::caret) && 729804a276fSDaniel Jasper ((Previous.is(tok::comma) && 730804a276fSDaniel Jasper !Previous.is(TT_OverloadedOperator)) || 73111a0ac66SDaniel Jasper (Previous.is(tok::colon) && Previous.is(TT_ObjCMethodExpr)))) { 7321f80396dSAlexander Kornienko State.Stack.back().LastSpace = State.Column; 733a6b6d51bSFrancois Ferrand } else if (Previous.is(TT_CtorInitializerColon) && 734a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers == 735a6b6d51bSFrancois Ferrand FormatStyle::BCIS_AfterColon) { 736a6b6d51bSFrancois Ferrand State.Stack.back().Indent = State.Column; 737a6b6d51bSFrancois Ferrand State.Stack.back().LastSpace = State.Column; 73811a0ac66SDaniel Jasper } else if ((Previous.isOneOf(TT_BinaryOperator, TT_ConditionalExpr, 739a98b7b01SDaniel Jasper TT_CtorInitializerColon)) && 7400e61784aSDaniel Jasper ((Previous.getPrecedence() != prec::Assignment && 7410e61784aSDaniel Jasper (Previous.isNot(tok::lessless) || Previous.OperatorIndex != 0 || 74200492f96SDaniel Jasper Previous.NextOperator)) || 74311a0ac66SDaniel Jasper Current.StartsBinaryExpression)) { 744602a727aSDaniel Jasper // Indent relative to the RHS of the expression unless this is a simple 745602a727aSDaniel Jasper // assignment without binary expression on the RHS. Also indent relative to 746602a727aSDaniel Jasper // unary operators and the colons of constructor initializers. 7474636debcSJonas Toth if (Style.BreakBeforeBinaryOperators == FormatStyle::BOS_None) 7481f80396dSAlexander Kornienko State.Stack.back().LastSpace = State.Column; 74911a0ac66SDaniel Jasper } else if (Previous.is(TT_InheritanceColon)) { 7501f80396dSAlexander Kornienko State.Stack.back().Indent = State.Column; 751f9a5e404SDaniel Jasper State.Stack.back().LastSpace = State.Column; 75278e2a3c6SJonathan Coe } else if (Current.is(TT_CSharpGenericTypeConstraintColon)) { 75378e2a3c6SJonathan Coe State.Stack.back().ColonPos = State.Column; 754f9a5e404SDaniel Jasper } else if (Previous.opensScope()) { 7551f80396dSAlexander Kornienko // If a function has a trailing call, indent all parameters from the 7561f80396dSAlexander Kornienko // opening parenthesis. This avoids confusing indents like: 7571f80396dSAlexander Kornienko // OuterFunction(InnerFunctionCall( // break 7581f80396dSAlexander Kornienko // ParameterToInnerFunction)) // break 7591f80396dSAlexander Kornienko // .SecondInnerFunctionCall(); 7601f80396dSAlexander Kornienko bool HasTrailingCall = false; 7611f80396dSAlexander Kornienko if (Previous.MatchingParen) { 7621f80396dSAlexander Kornienko const FormatToken *Next = Previous.MatchingParen->getNextNonComment(); 7631f80396dSAlexander Kornienko HasTrailingCall = Next && Next->isMemberAccess(); 7641f80396dSAlexander Kornienko } 765d97d5d5cSDaniel Jasper if (HasTrailingCall && State.Stack.size() > 1 && 7661f80396dSAlexander Kornienko State.Stack[State.Stack.size() - 2].CallContinuation == 0) 7671f80396dSAlexander Kornienko State.Stack.back().LastSpace = State.Column; 7681f80396dSAlexander Kornienko } 7691f80396dSAlexander Kornienko } 7701f80396dSAlexander Kornienko 7711f80396dSAlexander Kornienko unsigned ContinuationIndenter::addTokenOnNewLine(LineState &State, 7721f80396dSAlexander Kornienko bool DryRun) { 77371814b44SManuel Klimek FormatToken &Current = *State.NextToken; 7741f80396dSAlexander Kornienko const FormatToken &Previous = *State.NextToken->Previous; 7759f388d04SDaniel Jasper 7761f80396dSAlexander Kornienko // Extra penalty that needs to be added because of the way certain line 7771f80396dSAlexander Kornienko // breaks are chosen. 7781f80396dSAlexander Kornienko unsigned Penalty = 0; 7791f80396dSAlexander Kornienko 780a0407740SDaniel Jasper const FormatToken *PreviousNonComment = Current.getPreviousNonComment(); 781a0407740SDaniel Jasper const FormatToken *NextNonComment = Previous.getNextNonComment(); 782a0407740SDaniel Jasper if (!NextNonComment) 783a0407740SDaniel Jasper NextNonComment = &Current; 78405cd5868SDaniel Jasper // The first line break on any NestingLevel causes an extra penalty in order 7858de9ed05SDaniel Jasper // prefer similar line breaks. 7868de9ed05SDaniel Jasper if (!State.Stack.back().ContainsLineBreak) 7878de9ed05SDaniel Jasper Penalty += 15; 7888de9ed05SDaniel Jasper State.Stack.back().ContainsLineBreak = true; 7898de9ed05SDaniel Jasper 7908de9ed05SDaniel Jasper Penalty += State.NextToken->SplitPenalty; 7918de9ed05SDaniel Jasper 792de0328aeSDaniel Jasper // Breaking before the first "<<" is generally not desirable if the LHS is 7937aacf468SDaniel Jasper // short. Also always add the penalty if the LHS is split over multiple lines 7942b7556ecSDaniel Jasper // to avoid unnecessary line breaks that just work around this penalty. 795a0407740SDaniel Jasper if (NextNonComment->is(tok::lessless) && 796a0407740SDaniel Jasper State.Stack.back().FirstLessLess == 0 && 797004177eeSDaniel Jasper (State.Column <= Style.ColumnLimit / 3 || 79848437ce5SDaniel Jasper State.Stack.back().BreakBeforeParameter)) 7998de9ed05SDaniel Jasper Penalty += Style.PenaltyBreakFirstLessLess; 800de0328aeSDaniel Jasper 8019f388d04SDaniel Jasper State.Column = getNewLineColumn(State); 802ed3f3957SDaniel Jasper 8031e4d6d1cSMark Nauwelaerts // Add Penalty proportional to amount of whitespace away from FirstColumn 8041e4d6d1cSMark Nauwelaerts // This tends to penalize several lines that are far-right indented, 8051e4d6d1cSMark Nauwelaerts // and prefers a line-break prior to such a block, e.g: 8061e4d6d1cSMark Nauwelaerts // 8071e4d6d1cSMark Nauwelaerts // Constructor() : 8081e4d6d1cSMark Nauwelaerts // member(value), looooooooooooooooong_member( 8091e4d6d1cSMark Nauwelaerts // looooooooooong_call(param_1, param_2, param_3)) 8101e4d6d1cSMark Nauwelaerts // would then become 8111e4d6d1cSMark Nauwelaerts // Constructor() : 8121e4d6d1cSMark Nauwelaerts // member(value), 8131e4d6d1cSMark Nauwelaerts // looooooooooooooooong_member( 8141e4d6d1cSMark Nauwelaerts // looooooooooong_call(param_1, param_2, param_3)) 8151e4d6d1cSMark Nauwelaerts if (State.Column > State.FirstIndent) 8161e4d6d1cSMark Nauwelaerts Penalty += 8171e4d6d1cSMark Nauwelaerts Style.PenaltyIndentedWhitespace * (State.Column - State.FirstIndent); 8181e4d6d1cSMark Nauwelaerts 819ed3f3957SDaniel Jasper // Indent nested blocks relative to this column, unless in a very specific 820ed3f3957SDaniel Jasper // JavaScript special case where: 821ed3f3957SDaniel Jasper // 822ed3f3957SDaniel Jasper // var loooooong_name = 823ed3f3957SDaniel Jasper // function() { 824ed3f3957SDaniel Jasper // // code 825ed3f3957SDaniel Jasper // } 826ed3f3957SDaniel Jasper // 82787448c55SDaniel Jasper // is common and should be formatted like a free-standing function. The same 82887448c55SDaniel Jasper // goes for wrapping before the lambda return type arrow. 82987448c55SDaniel Jasper if (!Current.is(TT_LambdaArrow) && 830142e79b8Smydeveloperday (!Style.isJavaScript() || Current.NestingLevel != 0 || 831142e79b8Smydeveloperday !PreviousNonComment || !PreviousNonComment->is(tok::equal) || 83287448c55SDaniel Jasper !Current.isOneOf(Keywords.kw_async, Keywords.kw_function))) 83311a0ac66SDaniel Jasper State.Stack.back().NestedBlockIndent = State.Column; 834ed3f3957SDaniel Jasper 8359f388d04SDaniel Jasper if (NextNonComment->isMemberAccess()) { 8369f388d04SDaniel Jasper if (State.Stack.back().CallContinuation == 0) 837de0328aeSDaniel Jasper State.Stack.back().CallContinuation = State.Column; 838a98b7b01SDaniel Jasper } else if (NextNonComment->is(TT_SelectorName)) { 839b88b25feSDaniel Jasper if (!State.Stack.back().ObjCSelectorNameFound) { 840a0407740SDaniel Jasper if (NextNonComment->LongestObjCSelectorName == 0) { 841b88b25feSDaniel Jasper State.Stack.back().AlignColons = false; 842b88b25feSDaniel Jasper } else { 843b302f9a0SDaniel Jasper State.Stack.back().ColonPos = 844416348efSBen Hamilton (shouldIndentWrappedSelectorName(Style, State.Line->Type) 845a2a4d9c0SDaniel Jasper ? std::max(State.Stack.back().Indent, 846a2a4d9c0SDaniel Jasper State.FirstIndent + Style.ContinuationIndentWidth) 847a2a4d9c0SDaniel Jasper : State.Stack.back().Indent) + 84838d80134SFrancois Ferrand std::max(NextNonComment->LongestObjCSelectorName, 84938d80134SFrancois Ferrand NextNonComment->ColumnWidth); 850b88b25feSDaniel Jasper } 8519f388d04SDaniel Jasper } else if (State.Stack.back().AlignColons && 8529f388d04SDaniel Jasper State.Stack.back().ColonPos <= NextNonComment->ColumnWidth) { 853a0407740SDaniel Jasper State.Stack.back().ColonPos = State.Column + NextNonComment->ColumnWidth; 854de0328aeSDaniel Jasper } 8551fd6f1f8SDaniel Jasper } else if (PreviousNonComment && PreviousNonComment->is(tok::colon) && 856a98b7b01SDaniel Jasper PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) { 857b88b25feSDaniel Jasper // FIXME: This is hacky, find a better way. The problem is that in an ObjC 858b88b25feSDaniel Jasper // method expression, the block should be aligned to the line starting it, 859b88b25feSDaniel Jasper // e.g.: 860b88b25feSDaniel Jasper // [aaaaaaaaaaaaaaa aaaaaaaaa: \\ break for some reason 861b88b25feSDaniel Jasper // ^(int *i) { 862b88b25feSDaniel Jasper // // ... 863b88b25feSDaniel Jasper // }]; 86405cd5868SDaniel Jasper // Thus, we set LastSpace of the next higher NestingLevel, to which we move 865b88b25feSDaniel Jasper // when we consume all of the "}"'s FakeRParens at the "{". 8669a26e773SDaniel Jasper if (State.Stack.size() > 1) 8679f388d04SDaniel Jasper State.Stack[State.Stack.size() - 2].LastSpace = 8689f388d04SDaniel Jasper std::max(State.Stack.back().LastSpace, State.Stack.back().Indent) + 8699f388d04SDaniel Jasper Style.ContinuationIndentWidth; 870de0328aeSDaniel Jasper } 871de0328aeSDaniel Jasper 87235e4122fSDaniel Jasper if ((PreviousNonComment && 87335e4122fSDaniel Jasper PreviousNonComment->isOneOf(tok::comma, tok::semi) && 874de0328aeSDaniel Jasper !State.Stack.back().AvoidBinPacking) || 875a98b7b01SDaniel Jasper Previous.is(TT_BinaryOperator)) 876de0328aeSDaniel Jasper State.Stack.back().BreakBeforeParameter = false; 8770fb19de0SKrasimir Georgiev if (PreviousNonComment && 8780fb19de0SKrasimir Georgiev PreviousNonComment->isOneOf(TT_TemplateCloser, TT_JavaAnnotation) && 87939af6cd5SDaniel Jasper Current.NestingLevel == 0) 880de0328aeSDaniel Jasper State.Stack.back().BreakBeforeParameter = false; 881a0407740SDaniel Jasper if (NextNonComment->is(tok::question) || 882165b29e2SDaniel Jasper (PreviousNonComment && PreviousNonComment->is(tok::question))) 883165b29e2SDaniel Jasper State.Stack.back().BreakBeforeParameter = true; 8845962fa80SDaniel Jasper if (Current.is(TT_BinaryOperator) && Current.CanBreakBefore) 8855962fa80SDaniel Jasper State.Stack.back().BreakBeforeParameter = false; 886de0328aeSDaniel Jasper 887de0328aeSDaniel Jasper if (!DryRun) { 888a004b3f5SMartin Probst unsigned MaxEmptyLinesToKeep = Style.MaxEmptyLinesToKeep + 1; 889a004b3f5SMartin Probst if (Current.is(tok::r_brace) && Current.MatchingParen && 890a004b3f5SMartin Probst // Only strip trailing empty lines for l_braces that have children, i.e. 891a004b3f5SMartin Probst // for function expressions (lambdas, arrows, etc). 892a004b3f5SMartin Probst !Current.MatchingParen->Children.empty()) { 893a004b3f5SMartin Probst // lambdas and arrow functions are expressions, thus their r_brace is not 894a004b3f5SMartin Probst // on its own line, and thus not covered by UnwrappedLineFormatter's logic 895a004b3f5SMartin Probst // about removing empty lines on closing blocks. Special case them here. 896a004b3f5SMartin Probst MaxEmptyLinesToKeep = 1; 897a004b3f5SMartin Probst } 8985bcf99b4SPaul Hoad unsigned Newlines = 8995bcf99b4SPaul Hoad std::max(1u, std::min(Current.NewlinesBefore, MaxEmptyLinesToKeep)); 900a1c30937SKrasimir Georgiev bool ContinuePPDirective = 901a1c30937SKrasimir Georgiev State.Line->InPPDirective && State.Line->Type != LT_ImportStatement; 9027d42f3f7SDaniel Jasper Whitespaces.replaceWhitespace(Current, Newlines, State.Column, State.Column, 903e8111502Smydeveloperday State.Stack.back().IsAligned, 904a1c30937SKrasimir Georgiev ContinuePPDirective); 905de0328aeSDaniel Jasper } 906de0328aeSDaniel Jasper 907de0328aeSDaniel Jasper if (!Current.isTrailingComment()) 908de0328aeSDaniel Jasper State.Stack.back().LastSpace = State.Column; 909602a727aSDaniel Jasper if (Current.is(tok::lessless)) 910602a727aSDaniel Jasper // If we are breaking before a "<<", we always want to indent relative to 911602a727aSDaniel Jasper // RHS. This is necessary only for "<<", as we special-case it and don't 912602a727aSDaniel Jasper // always indent relative to the RHS. 913602a727aSDaniel Jasper State.Stack.back().LastSpace += 3; // 3 -> width of "<< ". 914602a727aSDaniel Jasper 91505cd5868SDaniel Jasper State.StartOfLineLevel = Current.NestingLevel; 91605cd5868SDaniel Jasper State.LowestLevelOnLine = Current.NestingLevel; 917de0328aeSDaniel Jasper 918de0328aeSDaniel Jasper // Any break on this level means that the parent level has been broken 919de0328aeSDaniel Jasper // and we need to avoid bin packing there. 9204b444495SDaniel Jasper bool NestedBlockSpecialCase = 92170c98671Smydeveloperday (!Style.isCpp() && Current.is(tok::r_brace) && State.Stack.size() > 1 && 92270c98671Smydeveloperday State.Stack[State.Stack.size() - 2].NestedBlockInlined) || 92370c98671Smydeveloperday (Style.Language == FormatStyle::LK_ObjC && Current.is(tok::r_brace) && 92470c98671Smydeveloperday State.Stack.size() > 1 && !Style.ObjCBreakBeforeNestedBlockParam); 9251699eca1SDaniel Jasper if (!NestedBlockSpecialCase) 9261699eca1SDaniel Jasper for (unsigned i = 0, e = State.Stack.size() - 1; i != e; ++i) 927de0328aeSDaniel Jasper State.Stack[i].BreakBeforeParameter = true; 928b16b969dSDaniel Jasper 9299e5ede0fSDaniel Jasper if (PreviousNonComment && 930a6b6d51bSFrancois Ferrand !PreviousNonComment->isOneOf(tok::comma, tok::colon, tok::semi) && 931a98b7b01SDaniel Jasper (PreviousNonComment->isNot(TT_TemplateCloser) || 9326c0ee17bSDaniel Jasper Current.NestingLevel != 0) && 93347bbda09SDaniel Jasper !PreviousNonComment->isOneOf( 93447bbda09SDaniel Jasper TT_BinaryOperator, TT_FunctionAnnotationRParen, TT_JavaAnnotation, 935a98b7b01SDaniel Jasper TT_LeadingJavaAnnotation) && 936a98b7b01SDaniel Jasper Current.isNot(TT_BinaryOperator) && !PreviousNonComment->opensScope()) 937de0328aeSDaniel Jasper State.Stack.back().BreakBeforeParameter = true; 938de0328aeSDaniel Jasper 9391db6c386SDaniel Jasper // If we break after { or the [ of an array initializer, we should also break 9401db6c386SDaniel Jasper // before the corresponding } or ]. 94190818057SDaniel Jasper if (PreviousNonComment && 94298e0b12dSDaniel Jasper (PreviousNonComment->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) || 943c10d97f9SMartin Probst opensProtoMessageField(*PreviousNonComment, Style))) 944de0328aeSDaniel Jasper State.Stack.back().BreakBeforeClosingBrace = true; 945de0328aeSDaniel Jasper 946de0328aeSDaniel Jasper if (State.Stack.back().AvoidBinPacking) { 947c6deae45SPaul Hoad // If we are breaking after '(', '{', '<', or this is the break after a ':' 948c6deae45SPaul Hoad // to start a member initializater list in a constructor, this should not 949c6deae45SPaul Hoad // be considered bin packing unless the relevant AllowAll option is false or 950c6deae45SPaul Hoad // this is a dict/object literal. 951c6deae45SPaul Hoad bool PreviousIsBreakingCtorInitializerColon = 952c6deae45SPaul Hoad Previous.is(TT_CtorInitializerColon) && 953c6deae45SPaul Hoad Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon; 954c6deae45SPaul Hoad if (!(Previous.isOneOf(tok::l_paren, tok::l_brace, TT_BinaryOperator) || 955c6deae45SPaul Hoad PreviousIsBreakingCtorInitializerColon) || 956de0328aeSDaniel Jasper (!Style.AllowAllParametersOfDeclarationOnNextLine && 9572a958321SDaniel Jasper State.Line->MustBeDeclaration) || 958c6deae45SPaul Hoad (!Style.AllowAllArgumentsOnNextLine && 959c6deae45SPaul Hoad !State.Line->MustBeDeclaration) || 9608a780a2fSowenca (Style.PackConstructorInitializers != FormatStyle::PCIS_NextLine && 961c6deae45SPaul Hoad PreviousIsBreakingCtorInitializerColon) || 962a98b7b01SDaniel Jasper Previous.is(TT_DictLiteral)) 963de0328aeSDaniel Jasper State.Stack.back().BreakBeforeParameter = true; 964c6deae45SPaul Hoad 965c6deae45SPaul Hoad // If we are breaking after a ':' to start a member initializer list, 966c6deae45SPaul Hoad // and we allow all arguments on the next line, we should not break 967c6deae45SPaul Hoad // before the next parameter. 968c6deae45SPaul Hoad if (PreviousIsBreakingCtorInitializerColon && 9698a780a2fSowenca Style.PackConstructorInitializers == FormatStyle::PCIS_NextLine) 970c6deae45SPaul Hoad State.Stack.back().BreakBeforeParameter = false; 971de0328aeSDaniel Jasper } 972de0328aeSDaniel Jasper 9731f80396dSAlexander Kornienko return Penalty; 974de0328aeSDaniel Jasper } 975de0328aeSDaniel Jasper 9769f388d04SDaniel Jasper unsigned ContinuationIndenter::getNewLineColumn(const LineState &State) { 9775d2587daSDaniel Jasper if (!State.NextToken || !State.NextToken->Previous) 9785d2587daSDaniel Jasper return 0; 97978e2a3c6SJonathan Coe 9809f388d04SDaniel Jasper FormatToken &Current = *State.NextToken; 98178e2a3c6SJonathan Coe 98278e2a3c6SJonathan Coe if (State.Stack.back().IsCSharpGenericTypeConstraint && 98378e2a3c6SJonathan Coe Current.isNot(TT_CSharpGenericTypeConstraint)) 98478e2a3c6SJonathan Coe return State.Stack.back().ColonPos + 2; 98578e2a3c6SJonathan Coe 9864281c5aeSDaniel Jasper const FormatToken &Previous = *Current.Previous; 9879f388d04SDaniel Jasper // If we are continuing an expression, we want to use the continuation indent. 9889f388d04SDaniel Jasper unsigned ContinuationIndent = 9899f388d04SDaniel Jasper std::max(State.Stack.back().LastSpace, State.Stack.back().Indent) + 9909f388d04SDaniel Jasper Style.ContinuationIndentWidth; 9919f388d04SDaniel Jasper const FormatToken *PreviousNonComment = Current.getPreviousNonComment(); 9929f388d04SDaniel Jasper const FormatToken *NextNonComment = Previous.getNextNonComment(); 9939f388d04SDaniel Jasper if (!NextNonComment) 9949f388d04SDaniel Jasper NextNonComment = &Current; 99550b4bd7cSDaniel Jasper 99650b4bd7cSDaniel Jasper // Java specific bits. 997d0ec0d62SDaniel Jasper if (Style.Language == FormatStyle::LK_Java && 998d0ec0d62SDaniel Jasper Current.isOneOf(Keywords.kw_implements, Keywords.kw_extends)) 99950b4bd7cSDaniel Jasper return std::max(State.Stack.back().LastSpace, 100050b4bd7cSDaniel Jasper State.Stack.back().Indent + Style.ContinuationIndentWidth); 100150b4bd7cSDaniel Jasper 1002a506ed25SPaul Hoad if (Style.BreakBeforeBraces == FormatStyle::BS_Whitesmiths && 1003a506ed25SPaul Hoad State.Line->First->is(tok::kw_enum)) 1004a506ed25SPaul Hoad return (Style.IndentWidth * State.Line->First->IndentLevel) + 1005a506ed25SPaul Hoad Style.IndentWidth; 1006a506ed25SPaul Hoad 1007f5acd11dSBruno Ricci if (NextNonComment->is(tok::l_brace) && NextNonComment->is(BK_Block)) 100805cd5868SDaniel Jasper return Current.NestingLevel == 0 ? State.FirstIndent 10099f388d04SDaniel Jasper : State.Stack.back().Indent; 1010ff747be4SKrasimir Georgiev if ((Current.isOneOf(tok::r_brace, tok::r_square) || 101126b144ccSKrasimir Georgiev (Current.is(tok::greater) && 101226b144ccSKrasimir Georgiev (Style.Language == FormatStyle::LK_Proto || 101326b144ccSKrasimir Georgiev Style.Language == FormatStyle::LK_TextProto))) && 1014ff747be4SKrasimir Georgiev State.Stack.size() > 1) { 1015bd73bcf8SDaniel Jasper if (Current.closesBlockOrBlockTypeList(Style)) 101611a0ac66SDaniel Jasper return State.Stack[State.Stack.size() - 2].NestedBlockIndent; 1017f5acd11dSBruno Ricci if (Current.MatchingParen && Current.MatchingParen->is(BK_BracedInit)) 10189f388d04SDaniel Jasper return State.Stack[State.Stack.size() - 2].LastSpace; 10199f388d04SDaniel Jasper return State.FirstIndent; 10209f388d04SDaniel Jasper } 1021e4f95e8eSOwen Pan // Indent a closing parenthesis at the previous level if followed by a semi, 1022e4f95e8eSOwen Pan // const, or opening brace. This allows indentations such as: 1023b2f06eaeSMartin Probst // foo( 1024b2f06eaeSMartin Probst // a, 1025b2f06eaeSMartin Probst // ); 1026e4f95e8eSOwen Pan // int Foo::getter( 1027e4f95e8eSOwen Pan // // 1028e4f95e8eSOwen Pan // ) const { 1029e4f95e8eSOwen Pan // return foo; 1030e4f95e8eSOwen Pan // } 1031b2f06eaeSMartin Probst // function foo( 1032b2f06eaeSMartin Probst // a, 1033b2f06eaeSMartin Probst // ) { 1034b2f06eaeSMartin Probst // code(); // 1035b2f06eaeSMartin Probst // } 1036b2f06eaeSMartin Probst if (Current.is(tok::r_paren) && State.Stack.size() > 1 && 1037e4f95e8eSOwen Pan (!Current.Next || 1038e4f95e8eSOwen Pan Current.Next->isOneOf(tok::semi, tok::kw_const, tok::l_brace))) 10392c1cdae2SMartin Probst return State.Stack[State.Stack.size() - 2].LastSpace; 104098e0b12dSDaniel Jasper if (NextNonComment->is(TT_TemplateString) && NextNonComment->closesScope()) 104198e0b12dSDaniel Jasper return State.Stack[State.Stack.size() - 2].LastSpace; 1042783bac6bSDaniel Jasper if (Current.is(tok::identifier) && Current.Next && 1043ddb19248SKrasimir Georgiev (Current.Next->is(TT_DictLiteral) || 1044ddb19248SKrasimir Georgiev ((Style.Language == FormatStyle::LK_Proto || 1045ddb19248SKrasimir Georgiev Style.Language == FormatStyle::LK_TextProto) && 1046a79d62d2SKrasimir Georgiev Current.Next->isOneOf(tok::less, tok::l_brace)))) 1047783bac6bSDaniel Jasper return State.Stack.back().Indent; 10480928553eSDaniel Jasper if (NextNonComment->is(TT_ObjCStringLiteral) && 10490928553eSDaniel Jasper State.StartOfStringLiteral != 0) 10500928553eSDaniel Jasper return State.StartOfStringLiteral - 1; 1051d4fa2e63SAlexander Kornienko if (NextNonComment->isStringLiteral() && State.StartOfStringLiteral != 0) 1052d4fa2e63SAlexander Kornienko return State.StartOfStringLiteral; 10539f388d04SDaniel Jasper if (NextNonComment->is(tok::lessless) && 10549f388d04SDaniel Jasper State.Stack.back().FirstLessLess != 0) 10559f388d04SDaniel Jasper return State.Stack.back().FirstLessLess; 10569f388d04SDaniel Jasper if (NextNonComment->isMemberAccess()) { 105724a14777SDaniel Jasper if (State.Stack.back().CallContinuation == 0) 10589f388d04SDaniel Jasper return ContinuationIndent; 10599f388d04SDaniel Jasper return State.Stack.back().CallContinuation; 10609f388d04SDaniel Jasper } 10619f388d04SDaniel Jasper if (State.Stack.back().QuestionColumn != 0 && 1062c0d606a5SDaniel Jasper ((NextNonComment->is(tok::colon) && 1063a98b7b01SDaniel Jasper NextNonComment->is(TT_ConditionalExpr)) || 10644db94094SFrancois Ferrand Previous.is(TT_ConditionalExpr))) { 10654db94094SFrancois Ferrand if (((NextNonComment->is(tok::colon) && NextNonComment->Next && 10664db94094SFrancois Ferrand !NextNonComment->Next->FakeLParens.empty() && 10674db94094SFrancois Ferrand NextNonComment->Next->FakeLParens.back() == prec::Conditional) || 10684db94094SFrancois Ferrand (Previous.is(tok::colon) && !Current.FakeLParens.empty() && 10694db94094SFrancois Ferrand Current.FakeLParens.back() == prec::Conditional)) && 10704db94094SFrancois Ferrand !State.Stack.back().IsWrappedConditional) { 10714db94094SFrancois Ferrand // NOTE: we may tweak this slightly: 10724db94094SFrancois Ferrand // * not remove the 'lead' ContinuationIndentWidth 10734db94094SFrancois Ferrand // * always un-indent by the operator when 10744db94094SFrancois Ferrand // BreakBeforeTernaryOperators=true 10750fad648bSKrasimir Georgiev unsigned Indent = State.Stack.back().Indent; 10760fad648bSKrasimir Georgiev if (Style.AlignOperands != FormatStyle::OAS_DontAlign) { 10770fad648bSKrasimir Georgiev Indent -= Style.ContinuationIndentWidth; 10780fad648bSKrasimir Georgiev } 10790ee04e6eSFrancois Ferrand if (Style.BreakBeforeTernaryOperators && 10800ee04e6eSFrancois Ferrand State.Stack.back().UnindentOperator) 10810ee04e6eSFrancois Ferrand Indent -= 2; 10824db94094SFrancois Ferrand return Indent; 10834db94094SFrancois Ferrand } 10849f388d04SDaniel Jasper return State.Stack.back().QuestionColumn; 10854db94094SFrancois Ferrand } 10869f388d04SDaniel Jasper if (Previous.is(tok::comma) && State.Stack.back().VariablePos != 0) 10879f388d04SDaniel Jasper return State.Stack.back().VariablePos; 1088e9ab42dfSDaniel Jasper if ((PreviousNonComment && 1089e9ab42dfSDaniel Jasper (PreviousNonComment->ClosesTemplateDeclaration || 109047bbda09SDaniel Jasper PreviousNonComment->isOneOf( 1091b060ad87SBen Hamilton TT_AttributeParen, TT_AttributeSquare, TT_FunctionAnnotationRParen, 1092b060ad87SBen Hamilton TT_JavaAnnotation, TT_LeadingJavaAnnotation))) || 1093c75e1effSDaniel Jasper (!Style.IndentWrappedFunctionNames && 1094a98b7b01SDaniel Jasper NextNonComment->isOneOf(tok::kw_operator, TT_FunctionDeclarationName))) 10959f388d04SDaniel Jasper return std::max(State.Stack.back().LastSpace, State.Stack.back().Indent); 1096a98b7b01SDaniel Jasper if (NextNonComment->is(TT_SelectorName)) { 10979f388d04SDaniel Jasper if (!State.Stack.back().ObjCSelectorNameFound) { 10984dc1cdc5SBen Hamilton unsigned MinIndent = State.Stack.back().Indent; 1099416348efSBen Hamilton if (shouldIndentWrappedSelectorName(Style, State.Line->Type)) 11004dc1cdc5SBen Hamilton MinIndent = std::max(MinIndent, 11014dc1cdc5SBen Hamilton State.FirstIndent + Style.ContinuationIndentWidth); 11024dc1cdc5SBen Hamilton // If LongestObjCSelectorName is 0, we are indenting the first 11034dc1cdc5SBen Hamilton // part of an ObjC selector (or a selector component which is 11044dc1cdc5SBen Hamilton // not colon-aligned due to block formatting). 11054dc1cdc5SBen Hamilton // 11064dc1cdc5SBen Hamilton // Otherwise, we are indenting a subsequent part of an ObjC 11074dc1cdc5SBen Hamilton // selector which should be colon-aligned to the longest 11084dc1cdc5SBen Hamilton // component of the ObjC selector. 11094dc1cdc5SBen Hamilton // 11104dc1cdc5SBen Hamilton // In either case, we want to respect Style.IndentWrappedFunctionNames. 11114dc1cdc5SBen Hamilton return MinIndent + 111238d80134SFrancois Ferrand std::max(NextNonComment->LongestObjCSelectorName, 111338d80134SFrancois Ferrand NextNonComment->ColumnWidth) - 11149f388d04SDaniel Jasper NextNonComment->ColumnWidth; 11159f388d04SDaniel Jasper } 111624a14777SDaniel Jasper if (!State.Stack.back().AlignColons) 11179f388d04SDaniel Jasper return State.Stack.back().Indent; 111824a14777SDaniel Jasper if (State.Stack.back().ColonPos > NextNonComment->ColumnWidth) 11199f388d04SDaniel Jasper return State.Stack.back().ColonPos - NextNonComment->ColumnWidth; 11209f388d04SDaniel Jasper return State.Stack.back().Indent; 11219f388d04SDaniel Jasper } 1122fd36f0b5SDaniel Jasper if (NextNonComment->is(tok::colon) && NextNonComment->is(TT_ObjCMethodExpr)) 1123fd36f0b5SDaniel Jasper return State.Stack.back().ColonPos; 1124a98b7b01SDaniel Jasper if (NextNonComment->is(TT_ArraySubscriptLSquare)) { 11259f388d04SDaniel Jasper if (State.Stack.back().StartOfArraySubscripts != 0) 11269f388d04SDaniel Jasper return State.Stack.back().StartOfArraySubscripts; 11270c28a093SJonathan Coe else if (Style.isCSharp()) // C# allows `["key"] = value` inside object 11280c28a093SJonathan Coe // initializers. 11290c28a093SJonathan Coe return State.Stack.back().Indent; 11309f388d04SDaniel Jasper return ContinuationIndent; 11319f388d04SDaniel Jasper } 11329c95013eSDaniel Jasper 11339c95013eSDaniel Jasper // This ensure that we correctly format ObjC methods calls without inputs, 11349c95013eSDaniel Jasper // i.e. where the last element isn't selector like: [callee method]; 11359c95013eSDaniel Jasper if (NextNonComment->is(tok::identifier) && NextNonComment->FakeRParens == 0 && 11369c95013eSDaniel Jasper NextNonComment->Next && NextNonComment->Next->is(TT_ObjCMethodExpr)) 1137eb53668cSDaniel Jasper return State.Stack.back().Indent; 11389c95013eSDaniel Jasper 1139b754a747SDaniel Jasper if (NextNonComment->isOneOf(TT_StartOfName, TT_PointerOrReference) || 1140b2328b1eSDaniel Jasper Previous.isOneOf(tok::coloncolon, tok::equal, TT_JsTypeColon)) 11419f388d04SDaniel Jasper return ContinuationIndent; 11429f388d04SDaniel Jasper if (PreviousNonComment && PreviousNonComment->is(tok::colon) && 1143a98b7b01SDaniel Jasper PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) 11449f388d04SDaniel Jasper return ContinuationIndent; 1145a98b7b01SDaniel Jasper if (NextNonComment->is(TT_CtorInitializerComma)) 11469f388d04SDaniel Jasper return State.Stack.back().Indent; 1147a6b6d51bSFrancois Ferrand if (PreviousNonComment && PreviousNonComment->is(TT_CtorInitializerColon) && 1148a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon) 1149a6b6d51bSFrancois Ferrand return State.Stack.back().Indent; 11506bb103f9SFrancois Ferrand if (PreviousNonComment && PreviousNonComment->is(TT_InheritanceColon) && 11516bb103f9SFrancois Ferrand Style.BreakInheritanceList == FormatStyle::BILS_AfterColon) 11526bb103f9SFrancois Ferrand return State.Stack.back().Indent; 11530ef8ee19SAndi-Bogdan Postelnicu if (NextNonComment->isOneOf(TT_CtorInitializerColon, TT_InheritanceColon, 11540ef8ee19SAndi-Bogdan Postelnicu TT_InheritanceComma)) 11550ef8ee19SAndi-Bogdan Postelnicu return State.FirstIndent + Style.ConstructorInitializerIndentWidth; 1156316ab38eSDaniel Jasper if (Previous.is(tok::r_paren) && !Current.isBinaryOperator() && 1157119ff533SDaniel Jasper !Current.isOneOf(tok::colon, tok::comment)) 1158316ab38eSDaniel Jasper return ContinuationIndent; 11594e290648SKrasimir Georgiev if (Current.is(TT_ProtoExtensionLSquare)) 11604e290648SKrasimir Georgiev return State.Stack.back().Indent; 11610ee04e6eSFrancois Ferrand if (Current.isBinaryOperator() && State.Stack.back().UnindentOperator) 11620ee04e6eSFrancois Ferrand return State.Stack.back().Indent - Current.Tok.getLength() - 11630ee04e6eSFrancois Ferrand Current.SpacesRequiredBefore; 11640ee04e6eSFrancois Ferrand if (Current.isOneOf(tok::comment, TT_BlockComment, TT_LineComment) && 11650ee04e6eSFrancois Ferrand NextNonComment->isBinaryOperator() && State.Stack.back().UnindentOperator) 11660ee04e6eSFrancois Ferrand return State.Stack.back().Indent - NextNonComment->Tok.getLength() - 11670ee04e6eSFrancois Ferrand NextNonComment->SpacesRequiredBefore; 11685d2587daSDaniel Jasper if (State.Stack.back().Indent == State.FirstIndent && PreviousNonComment && 1169fca07890SOwen Pan !PreviousNonComment->isOneOf(tok::r_brace, TT_CtorInitializerComma)) 11709f388d04SDaniel Jasper // Ensure that we fall back to the continuation indent width instead of 11719f388d04SDaniel Jasper // just flushing continuations left. 11729f388d04SDaniel Jasper return State.Stack.back().Indent + Style.ContinuationIndentWidth; 11739f388d04SDaniel Jasper return State.Stack.back().Indent; 11749f388d04SDaniel Jasper } 11759f388d04SDaniel Jasper 1176fa0118e6SWawha static bool hasNestedBlockInlined(const FormatToken *Previous, 1177fa0118e6SWawha const FormatToken &Current, 1178fa0118e6SWawha const FormatStyle &Style) { 1179fa0118e6SWawha if (Previous->isNot(tok::l_paren)) 1180fa0118e6SWawha return true; 1181fa0118e6SWawha if (Previous->ParameterCount > 1) 1182fa0118e6SWawha return true; 1183fa0118e6SWawha 1184fa0118e6SWawha // Also a nested block if contains a lambda inside function with 1 parameter 1185fa0118e6SWawha return (Style.BraceWrapping.BeforeLambdaBody && Current.is(TT_LambdaLSquare)); 1186fa0118e6SWawha } 1187fa0118e6SWawha 1188de0328aeSDaniel Jasper unsigned ContinuationIndenter::moveStateToNextToken(LineState &State, 1189de0328aeSDaniel Jasper bool DryRun, bool Newline) { 1190de0328aeSDaniel Jasper assert(State.Stack.size()); 119160553be5SDaniel Jasper const FormatToken &Current = *State.NextToken; 1192de0328aeSDaniel Jasper 119378e2a3c6SJonathan Coe if (Current.is(TT_CSharpGenericTypeConstraint)) 119478e2a3c6SJonathan Coe State.Stack.back().IsCSharpGenericTypeConstraint = true; 1195240527caSDaniel Jasper if (Current.isOneOf(tok::comma, TT_BinaryOperator)) 1196240527caSDaniel Jasper State.Stack.back().NoLineBreakInOperand = false; 119778e2a3c6SJonathan Coe if (Current.isOneOf(TT_InheritanceColon, TT_CSharpGenericTypeConstraintColon)) 1198de0328aeSDaniel Jasper State.Stack.back().AvoidBinPacking = true; 1199a98b7b01SDaniel Jasper if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator)) { 1200c0d606a5SDaniel Jasper if (State.Stack.back().FirstLessLess == 0) 1201de0328aeSDaniel Jasper State.Stack.back().FirstLessLess = State.Column; 1202c0d606a5SDaniel Jasper else 1203c0d606a5SDaniel Jasper State.Stack.back().LastOperatorWrapped = Newline; 1204c0d606a5SDaniel Jasper } 1205240527caSDaniel Jasper if (Current.is(TT_BinaryOperator) && Current.isNot(tok::lessless)) 1206240527caSDaniel Jasper State.Stack.back().LastOperatorWrapped = Newline; 1207240527caSDaniel Jasper if (Current.is(TT_ConditionalExpr) && Current.Previous && 1208240527caSDaniel Jasper !Current.Previous->is(TT_ConditionalExpr)) 1209c0d606a5SDaniel Jasper State.Stack.back().LastOperatorWrapped = Newline; 1210a98b7b01SDaniel Jasper if (Current.is(TT_ArraySubscriptLSquare) && 1211de0328aeSDaniel Jasper State.Stack.back().StartOfArraySubscripts == 0) 1212de0328aeSDaniel Jasper State.Stack.back().StartOfArraySubscripts = State.Column; 12134db94094SFrancois Ferrand if (Current.is(TT_ConditionalExpr) && Current.is(tok::question) && 12144db94094SFrancois Ferrand ((Current.MustBreakBefore) || 12154db94094SFrancois Ferrand (Current.getNextNonComment() && 12164db94094SFrancois Ferrand Current.getNextNonComment()->MustBreakBefore))) 12174db94094SFrancois Ferrand State.Stack.back().IsWrappedConditional = true; 121845860facSDaniel Jasper if (Style.BreakBeforeTernaryOperators && Current.is(tok::question)) 1219de0328aeSDaniel Jasper State.Stack.back().QuestionColumn = State.Column; 122045860facSDaniel Jasper if (!Style.BreakBeforeTernaryOperators && Current.isNot(tok::colon)) { 122145860facSDaniel Jasper const FormatToken *Previous = Current.Previous; 122245860facSDaniel Jasper while (Previous && Previous->isTrailingComment()) 122345860facSDaniel Jasper Previous = Previous->Previous; 122445860facSDaniel Jasper if (Previous && Previous->is(tok::question)) 122545860facSDaniel Jasper State.Stack.back().QuestionColumn = State.Column; 122645860facSDaniel Jasper } 1227cab46171SDaniel Jasper if (!Current.opensScope() && !Current.closesScope() && 1228cab46171SDaniel Jasper !Current.is(TT_PointerOrReference)) 1229de0328aeSDaniel Jasper State.LowestLevelOnLine = 123005cd5868SDaniel Jasper std::min(State.LowestLevelOnLine, Current.NestingLevel); 12314c6e0059SDaniel Jasper if (Current.isMemberAccess()) 1232de0328aeSDaniel Jasper State.Stack.back().StartOfFunctionCall = 123300492f96SDaniel Jasper !Current.NextOperator ? 0 : State.Column; 1234f51ce777SJacek Olesiak if (Current.is(TT_SelectorName)) 1235b88b25feSDaniel Jasper State.Stack.back().ObjCSelectorNameFound = true; 1236a6b6d51bSFrancois Ferrand if (Current.is(TT_CtorInitializerColon) && 1237a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers != FormatStyle::BCIS_AfterColon) { 1238de0328aeSDaniel Jasper // Indent 2 from the column, so: 1239de0328aeSDaniel Jasper // SomeClass::SomeClass() 1240de0328aeSDaniel Jasper // : First(...), ... 1241de0328aeSDaniel Jasper // Next(...) 1242de0328aeSDaniel Jasper // ^ line up here. 1243de0328aeSDaniel Jasper State.Stack.back().Indent = 124489628f64SManuel Klimek State.Column + 124589628f64SManuel Klimek (Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma 124689628f64SManuel Klimek ? 0 124789628f64SManuel Klimek : 2); 1248d6a1cab1SDaniel Jasper State.Stack.back().NestedBlockIndent = State.Stack.back().Indent; 12498a780a2fSowenca if (Style.PackConstructorInitializers > FormatStyle::PCIS_BinPack) { 1250de0328aeSDaniel Jasper State.Stack.back().AvoidBinPacking = true; 1251c6deae45SPaul Hoad State.Stack.back().BreakBeforeParameter = 12528a780a2fSowenca Style.PackConstructorInitializers != FormatStyle::PCIS_NextLine; 1253c6deae45SPaul Hoad } else { 1254de0328aeSDaniel Jasper State.Stack.back().BreakBeforeParameter = false; 1255de0328aeSDaniel Jasper } 1256c6deae45SPaul Hoad } 1257a6b6d51bSFrancois Ferrand if (Current.is(TT_CtorInitializerColon) && 1258a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon) { 1259a6b6d51bSFrancois Ferrand State.Stack.back().Indent = 1260a6b6d51bSFrancois Ferrand State.FirstIndent + Style.ConstructorInitializerIndentWidth; 1261a6b6d51bSFrancois Ferrand State.Stack.back().NestedBlockIndent = State.Stack.back().Indent; 12628a780a2fSowenca if (Style.PackConstructorInitializers > FormatStyle::PCIS_BinPack) 1263a6b6d51bSFrancois Ferrand State.Stack.back().AvoidBinPacking = true; 1264a6b6d51bSFrancois Ferrand } 12650ef8ee19SAndi-Bogdan Postelnicu if (Current.is(TT_InheritanceColon)) 12660ef8ee19SAndi-Bogdan Postelnicu State.Stack.back().Indent = 12676bb103f9SFrancois Ferrand State.FirstIndent + Style.ConstructorInitializerIndentWidth; 1268c4144ea4SDaniel Jasper if (Current.isOneOf(TT_BinaryOperator, TT_ConditionalExpr) && Newline) 1269c4144ea4SDaniel Jasper State.Stack.back().NestedBlockIndent = 1270c4144ea4SDaniel Jasper State.Column + Current.ColumnWidth + 1; 1271d9b319e3SDaniel Jasper if (Current.isOneOf(TT_LambdaLSquare, TT_LambdaArrow)) 1272d9b319e3SDaniel Jasper State.Stack.back().LastSpace = State.Column; 1273de0328aeSDaniel Jasper 1274de0328aeSDaniel Jasper // Insert scopes created by fake parenthesis. 1275de0328aeSDaniel Jasper const FormatToken *Previous = Current.getPreviousNonComment(); 1276b16b969dSDaniel Jasper 1277b16b969dSDaniel Jasper // Add special behavior to support a format commonly used for JavaScript 1278b16b969dSDaniel Jasper // closures: 1279b16b969dSDaniel Jasper // SomeFunction(function() { 1280b16b969dSDaniel Jasper // foo(); 1281b16b969dSDaniel Jasper // bar(); 1282b16b969dSDaniel Jasper // }, a, b, c); 1283b2ad4d4cSDaniel Jasper if (Current.isNot(tok::comment) && Previous && 1284b2ad4d4cSDaniel Jasper Previous->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) && 12855528caceSKrasimir Georgiev !Previous->is(TT_DictLiteral) && State.Stack.size() > 1 && 12865528caceSKrasimir Georgiev !State.Stack.back().HasMultipleNestedBlocks) { 12871699eca1SDaniel Jasper if (State.Stack[State.Stack.size() - 2].NestedBlockInlined && Newline) 12881699eca1SDaniel Jasper for (unsigned i = 0, e = State.Stack.size() - 1; i != e; ++i) 1289b16b969dSDaniel Jasper State.Stack[i].NoLineBreak = true; 12904b444495SDaniel Jasper State.Stack[State.Stack.size() - 2].NestedBlockInlined = false; 1291b16b969dSDaniel Jasper } 1292d48d1f8eSBjörn Schäpers if (Previous && (Previous->isOneOf(TT_BinaryOperator, TT_ConditionalExpr) || 1293d48d1f8eSBjörn Schäpers (Previous->isOneOf(tok::l_paren, tok::comma, tok::colon) && 1294d48d1f8eSBjörn Schäpers !Previous->isOneOf(TT_DictLiteral, TT_ObjCMethodExpr)))) { 12954b444495SDaniel Jasper State.Stack.back().NestedBlockInlined = 1296fa0118e6SWawha !Newline && hasNestedBlockInlined(Previous, Current, Style); 1297b16b969dSDaniel Jasper } 1298b16b969dSDaniel Jasper 129960553be5SDaniel Jasper moveStatePastFakeLParens(State, Newline); 130060553be5SDaniel Jasper moveStatePastScopeCloser(State); 130145ab559eSManuel Klimek bool AllowBreak = !State.Stack.back().NoLineBreak && 1302bc05ebaaSKrasimir Georgiev !State.Stack.back().NoLineBreakInOperand; 1303c06f6da3SDaniel Jasper moveStatePastScopeOpener(State, Newline); 130460553be5SDaniel Jasper moveStatePastFakeRParens(State); 130560553be5SDaniel Jasper 13060928553eSDaniel Jasper if (Current.is(TT_ObjCStringLiteral) && State.StartOfStringLiteral == 0) 13070928553eSDaniel Jasper State.StartOfStringLiteral = State.Column + 1; 1308cbb726d0SPaul Hoad if (Current.is(TT_CSharpStringLiteral) && State.StartOfStringLiteral == 0) 1309cbb726d0SPaul Hoad State.StartOfStringLiteral = State.Column + 1; 1310d4fa2e63SAlexander Kornienko else if (Current.isStringLiteral() && State.StartOfStringLiteral == 0) 1311d4fa2e63SAlexander Kornienko State.StartOfStringLiteral = State.Column; 13120928553eSDaniel Jasper else if (!Current.isOneOf(tok::comment, tok::identifier, tok::hash) && 13130928553eSDaniel Jasper !Current.isStringLiteral()) 131460553be5SDaniel Jasper State.StartOfStringLiteral = 0; 131560553be5SDaniel Jasper 131660553be5SDaniel Jasper State.Column += Current.ColumnWidth; 131760553be5SDaniel Jasper State.NextToken = State.NextToken->Next; 131845ab559eSManuel Klimek 1319bda8482aSKrasimir Georgiev unsigned Penalty = 1320bda8482aSKrasimir Georgiev handleEndOfLine(Current, State, DryRun, AllowBreak, Newline); 132160553be5SDaniel Jasper 132260553be5SDaniel Jasper if (Current.Role) 132360553be5SDaniel Jasper Current.Role->formatFromToken(State, this, DryRun); 132460553be5SDaniel Jasper // If the previous has a special role, let it consume tokens as appropriate. 132560553be5SDaniel Jasper // It is necessary to start at the previous token for the only implemented 132660553be5SDaniel Jasper // role (comma separated list). That way, the decision whether or not to break 132760553be5SDaniel Jasper // after the "{" is already done and both options are tried and evaluated. 132860553be5SDaniel Jasper // FIXME: This is ugly, find a better way. 132960553be5SDaniel Jasper if (Previous && Previous->Role) 133060553be5SDaniel Jasper Penalty += Previous->Role->formatAfterToken(State, this, DryRun); 133160553be5SDaniel Jasper 133260553be5SDaniel Jasper return Penalty; 133360553be5SDaniel Jasper } 133460553be5SDaniel Jasper 133560553be5SDaniel Jasper void ContinuationIndenter::moveStatePastFakeLParens(LineState &State, 133660553be5SDaniel Jasper bool Newline) { 133760553be5SDaniel Jasper const FormatToken &Current = *State.NextToken; 13384041f16bSBjörn Schäpers if (Current.FakeLParens.empty()) 13394041f16bSBjörn Schäpers return; 13404041f16bSBjörn Schäpers 134160553be5SDaniel Jasper const FormatToken *Previous = Current.getPreviousNonComment(); 134260553be5SDaniel Jasper 1343de0328aeSDaniel Jasper // Don't add extra indentation for the first fake parenthesis after 13440db806b7SDinesh Dwivedi // 'return', assignments or opening <({[. The indentation for these cases 1345de0328aeSDaniel Jasper // is special cased. 1346de0328aeSDaniel Jasper bool SkipFirstExtraIndent = 134798f8ae34SDaniel Jasper (Previous && (Previous->opensScope() || 134898f8ae34SDaniel Jasper Previous->isOneOf(tok::semi, tok::kw_return) || 13493219e43cSDaniel Jasper (Previous->getPrecedence() == prec::Assignment && 13500ee04e6eSFrancois Ferrand Style.AlignOperands != FormatStyle::OAS_DontAlign) || 1351a98b7b01SDaniel Jasper Previous->is(TT_ObjCMethodExpr))); 135288fa4bfeSBjörn Schäpers for (const auto &PrecedenceLevel : llvm::reverse(Current.FakeLParens)) { 1353de0328aeSDaniel Jasper ParenState NewParenState = State.Stack.back(); 13548b4bc76aSKrasimir Georgiev NewParenState.Tok = nullptr; 1355de0328aeSDaniel Jasper NewParenState.ContainsLineBreak = false; 135604bbda99SDaniel Jasper NewParenState.LastOperatorWrapped = true; 13574db94094SFrancois Ferrand NewParenState.IsChainedConditional = false; 13584db94094SFrancois Ferrand NewParenState.IsWrappedConditional = false; 13590ee04e6eSFrancois Ferrand NewParenState.UnindentOperator = false; 1360240527caSDaniel Jasper NewParenState.NoLineBreak = 1361240527caSDaniel Jasper NewParenState.NoLineBreak || State.Stack.back().NoLineBreakInOperand; 1362eabede6dSDaniel Jasper 1363988e7e4fSDaniel Jasper // Don't propagate AvoidBinPacking into subexpressions of arg/param lists. 136488fa4bfeSBjörn Schäpers if (PrecedenceLevel > prec::Comma) 1365988e7e4fSDaniel Jasper NewParenState.AvoidBinPacking = false; 1366988e7e4fSDaniel Jasper 13673aa9a6a1SDaniel Jasper // Indent from 'LastSpace' unless these are fake parentheses encapsulating 13683aa9a6a1SDaniel Jasper // a builder type call after 'return' or, if the alignment after opening 13693aa9a6a1SDaniel Jasper // brackets is disabled. 13704281c5aeSDaniel Jasper if (!Current.isTrailingComment() && 13710ee04e6eSFrancois Ferrand (Style.AlignOperands != FormatStyle::OAS_DontAlign || 137288fa4bfeSBjörn Schäpers PrecedenceLevel < prec::Assignment) && 13736cab6784SDaniel Jasper (!Previous || Previous->isNot(tok::kw_return) || 137488fa4bfeSBjörn Schäpers (Style.Language != FormatStyle::LK_Java && PrecedenceLevel > 0)) && 13756501f7e8SDaniel Jasper (Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign || 137688fa4bfeSBjörn Schäpers PrecedenceLevel != prec::Comma || Current.NestingLevel == 0)) { 1377de0328aeSDaniel Jasper NewParenState.Indent = 1378de0328aeSDaniel Jasper std::max(std::max(State.Column, NewParenState.Indent), 1379de0328aeSDaniel Jasper State.Stack.back().LastSpace); 13800ee04e6eSFrancois Ferrand } 13810ee04e6eSFrancois Ferrand 1382b18c63e8SMaximilian Fickert if (Previous && 13830ee04e6eSFrancois Ferrand (Previous->getPrecedence() == prec::Assignment || 13840ee04e6eSFrancois Ferrand Previous->is(tok::kw_return) || 138588fa4bfeSBjörn Schäpers (PrecedenceLevel == prec::Conditional && Previous->is(tok::question) && 13860ee04e6eSFrancois Ferrand Previous->is(TT_ConditionalExpr))) && 1387b18c63e8SMaximilian Fickert !Newline) { 1388b18c63e8SMaximilian Fickert // If BreakBeforeBinaryOperators is set, un-indent a bit to account for 1389b18c63e8SMaximilian Fickert // the operator and keep the operands aligned 1390b18c63e8SMaximilian Fickert if (Style.AlignOperands == FormatStyle::OAS_AlignAfterOperator) 13910ee04e6eSFrancois Ferrand NewParenState.UnindentOperator = true; 1392b18c63e8SMaximilian Fickert // Mark indentation as alignment if the expression is aligned. 1393b18c63e8SMaximilian Fickert if (Style.AlignOperands != FormatStyle::OAS_DontAlign) 1394b18c63e8SMaximilian Fickert NewParenState.IsAligned = true; 1395b18c63e8SMaximilian Fickert } 1396eabede6dSDaniel Jasper 13979fe0e8daSDaniel Jasper // Do not indent relative to the fake parentheses inserted for "." or "->". 13989fe0e8daSDaniel Jasper // This is a special case to make the following to statements consistent: 13999fe0e8daSDaniel Jasper // OuterFunction(InnerFunctionCall( // break 14009fe0e8daSDaniel Jasper // ParameterToInnerFunction)); 14019fe0e8daSDaniel Jasper // OuterFunction(SomeObject.InnerFunctionCall( // break 14029fe0e8daSDaniel Jasper // ParameterToInnerFunction)); 140388fa4bfeSBjörn Schäpers if (PrecedenceLevel > prec::Unknown) 14049fe0e8daSDaniel Jasper NewParenState.LastSpace = std::max(NewParenState.LastSpace, State.Column); 140588fa4bfeSBjörn Schäpers if (PrecedenceLevel != prec::Conditional && !Current.is(TT_UnaryOperator) && 14062a9f7201SDaniel Jasper Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign) 140796964359SDaniel Jasper NewParenState.StartOfFunctionCall = State.Column; 1408de0328aeSDaniel Jasper 14094db94094SFrancois Ferrand // Indent conditional expressions, unless they are chained "else-if" 14104db94094SFrancois Ferrand // conditionals. Never indent expression where the 'operator' is ',', ';' or 14114db94094SFrancois Ferrand // an assignment (i.e. *I <= prec::Assignment) as those have different 14124db94094SFrancois Ferrand // indentation rules. Indent other expression, unless the indentation needs 14134db94094SFrancois Ferrand // to be skipped. 141488fa4bfeSBjörn Schäpers if (PrecedenceLevel == prec::Conditional && Previous && 141588fa4bfeSBjörn Schäpers Previous->is(tok::colon) && Previous->is(TT_ConditionalExpr) && 141688fa4bfeSBjörn Schäpers &PrecedenceLevel == &Current.FakeLParens.back() && 14174db94094SFrancois Ferrand !State.Stack.back().IsWrappedConditional) { 14184db94094SFrancois Ferrand NewParenState.IsChainedConditional = true; 14190ee04e6eSFrancois Ferrand NewParenState.UnindentOperator = State.Stack.back().UnindentOperator; 142088fa4bfeSBjörn Schäpers } else if (PrecedenceLevel == prec::Conditional || 142188fa4bfeSBjörn Schäpers (!SkipFirstExtraIndent && PrecedenceLevel > prec::Assignment && 14224db94094SFrancois Ferrand !Current.isTrailingComment())) { 14236633ab8aSDaniel Jasper NewParenState.Indent += Style.ContinuationIndentWidth; 14244db94094SFrancois Ferrand } 142588fa4bfeSBjörn Schäpers if ((Previous && !Previous->opensScope()) || PrecedenceLevel != prec::Comma) 1426de0328aeSDaniel Jasper NewParenState.BreakBeforeParameter = false; 1427de0328aeSDaniel Jasper State.Stack.push_back(NewParenState); 1428de0328aeSDaniel Jasper SkipFirstExtraIndent = false; 1429de0328aeSDaniel Jasper } 143060553be5SDaniel Jasper } 1431de0328aeSDaniel Jasper 143211a0ac66SDaniel Jasper void ContinuationIndenter::moveStatePastFakeRParens(LineState &State) { 143311a0ac66SDaniel Jasper for (unsigned i = 0, e = State.NextToken->FakeRParens; i != e; ++i) { 143460553be5SDaniel Jasper unsigned VariablePos = State.Stack.back().VariablePos; 1435819788daSManuel Klimek if (State.Stack.size() == 1) { 1436819788daSManuel Klimek // Do not pop the last element. 1437819788daSManuel Klimek break; 1438819788daSManuel Klimek } 14399fe0e8daSDaniel Jasper State.Stack.pop_back(); 144060553be5SDaniel Jasper State.Stack.back().VariablePos = VariablePos; 1441819788daSManuel Klimek } 144260553be5SDaniel Jasper } 1443335ff266SDaniel Jasper 144460553be5SDaniel Jasper void ContinuationIndenter::moveStatePastScopeOpener(LineState &State, 144560553be5SDaniel Jasper bool Newline) { 144660553be5SDaniel Jasper const FormatToken &Current = *State.NextToken; 144760553be5SDaniel Jasper if (!Current.opensScope()) 144860553be5SDaniel Jasper return; 144960553be5SDaniel Jasper 145078e2a3c6SJonathan Coe // Don't allow '<' or '(' in C# generic type constraints to start new scopes. 145178e2a3c6SJonathan Coe if (Current.isOneOf(tok::less, tok::l_paren) && 145278e2a3c6SJonathan Coe State.Stack.back().IsCSharpGenericTypeConstraint) 145378e2a3c6SJonathan Coe return; 145478e2a3c6SJonathan Coe 1455f5acd11dSBruno Ricci if (Current.MatchingParen && Current.is(BK_Block)) { 145660553be5SDaniel Jasper moveStateToNewBlock(State); 145760553be5SDaniel Jasper return; 145860553be5SDaniel Jasper } 145960553be5SDaniel Jasper 146060553be5SDaniel Jasper unsigned NewIndent; 1461de7ca75cSDaniel Jasper unsigned LastSpace = State.Stack.back().LastSpace; 146260553be5SDaniel Jasper bool AvoidBinPacking; 146360553be5SDaniel Jasper bool BreakBeforeParameter = false; 1464ea40cee2SDaniel Jasper unsigned NestedBlockIndent = std::max(State.Stack.back().StartOfFunctionCall, 1465ea40cee2SDaniel Jasper State.Stack.back().NestedBlockIndent); 1466ff747be4SKrasimir Georgiev if (Current.isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) || 146726b144ccSKrasimir Georgiev opensProtoMessageField(Current, Style)) { 1468bd73bcf8SDaniel Jasper if (Current.opensBlockOrBlockTypeList(Style)) { 146938423276SMartin Probst NewIndent = Style.IndentWidth + 147038423276SMartin Probst std::min(State.Column, State.Stack.back().NestedBlockIndent); 1471b8f61682SDaniel Jasper } else { 147211a0ac66SDaniel Jasper NewIndent = State.Stack.back().LastSpace + Style.ContinuationIndentWidth; 1473e2e0387fSAlexander Kornienko } 1474de0328aeSDaniel Jasper const FormatToken *NextNoComment = Current.getNextNonComment(); 147550780ce1SDaniel Jasper bool EndsInComma = Current.MatchingParen && 147650780ce1SDaniel Jasper Current.MatchingParen->Previous && 147750780ce1SDaniel Jasper Current.MatchingParen->Previous->is(tok::comma); 147889628f64SManuel Klimek AvoidBinPacking = EndsInComma || Current.is(TT_DictLiteral) || 147926b144ccSKrasimir Georgiev Style.Language == FormatStyle::LK_Proto || 148026b144ccSKrasimir Georgiev Style.Language == FormatStyle::LK_TextProto || 148126b144ccSKrasimir Georgiev !Style.BinPackArguments || 14825f07f443SFrancois Ferrand (NextNoComment && 14835f07f443SFrancois Ferrand NextNoComment->isOneOf(TT_DesignatedInitializerPeriod, 14845f07f443SFrancois Ferrand TT_DesignatedInitializerLSquare)); 1485d2130f51SFrancois Ferrand BreakBeforeParameter = EndsInComma; 1486ea40cee2SDaniel Jasper if (Current.ParameterCount > 1) 1487ea40cee2SDaniel Jasper NestedBlockIndent = std::max(NestedBlockIndent, State.Column + 1); 1488de0328aeSDaniel Jasper } else { 14896633ab8aSDaniel Jasper NewIndent = Style.ContinuationIndentWidth + 14906633ab8aSDaniel Jasper std::max(State.Stack.back().LastSpace, 14919fe0e8daSDaniel Jasper State.Stack.back().StartOfFunctionCall); 1492de7ca75cSDaniel Jasper 1493de7ca75cSDaniel Jasper // Ensure that different different brackets force relative alignment, e.g.: 1494de7ca75cSDaniel Jasper // void SomeFunction(vector< // break 1495de7ca75cSDaniel Jasper // int> v); 1496de7ca75cSDaniel Jasper // FIXME: We likely want to do this for more combinations of brackets. 14973f11941dSDaniel Jasper if (Current.is(tok::less) && Current.ParentBracket == tok::l_paren) { 1498de7ca75cSDaniel Jasper NewIndent = std::max(NewIndent, State.Stack.back().Indent); 1499de7ca75cSDaniel Jasper LastSpace = std::max(LastSpace, State.Stack.back().Indent); 1500de7ca75cSDaniel Jasper } 1501de7ca75cSDaniel Jasper 15022c1cdae2SMartin Probst bool EndsInComma = 15032c1cdae2SMartin Probst Current.MatchingParen && 15042c1cdae2SMartin Probst Current.MatchingParen->getPreviousNonComment() && 15052c1cdae2SMartin Probst Current.MatchingParen->getPreviousNonComment()->is(tok::comma); 15062c1cdae2SMartin Probst 15074dc658caSBen Hamilton // If ObjCBinPackProtocolList is unspecified, fall back to BinPackParameters 15084dc658caSBen Hamilton // for backwards compatibility. 15094dc658caSBen Hamilton bool ObjCBinPackProtocolList = 15104dc658caSBen Hamilton (Style.ObjCBinPackProtocolList == FormatStyle::BPS_Auto && 15114dc658caSBen Hamilton Style.BinPackParameters) || 15124dc658caSBen Hamilton Style.ObjCBinPackProtocolList == FormatStyle::BPS_Always; 15134dc658caSBen Hamilton 15144dc658caSBen Hamilton bool BinPackDeclaration = 15154dc658caSBen Hamilton (State.Line->Type != LT_ObjCDecl && Style.BinPackParameters) || 15164dc658caSBen Hamilton (State.Line->Type == LT_ObjCDecl && ObjCBinPackProtocolList); 15174dc658caSBen Hamilton 151818210d7dSDaniel Jasper AvoidBinPacking = 151978e2a3c6SJonathan Coe (State.Stack.back().IsCSharpGenericTypeConstraint) || 1520142e79b8Smydeveloperday (Style.isJavaScript() && EndsInComma) || 15214dc658caSBen Hamilton (State.Line->MustBeDeclaration && !BinPackDeclaration) || 152218210d7dSDaniel Jasper (!State.Line->MustBeDeclaration && !Style.BinPackArguments) || 1523de0328aeSDaniel Jasper (Style.ExperimentalAutoDetectBinPacking && 1524f5acd11dSBruno Ricci (Current.is(PPK_OnePerLine) || 1525f5acd11dSBruno Ricci (!BinPackInconclusiveFunctions && Current.is(PPK_Inconclusive)))); 15262c1cdae2SMartin Probst 152770c98671Smydeveloperday if (Current.is(TT_ObjCMethodExpr) && Current.MatchingParen && 152870c98671Smydeveloperday Style.ObjCBreakBeforeNestedBlockParam) { 1529289afc07SDaniel Jasper if (Style.ColumnLimit) { 15301db6c386SDaniel Jasper // If this '[' opens an ObjC call, determine whether all parameters fit 15311db6c386SDaniel Jasper // into one line and put one per line if they don't. 15328b4bc76aSKrasimir Georgiev if (getLengthToMatchingParen(Current, State.Stack) + State.Column > 15331db6c386SDaniel Jasper getColumnLimit(State)) 15341db6c386SDaniel Jasper BreakBeforeParameter = true; 1535289afc07SDaniel Jasper } else { 1536289afc07SDaniel Jasper // For ColumnLimit = 0, we have to figure out whether there is or has to 1537289afc07SDaniel Jasper // be a line break within this call. 1538289afc07SDaniel Jasper for (const FormatToken *Tok = &Current; 1539289afc07SDaniel Jasper Tok && Tok != Current.MatchingParen; Tok = Tok->Next) { 1540289afc07SDaniel Jasper if (Tok->MustBreakBefore || 1541289afc07SDaniel Jasper (Tok->CanBreakBefore && Tok->NewlinesBefore > 0)) { 1542289afc07SDaniel Jasper BreakBeforeParameter = true; 1543289afc07SDaniel Jasper break; 1544289afc07SDaniel Jasper } 1545289afc07SDaniel Jasper } 1546289afc07SDaniel Jasper } 1547289afc07SDaniel Jasper } 15482c1cdae2SMartin Probst 1549142e79b8Smydeveloperday if (Style.isJavaScript() && EndsInComma) 15502c1cdae2SMartin Probst BreakBeforeParameter = true; 1551de0328aeSDaniel Jasper } 1552bd73bcf8SDaniel Jasper // Generally inherit NoLineBreak from the current scope to nested scope. 1553bd73bcf8SDaniel Jasper // However, don't do this for non-empty nested blocks, dict literals and 1554bd73bcf8SDaniel Jasper // array literals as these follow different indentation rules. 1555bd73bcf8SDaniel Jasper bool NoLineBreak = 1556bd73bcf8SDaniel Jasper Current.Children.empty() && 1557bd73bcf8SDaniel Jasper !Current.isOneOf(TT_DictLiteral, TT_ArrayInitializerLSquare) && 1558bd73bcf8SDaniel Jasper (State.Stack.back().NoLineBreak || 1559240527caSDaniel Jasper State.Stack.back().NoLineBreakInOperand || 1560bd73bcf8SDaniel Jasper (Current.is(TT_TemplateOpener) && 1561240527caSDaniel Jasper State.Stack.back().ContainsUnwrappedBuilder)); 15627d42f3f7SDaniel Jasper State.Stack.push_back( 15638b4bc76aSKrasimir Georgiev ParenState(&Current, NewIndent, LastSpace, AvoidBinPacking, NoLineBreak)); 156411a0ac66SDaniel Jasper State.Stack.back().NestedBlockIndent = NestedBlockIndent; 15651db6c386SDaniel Jasper State.Stack.back().BreakBeforeParameter = BreakBeforeParameter; 1566f594e3d2Smydeveloperday State.Stack.back().HasMultipleNestedBlocks = 1567f594e3d2Smydeveloperday (Current.BlockParameterCount > 1); 1568fa0118e6SWawha 1569f594e3d2Smydeveloperday if (Style.BraceWrapping.BeforeLambdaBody && Current.Next != nullptr && 1570f594e3d2Smydeveloperday Current.Tok.is(tok::l_paren)) { 1571fa0118e6SWawha // Search for any parameter that is a lambda 1572fa0118e6SWawha FormatToken const *next = Current.Next; 1573fa0118e6SWawha while (next != nullptr) { 1574fa0118e6SWawha if (next->is(TT_LambdaLSquare)) { 1575fa0118e6SWawha State.Stack.back().HasMultipleNestedBlocks = true; 1576fa0118e6SWawha break; 1577fa0118e6SWawha } 1578fa0118e6SWawha next = next->Next; 1579fa0118e6SWawha } 1580fa0118e6SWawha } 1581fa0118e6SWawha 158209051f29SBen Hamilton State.Stack.back().IsInsideObjCArrayLiteral = 158309051f29SBen Hamilton Current.is(TT_ArrayInitializerLSquare) && Current.Previous && 158409051f29SBen Hamilton Current.Previous->is(tok::at); 1585de0328aeSDaniel Jasper } 1586de0328aeSDaniel Jasper 158760553be5SDaniel Jasper void ContinuationIndenter::moveStatePastScopeCloser(LineState &State) { 158860553be5SDaniel Jasper const FormatToken &Current = *State.NextToken; 158960553be5SDaniel Jasper if (!Current.closesScope()) 159060553be5SDaniel Jasper return; 159160553be5SDaniel Jasper 1592de0328aeSDaniel Jasper // If we encounter a closing ), ], } or >, we can remove a level from our 1593de0328aeSDaniel Jasper // stacks. 159496df37a6SDaniel Jasper if (State.Stack.size() > 1 && 1595c06f6da3SDaniel Jasper (Current.isOneOf(tok::r_paren, tok::r_square, TT_TemplateString) || 15969fe0e8daSDaniel Jasper (Current.is(tok::r_brace) && State.NextToken != State.Line->First) || 1597a79d62d2SKrasimir Georgiev State.NextToken->is(TT_TemplateCloser) || 1598a79d62d2SKrasimir Georgiev (Current.is(tok::greater) && Current.is(TT_DictLiteral)))) 1599de0328aeSDaniel Jasper State.Stack.pop_back(); 1600335ff266SDaniel Jasper 16012dc3dac9SJacek Olesiak // Reevaluate whether ObjC message arguments fit into one line. 16022dc3dac9SJacek Olesiak // If a receiver spans multiple lines, e.g.: 16032dc3dac9SJacek Olesiak // [[object block:^{ 16042dc3dac9SJacek Olesiak // return 42; 16052dc3dac9SJacek Olesiak // }] a:42 b:42]; 16062dc3dac9SJacek Olesiak // BreakBeforeParameter is calculated based on an incorrect assumption 16072dc3dac9SJacek Olesiak // (it is checked whether the whole expression fits into one line without 16082dc3dac9SJacek Olesiak // considering a line break inside a message receiver). 1609f1191705SNico Weber // We check whether arguments fit after receiver scope closer (into the same 16102dc3dac9SJacek Olesiak // line). 16112dc3dac9SJacek Olesiak if (State.Stack.back().BreakBeforeParameter && Current.MatchingParen && 16122dc3dac9SJacek Olesiak Current.MatchingParen->Previous) { 16132dc3dac9SJacek Olesiak const FormatToken &CurrentScopeOpener = *Current.MatchingParen->Previous; 16142dc3dac9SJacek Olesiak if (CurrentScopeOpener.is(TT_ObjCMethodExpr) && 16152dc3dac9SJacek Olesiak CurrentScopeOpener.MatchingParen) { 16162dc3dac9SJacek Olesiak int NecessarySpaceInLine = 16172dc3dac9SJacek Olesiak getLengthToMatchingParen(CurrentScopeOpener, State.Stack) + 16182dc3dac9SJacek Olesiak CurrentScopeOpener.TotalLength - Current.TotalLength - 1; 16192dc3dac9SJacek Olesiak if (State.Column + Current.ColumnWidth + NecessarySpaceInLine <= 16202dc3dac9SJacek Olesiak Style.ColumnLimit) 16212dc3dac9SJacek Olesiak State.Stack.back().BreakBeforeParameter = false; 16222dc3dac9SJacek Olesiak } 16232dc3dac9SJacek Olesiak } 16242dc3dac9SJacek Olesiak 1625de0328aeSDaniel Jasper if (Current.is(tok::r_square)) { 1626de0328aeSDaniel Jasper // If this ends the array subscript expr, reset the corresponding value. 1627de0328aeSDaniel Jasper const FormatToken *NextNonComment = Current.getNextNonComment(); 1628de0328aeSDaniel Jasper if (NextNonComment && NextNonComment->isNot(tok::l_square)) 1629de0328aeSDaniel Jasper State.Stack.back().StartOfArraySubscripts = 0; 1630de0328aeSDaniel Jasper } 163160553be5SDaniel Jasper } 1632de0328aeSDaniel Jasper 163360553be5SDaniel Jasper void ContinuationIndenter::moveStateToNewBlock(LineState &State) { 163411a0ac66SDaniel Jasper unsigned NestedBlockIndent = State.Stack.back().NestedBlockIndent; 163550d634b3SDaniel Jasper // ObjC block sometimes follow special indentation rules. 1636a98b7b01SDaniel Jasper unsigned NewIndent = 163711a0ac66SDaniel Jasper NestedBlockIndent + (State.NextToken->is(TT_ObjCBlockLBrace) 163850d634b3SDaniel Jasper ? Style.ObjCBlockIndentWidth 163960553be5SDaniel Jasper : Style.IndentWidth); 16408b4bc76aSKrasimir Georgiev State.Stack.push_back(ParenState(State.NextToken, NewIndent, 16418b4bc76aSKrasimir Georgiev State.Stack.back().LastSpace, 16427d42f3f7SDaniel Jasper /*AvoidBinPacking=*/true, 16436501f7e8SDaniel Jasper /*NoLineBreak=*/false)); 164411a0ac66SDaniel Jasper State.Stack.back().NestedBlockIndent = NestedBlockIndent; 164560553be5SDaniel Jasper State.Stack.back().BreakBeforeParameter = true; 1646de0328aeSDaniel Jasper } 1647de0328aeSDaniel Jasper 16489ad83fe7SKrasimir Georgiev static unsigned getLastLineEndColumn(StringRef Text, unsigned StartColumn, 16499ad83fe7SKrasimir Georgiev unsigned TabWidth, 16509ad83fe7SKrasimir Georgiev encoding::Encoding Encoding) { 16519ad83fe7SKrasimir Georgiev size_t LastNewlinePos = Text.find_last_of("\n"); 16529ad83fe7SKrasimir Georgiev if (LastNewlinePos == StringRef::npos) { 16539ad83fe7SKrasimir Georgiev return StartColumn + 16549ad83fe7SKrasimir Georgiev encoding::columnWidthWithTabs(Text, StartColumn, TabWidth, Encoding); 16559ad83fe7SKrasimir Georgiev } else { 16569ad83fe7SKrasimir Georgiev return encoding::columnWidthWithTabs(Text.substr(LastNewlinePos), 16579ad83fe7SKrasimir Georgiev /*StartColumn=*/0, TabWidth, Encoding); 16589ad83fe7SKrasimir Georgiev } 16599ad83fe7SKrasimir Georgiev } 16609ad83fe7SKrasimir Georgiev 16619ad83fe7SKrasimir Georgiev unsigned ContinuationIndenter::reformatRawStringLiteral( 166245ab559eSManuel Klimek const FormatToken &Current, LineState &State, 1663bda8482aSKrasimir Georgiev const FormatStyle &RawStringStyle, bool DryRun, bool Newline) { 166445ab559eSManuel Klimek unsigned StartColumn = State.Column - Current.ColumnWidth; 1665412ed095SKrasimir Georgiev StringRef OldDelimiter = *getRawStringDelimiter(Current.TokenText); 1666412ed095SKrasimir Georgiev StringRef NewDelimiter = 1667412ed095SKrasimir Georgiev getCanonicalRawStringDelimiter(Style, RawStringStyle.Language); 1668587859d9SKrasimir Georgiev if (NewDelimiter.empty()) 1669412ed095SKrasimir Georgiev NewDelimiter = OldDelimiter; 16709ad83fe7SKrasimir Georgiev // The text of a raw string is between the leading 'R"delimiter(' and the 16719ad83fe7SKrasimir Georgiev // trailing 'delimiter)"'. 1672412ed095SKrasimir Georgiev unsigned OldPrefixSize = 3 + OldDelimiter.size(); 1673412ed095SKrasimir Georgiev unsigned OldSuffixSize = 2 + OldDelimiter.size(); 1674412ed095SKrasimir Georgiev // We create a virtual text environment which expects a null-terminated 1675412ed095SKrasimir Georgiev // string, so we cannot use StringRef. 1676adcd0268SBenjamin Kramer std::string RawText = std::string( 1677adcd0268SBenjamin Kramer Current.TokenText.substr(OldPrefixSize).drop_back(OldSuffixSize)); 1678412ed095SKrasimir Georgiev if (NewDelimiter != OldDelimiter) { 1679412ed095SKrasimir Georgiev // Don't update to the canonical delimiter 'deli' if ')deli"' occurs in the 1680412ed095SKrasimir Georgiev // raw string. 1681412ed095SKrasimir Georgiev std::string CanonicalDelimiterSuffix = (")" + NewDelimiter + "\"").str(); 1682412ed095SKrasimir Georgiev if (StringRef(RawText).contains(CanonicalDelimiterSuffix)) 1683412ed095SKrasimir Georgiev NewDelimiter = OldDelimiter; 1684412ed095SKrasimir Georgiev } 16859ad83fe7SKrasimir Georgiev 1686412ed095SKrasimir Georgiev unsigned NewPrefixSize = 3 + NewDelimiter.size(); 1687412ed095SKrasimir Georgiev unsigned NewSuffixSize = 2 + NewDelimiter.size(); 1688412ed095SKrasimir Georgiev 1689412ed095SKrasimir Georgiev // The first start column is the column the raw text starts after formatting. 1690412ed095SKrasimir Georgiev unsigned FirstStartColumn = StartColumn + NewPrefixSize; 16919ad83fe7SKrasimir Georgiev 16929ad83fe7SKrasimir Georgiev // The next start column is the intended indentation a line break inside 16939ad83fe7SKrasimir Georgiev // the raw string at level 0. It is determined by the following rules: 16949ad83fe7SKrasimir Georgiev // - if the content starts on newline, it is one level more than the current 16959ad83fe7SKrasimir Georgiev // indent, and 16969ad83fe7SKrasimir Georgiev // - if the content does not start on a newline, it is the first start 16979ad83fe7SKrasimir Georgiev // column. 16989ad83fe7SKrasimir Georgiev // These rules have the advantage that the formatted content both does not 16999ad83fe7SKrasimir Georgiev // violate the rectangle rule and visually flows within the surrounding 17009ad83fe7SKrasimir Georgiev // source. 1701412ed095SKrasimir Georgiev bool ContentStartsOnNewline = Current.TokenText[OldPrefixSize] == '\n'; 1702128fcffbSKrasimir Georgiev // If this token is the last parameter (checked by looking if it's followed by 1703bda8482aSKrasimir Georgiev // `)` and is not on a newline, the base the indent off the line's nested 1704bda8482aSKrasimir Georgiev // block indent. Otherwise, base the indent off the arguments indent, so we 1705bda8482aSKrasimir Georgiev // can achieve: 1706bda8482aSKrasimir Georgiev // 1707128fcffbSKrasimir Georgiev // fffffffffff(1, 2, 3, R"pb( 1708128fcffbSKrasimir Georgiev // key1: 1 # 1709128fcffbSKrasimir Georgiev // key2: 2)pb"); 1710128fcffbSKrasimir Georgiev // 1711128fcffbSKrasimir Georgiev // fffffffffff(1, 2, 3, 1712128fcffbSKrasimir Georgiev // R"pb( 1713128fcffbSKrasimir Georgiev // key1: 1 # 1714128fcffbSKrasimir Georgiev // key2: 2 1715bda8482aSKrasimir Georgiev // )pb"); 1716bda8482aSKrasimir Georgiev // 1717bda8482aSKrasimir Georgiev // fffffffffff(1, 2, 3, 1718bda8482aSKrasimir Georgiev // R"pb( 1719bda8482aSKrasimir Georgiev // key1: 1 # 1720bda8482aSKrasimir Georgiev // key2: 2 1721128fcffbSKrasimir Georgiev // )pb", 1722128fcffbSKrasimir Georgiev // 5); 1723bda8482aSKrasimir Georgiev unsigned CurrentIndent = 1724bda8482aSKrasimir Georgiev (!Newline && Current.Next && Current.Next->is(tok::r_paren)) 1725128fcffbSKrasimir Georgiev ? State.Stack.back().NestedBlockIndent 1726128fcffbSKrasimir Georgiev : State.Stack.back().Indent; 1727128fcffbSKrasimir Georgiev unsigned NextStartColumn = ContentStartsOnNewline 1728128fcffbSKrasimir Georgiev ? CurrentIndent + Style.IndentWidth 17299ad83fe7SKrasimir Georgiev : FirstStartColumn; 17309ad83fe7SKrasimir Georgiev 17319ad83fe7SKrasimir Georgiev // The last start column is the column the raw string suffix starts if it is 17329ad83fe7SKrasimir Georgiev // put on a newline. 17339ad83fe7SKrasimir Georgiev // The last start column is the intended indentation of the raw string postfix 17349ad83fe7SKrasimir Georgiev // if it is put on a newline. It is determined by the following rules: 17359ad83fe7SKrasimir Georgiev // - if the raw string prefix starts on a newline, it is the column where 17369ad83fe7SKrasimir Georgiev // that raw string prefix starts, and 17379ad83fe7SKrasimir Georgiev // - if the raw string prefix does not start on a newline, it is the current 17389ad83fe7SKrasimir Georgiev // indent. 17395bcf99b4SPaul Hoad unsigned LastStartColumn = 17405bcf99b4SPaul Hoad Current.NewlinesBefore ? FirstStartColumn - NewPrefixSize : CurrentIndent; 17419ad83fe7SKrasimir Georgiev 17429ad83fe7SKrasimir Georgiev std::pair<tooling::Replacements, unsigned> Fixes = internal::reformat( 17439ad83fe7SKrasimir Georgiev RawStringStyle, RawText, {tooling::Range(0, RawText.size())}, 17449ad83fe7SKrasimir Georgiev FirstStartColumn, NextStartColumn, LastStartColumn, "<stdin>", 17450fcb580bSKrasimir Georgiev /*Status=*/nullptr); 17469ad83fe7SKrasimir Georgiev 17479ad83fe7SKrasimir Georgiev auto NewCode = applyAllReplacements(RawText, Fixes.first); 17489ad83fe7SKrasimir Georgiev tooling::Replacements NoFixes; 17499ad83fe7SKrasimir Georgiev if (!NewCode) { 1750128fcffbSKrasimir Georgiev return addMultilineToken(Current, State); 17519ad83fe7SKrasimir Georgiev } 17529ad83fe7SKrasimir Georgiev if (!DryRun) { 1753412ed095SKrasimir Georgiev if (NewDelimiter != OldDelimiter) { 1754412ed095SKrasimir Georgiev // In 'R"delimiter(...', the delimiter starts 2 characters after the start 1755412ed095SKrasimir Georgiev // of the token. 1756412ed095SKrasimir Georgiev SourceLocation PrefixDelimiterStart = 1757412ed095SKrasimir Georgiev Current.Tok.getLocation().getLocWithOffset(2); 1758412ed095SKrasimir Georgiev auto PrefixErr = Whitespaces.addReplacement(tooling::Replacement( 1759412ed095SKrasimir Georgiev SourceMgr, PrefixDelimiterStart, OldDelimiter.size(), NewDelimiter)); 1760412ed095SKrasimir Georgiev if (PrefixErr) { 1761412ed095SKrasimir Georgiev llvm::errs() 1762412ed095SKrasimir Georgiev << "Failed to update the prefix delimiter of a raw string: " 1763412ed095SKrasimir Georgiev << llvm::toString(std::move(PrefixErr)) << "\n"; 1764412ed095SKrasimir Georgiev } 1765412ed095SKrasimir Georgiev // In 'R"delimiter(...)delimiter"', the suffix delimiter starts at 1766412ed095SKrasimir Georgiev // position length - 1 - |delimiter|. 1767412ed095SKrasimir Georgiev SourceLocation SuffixDelimiterStart = 1768412ed095SKrasimir Georgiev Current.Tok.getLocation().getLocWithOffset(Current.TokenText.size() - 1769412ed095SKrasimir Georgiev 1 - OldDelimiter.size()); 1770412ed095SKrasimir Georgiev auto SuffixErr = Whitespaces.addReplacement(tooling::Replacement( 1771412ed095SKrasimir Georgiev SourceMgr, SuffixDelimiterStart, OldDelimiter.size(), NewDelimiter)); 1772412ed095SKrasimir Georgiev if (SuffixErr) { 1773412ed095SKrasimir Georgiev llvm::errs() 1774412ed095SKrasimir Georgiev << "Failed to update the suffix delimiter of a raw string: " 1775412ed095SKrasimir Georgiev << llvm::toString(std::move(SuffixErr)) << "\n"; 1776412ed095SKrasimir Georgiev } 1777412ed095SKrasimir Georgiev } 17789ad83fe7SKrasimir Georgiev SourceLocation OriginLoc = 1779412ed095SKrasimir Georgiev Current.Tok.getLocation().getLocWithOffset(OldPrefixSize); 17809ad83fe7SKrasimir Georgiev for (const tooling::Replacement &Fix : Fixes.first) { 17819ad83fe7SKrasimir Georgiev auto Err = Whitespaces.addReplacement(tooling::Replacement( 17829ad83fe7SKrasimir Georgiev SourceMgr, OriginLoc.getLocWithOffset(Fix.getOffset()), 17839ad83fe7SKrasimir Georgiev Fix.getLength(), Fix.getReplacementText())); 17849ad83fe7SKrasimir Georgiev if (Err) { 17859ad83fe7SKrasimir Georgiev llvm::errs() << "Failed to reformat raw string: " 17869ad83fe7SKrasimir Georgiev << llvm::toString(std::move(Err)) << "\n"; 17879ad83fe7SKrasimir Georgiev } 17889ad83fe7SKrasimir Georgiev } 17899ad83fe7SKrasimir Georgiev } 17909ad83fe7SKrasimir Georgiev unsigned RawLastLineEndColumn = getLastLineEndColumn( 17919ad83fe7SKrasimir Georgiev *NewCode, FirstStartColumn, Style.TabWidth, Encoding); 1792412ed095SKrasimir Georgiev State.Column = RawLastLineEndColumn + NewSuffixSize; 1793ce00978bSKrasimir Georgiev // Since we're updating the column to after the raw string literal here, we 1794ce00978bSKrasimir Georgiev // have to manually add the penalty for the prefix R"delim( over the column 1795ce00978bSKrasimir Georgiev // limit. 1796ce00978bSKrasimir Georgiev unsigned PrefixExcessCharacters = 17975bcf99b4SPaul Hoad StartColumn + NewPrefixSize > Style.ColumnLimit 17985bcf99b4SPaul Hoad ? StartColumn + NewPrefixSize - Style.ColumnLimit 17995bcf99b4SPaul Hoad : 0; 1800128fcffbSKrasimir Georgiev bool IsMultiline = 1801128fcffbSKrasimir Georgiev ContentStartsOnNewline || (NewCode->find('\n') != std::string::npos); 1802128fcffbSKrasimir Georgiev if (IsMultiline) { 1803128fcffbSKrasimir Georgiev // Break before further function parameters on all levels. 1804128fcffbSKrasimir Georgiev for (unsigned i = 0, e = State.Stack.size(); i != e; ++i) 1805128fcffbSKrasimir Georgiev State.Stack[i].BreakBeforeParameter = true; 1806128fcffbSKrasimir Georgiev } 1807ce00978bSKrasimir Georgiev return Fixes.second + PrefixExcessCharacters * Style.PenaltyExcessCharacter; 18089ad83fe7SKrasimir Georgiev } 18099ad83fe7SKrasimir Georgiev 1810917f9e04SAlexander Kornienko unsigned ContinuationIndenter::addMultilineToken(const FormatToken &Current, 1811d7b837e7SAlexander Kornienko LineState &State) { 1812d7b837e7SAlexander Kornienko // Break before further function parameters on all levels. 1813d7b837e7SAlexander Kornienko for (unsigned i = 0, e = State.Stack.size(); i != e; ++i) 1814d7b837e7SAlexander Kornienko State.Stack[i].BreakBeforeParameter = true; 1815d7b837e7SAlexander Kornienko 181639856b71SAlexander Kornienko unsigned ColumnsUsed = State.Column; 1817632abb9bSAlexander Kornienko // We can only affect layout of the first and the last line, so the penalty 1818632abb9bSAlexander Kornienko // for all other lines is constant, and we ignore it. 1819ebb43caaSAlexander Kornienko State.Column = Current.LastLineColumnWidth; 1820632abb9bSAlexander Kornienko 18219fe0e8daSDaniel Jasper if (ColumnsUsed > getColumnLimit(State)) 18229fe0e8daSDaniel Jasper return Style.PenaltyExcessCharacter * (ColumnsUsed - getColumnLimit(State)); 1823d7b837e7SAlexander Kornienko return 0; 1824d7b837e7SAlexander Kornienko } 1825d7b837e7SAlexander Kornienko 182645ab559eSManuel Klimek unsigned ContinuationIndenter::handleEndOfLine(const FormatToken &Current, 182745ab559eSManuel Klimek LineState &State, bool DryRun, 1828bda8482aSKrasimir Georgiev bool AllowBreak, bool Newline) { 182945ab559eSManuel Klimek unsigned Penalty = 0; 18309ad83fe7SKrasimir Georgiev // Compute the raw string style to use in case this is a raw string literal 18319ad83fe7SKrasimir Georgiev // that can be reformatted. 183245ab559eSManuel Klimek auto RawStringStyle = getRawStringStyle(Current, State); 1833dd0c4e1dSDaniel Jasper if (RawStringStyle && !Current.Finalized) { 1834bda8482aSKrasimir Georgiev Penalty = reformatRawStringLiteral(Current, State, *RawStringStyle, DryRun, 1835bda8482aSKrasimir Georgiev Newline); 183645ab559eSManuel Klimek } else if (Current.IsMultiline && Current.isNot(TT_BlockComment)) { 18379ad83fe7SKrasimir Georgiev // Don't break multi-line tokens other than block comments and raw string 18389ad83fe7SKrasimir Georgiev // literals. Instead, just update the state. 183945ab559eSManuel Klimek Penalty = addMultilineToken(Current, State); 184045ab559eSManuel Klimek } else if (State.Line->Type != LT_ImportStatement) { 184145ab559eSManuel Klimek // We generally don't break import statements. 18420b58c328SManuel Klimek LineState OriginalState = State; 18430b58c328SManuel Klimek 18440b58c328SManuel Klimek // Whether we force the reflowing algorithm to stay strictly within the 18450b58c328SManuel Klimek // column limit. 18460b58c328SManuel Klimek bool Strict = false; 18470b58c328SManuel Klimek // Whether the first non-strict attempt at reflowing did intentionally 18480b58c328SManuel Klimek // exceed the column limit. 18490b58c328SManuel Klimek bool Exceeded = false; 18500b58c328SManuel Klimek std::tie(Penalty, Exceeded) = breakProtrudingToken( 18510b58c328SManuel Klimek Current, State, AllowBreak, /*DryRun=*/true, Strict); 18520b58c328SManuel Klimek if (Exceeded) { 18530b58c328SManuel Klimek // If non-strict reflowing exceeds the column limit, try whether strict 18540b58c328SManuel Klimek // reflowing leads to an overall lower penalty. 18550b58c328SManuel Klimek LineState StrictState = OriginalState; 18560b58c328SManuel Klimek unsigned StrictPenalty = 18570b58c328SManuel Klimek breakProtrudingToken(Current, StrictState, AllowBreak, 18580b58c328SManuel Klimek /*DryRun=*/true, /*Strict=*/true) 18590b58c328SManuel Klimek .first; 18600b58c328SManuel Klimek Strict = StrictPenalty <= Penalty; 18610b58c328SManuel Klimek if (Strict) { 18620b58c328SManuel Klimek Penalty = StrictPenalty; 18630b58c328SManuel Klimek State = StrictState; 18640b58c328SManuel Klimek } 18650b58c328SManuel Klimek } 18660b58c328SManuel Klimek if (!DryRun) { 18670b58c328SManuel Klimek // If we're not in dry-run mode, apply the changes with the decision on 18680b58c328SManuel Klimek // strictness made above. 18690b58c328SManuel Klimek breakProtrudingToken(Current, OriginalState, AllowBreak, /*DryRun=*/false, 18700b58c328SManuel Klimek Strict); 18710b58c328SManuel Klimek } 187245ab559eSManuel Klimek } 187345ab559eSManuel Klimek if (State.Column > getColumnLimit(State)) { 187445ab559eSManuel Klimek unsigned ExcessCharacters = State.Column - getColumnLimit(State); 187545ab559eSManuel Klimek Penalty += Style.PenaltyExcessCharacter * ExcessCharacters; 187645ab559eSManuel Klimek } 187745ab559eSManuel Klimek return Penalty; 187845ab559eSManuel Klimek } 1879917f9e04SAlexander Kornienko 18802537e220SKrasimir Georgiev // Returns the enclosing function name of a token, or the empty string if not 18812537e220SKrasimir Georgiev // found. 18822537e220SKrasimir Georgiev static StringRef getEnclosingFunctionName(const FormatToken &Current) { 18832537e220SKrasimir Georgiev // Look for: 'function(' or 'function<templates>(' before Current. 18842537e220SKrasimir Georgiev auto Tok = Current.getPreviousNonComment(); 18852537e220SKrasimir Georgiev if (!Tok || !Tok->is(tok::l_paren)) 18862537e220SKrasimir Georgiev return ""; 18872537e220SKrasimir Georgiev Tok = Tok->getPreviousNonComment(); 18882537e220SKrasimir Georgiev if (!Tok) 18892537e220SKrasimir Georgiev return ""; 18902537e220SKrasimir Georgiev if (Tok->is(TT_TemplateCloser)) { 18912537e220SKrasimir Georgiev Tok = Tok->MatchingParen; 18922537e220SKrasimir Georgiev if (Tok) 18932537e220SKrasimir Georgiev Tok = Tok->getPreviousNonComment(); 18942537e220SKrasimir Georgiev } 18952537e220SKrasimir Georgiev if (!Tok || !Tok->is(tok::identifier)) 18962537e220SKrasimir Georgiev return ""; 18972537e220SKrasimir Georgiev return Tok->TokenText; 18982537e220SKrasimir Georgiev } 18992537e220SKrasimir Georgiev 190045ab559eSManuel Klimek llvm::Optional<FormatStyle> 190145ab559eSManuel Klimek ContinuationIndenter::getRawStringStyle(const FormatToken &Current, 190245ab559eSManuel Klimek const LineState &State) { 190345ab559eSManuel Klimek if (!Current.isStringLiteral()) 190445ab559eSManuel Klimek return None; 190545ab559eSManuel Klimek auto Delimiter = getRawStringDelimiter(Current.TokenText); 190645ab559eSManuel Klimek if (!Delimiter) 190745ab559eSManuel Klimek return None; 19082537e220SKrasimir Georgiev auto RawStringStyle = RawStringFormats.getDelimiterStyle(*Delimiter); 19092d69f5d8SKrasimir Georgiev if (!RawStringStyle && Delimiter->empty()) 19102537e220SKrasimir Georgiev RawStringStyle = RawStringFormats.getEnclosingFunctionStyle( 19112537e220SKrasimir Georgiev getEnclosingFunctionName(Current)); 191245ab559eSManuel Klimek if (!RawStringStyle) 191345ab559eSManuel Klimek return None; 191445ab559eSManuel Klimek RawStringStyle->ColumnLimit = getColumnLimit(State); 191545ab559eSManuel Klimek return RawStringStyle; 191645ab559eSManuel Klimek } 19179885784dSDaniel Jasper 19185bcf99b4SPaul Hoad std::unique_ptr<BreakableToken> 19195bcf99b4SPaul Hoad ContinuationIndenter::createBreakableToken(const FormatToken &Current, 19205bcf99b4SPaul Hoad LineState &State, bool AllowBreak) { 192139856b71SAlexander Kornienko unsigned StartColumn = State.Column - Current.ColumnWidth; 192204b6a081SDaniel Jasper if (Current.isStringLiteral()) { 19238b7881a0Smydeveloperday // FIXME: String literal breaking is currently disabled for C#, Java, Json 19248b7881a0Smydeveloperday // and JavaScript, as it requires strings to be merged using "+" which we 1925cbb726d0SPaul Hoad // don't support. 1926142e79b8Smydeveloperday if (Style.Language == FormatStyle::LK_Java || Style.isJavaScript() || 1927142e79b8Smydeveloperday Style.isCSharp() || Style.isJson() || !Style.BreakStringLiterals || 1928142e79b8Smydeveloperday !AllowBreak) 192945ab559eSManuel Klimek return nullptr; 1930428f0b14SDaniel Jasper 1931384b40b9SAlexander Kornienko // Don't break string literals inside preprocessor directives (except for 1932384b40b9SAlexander Kornienko // #define directives, as their contents are stored in separate lines and 1933384b40b9SAlexander Kornienko // are not affected by this check). 1934384b40b9SAlexander Kornienko // This way we avoid breaking code with line directives and unknown 1935384b40b9SAlexander Kornienko // preprocessor directives that contain long string literals. 1936384b40b9SAlexander Kornienko if (State.Line->Type == LT_PreprocessorDirective) 193745ab559eSManuel Klimek return nullptr; 1938de0328aeSDaniel Jasper // Exempts unterminated string literals from line breaking. The user will 1939de0328aeSDaniel Jasper // likely want to terminate the string before any line breaking is done. 1940de0328aeSDaniel Jasper if (Current.IsUnterminatedLiteral) 194145ab559eSManuel Klimek return nullptr; 194209051f29SBen Hamilton // Don't break string literals inside Objective-C array literals (doing so 194309051f29SBen Hamilton // raises the warning -Wobjc-string-concatenation). 194409051f29SBen Hamilton if (State.Stack.back().IsInsideObjCArrayLiteral) { 194509051f29SBen Hamilton return nullptr; 194609051f29SBen Hamilton } 1947de0328aeSDaniel Jasper 194881e3294eSAlexander Kornienko StringRef Text = Current.TokenText; 194981e3294eSAlexander Kornienko StringRef Prefix; 195081e3294eSAlexander Kornienko StringRef Postfix; 195181e3294eSAlexander Kornienko // FIXME: Handle whitespace between '_T', '(', '"..."', and ')'. 195281e3294eSAlexander Kornienko // FIXME: Store Prefix and Suffix (or PrefixLength and SuffixLength to 195381e3294eSAlexander Kornienko // reduce the overhead) for each FormatToken, which is a string, so that we 195481e3294eSAlexander Kornienko // don't run multiple checks here on the hot path. 195581e3294eSAlexander Kornienko if ((Text.endswith(Postfix = "\"") && 1956d4fa2e63SAlexander Kornienko (Text.startswith(Prefix = "@\"") || Text.startswith(Prefix = "\"") || 1957174b0125SDaniel Jasper Text.startswith(Prefix = "u\"") || Text.startswith(Prefix = "U\"") || 1958174b0125SDaniel Jasper Text.startswith(Prefix = "u8\"") || 195981e3294eSAlexander Kornienko Text.startswith(Prefix = "L\""))) || 1960732b6bd4SAlexander Kornienko (Text.startswith(Prefix = "_T(\"") && Text.endswith(Postfix = "\")"))) { 196155c23a10SKrasimir Georgiev // We need this to address the case where there is an unbreakable tail 196255c23a10SKrasimir Georgiev // only if certain other formatting decisions have been taken. The 196355c23a10SKrasimir Georgiev // UnbreakableTailLength of Current is an overapproximation is that case 196455c23a10SKrasimir Georgiev // and we need to be correct here. 196555c23a10SKrasimir Georgiev unsigned UnbreakableTailLength = (State.NextToken && canBreak(State)) 196655c23a10SKrasimir Georgiev ? 0 196755c23a10SKrasimir Georgiev : Current.UnbreakableTailLength; 19682b3d49b6SJonas Devlieghere return std::make_unique<BreakableStringLiteral>( 196955c23a10SKrasimir Georgiev Current, StartColumn, Prefix, Postfix, UnbreakableTailLength, 197055c23a10SKrasimir Georgiev State.Line->InPPDirective, Encoding, Style); 197181e3294eSAlexander Kornienko } 1972acadc8e0SDaniel Jasper } else if (Current.is(TT_BlockComment)) { 197335599fdfSKrasimir Georgiev if (!Style.ReflowComments || 197491834227SKrasimir Georgiev // If a comment token switches formatting, like 197591834227SKrasimir Georgiev // /* clang-format on */, we don't want to break it further, 197691834227SKrasimir Georgiev // but we may still want to adjust its indentation. 197745ab559eSManuel Klimek switchesFormatting(Current)) { 197845ab559eSManuel Klimek return nullptr; 197945ab559eSManuel Klimek } 19802b3d49b6SJonas Devlieghere return std::make_unique<BreakableBlockComment>( 19817d42f3f7SDaniel Jasper Current, StartColumn, Current.OriginalColumn, !Current.Previous, 1982108cbbc2SOwen Pan State.Line->InPPDirective, Encoding, Style, Whitespaces.useCRLF()); 1983a98b7b01SDaniel Jasper } else if (Current.is(TT_LineComment) && 19842145bc02SCraig Topper (Current.Previous == nullptr || 1985a98b7b01SDaniel Jasper Current.Previous->isNot(TT_ImplicitStringLiteral))) { 19864dcfae6aSManuel Klimek bool RegularComments = [&]() { 19874dcfae6aSManuel Klimek for (const FormatToken *T = &Current; T && T->is(TT_LineComment); 19884dcfae6aSManuel Klimek T = T->Next) { 19894dcfae6aSManuel Klimek if (!(T->TokenText.startswith("//") || T->TokenText.startswith("#"))) 19904dcfae6aSManuel Klimek return false; 19914dcfae6aSManuel Klimek } 19924dcfae6aSManuel Klimek return true; 19934dcfae6aSManuel Klimek }(); 1994a0a5039dSDaniel Jasper if (!Style.ReflowComments || 199591834227SKrasimir Georgiev CommentPragmasRegex.match(Current.TokenText.substr(2)) || 19964dcfae6aSManuel Klimek switchesFormatting(Current) || !RegularComments) 199745ab559eSManuel Klimek return nullptr; 19982b3d49b6SJonas Devlieghere return std::make_unique<BreakableLineCommentSection>( 19999aa38a06SBjörn Schäpers Current, StartColumn, /*InPPDirective=*/false, Encoding, Style); 200045ab559eSManuel Klimek } 200145ab559eSManuel Klimek return nullptr; 200245ab559eSManuel Klimek } 200345ab559eSManuel Klimek 20040b58c328SManuel Klimek std::pair<unsigned, bool> 20050b58c328SManuel Klimek ContinuationIndenter::breakProtrudingToken(const FormatToken &Current, 20060b58c328SManuel Klimek LineState &State, bool AllowBreak, 20070b58c328SManuel Klimek bool DryRun, bool Strict) { 200893699f4dSManuel Klimek std::unique_ptr<const BreakableToken> Token = 200945ab559eSManuel Klimek createBreakableToken(Current, State, AllowBreak); 201045ab559eSManuel Klimek if (!Token) 20110b58c328SManuel Klimek return {0, false}; 201293699f4dSManuel Klimek assert(Token->getLineCount() > 0); 201345ab559eSManuel Klimek unsigned ColumnLimit = getColumnLimit(State); 201445ab559eSManuel Klimek if (Current.is(TT_LineComment)) { 20153abbb8a4SAlexander Kornienko // We don't insert backslashes when breaking line comments. 20163abbb8a4SAlexander Kornienko ColumnLimit = Style.ColumnLimit; 2017de0328aeSDaniel Jasper } 2018418b4a7bSBjörn Schäpers if (ColumnLimit == 0) { 2019418b4a7bSBjörn Schäpers // To make the rest of the function easier set the column limit to the 2020418b4a7bSBjörn Schäpers // maximum, if there should be no limit. 2021418b4a7bSBjörn Schäpers ColumnLimit = std::numeric_limits<decltype(ColumnLimit)>::max(); 2022418b4a7bSBjörn Schäpers } 20233abbb8a4SAlexander Kornienko if (Current.UnbreakableTailLength >= ColumnLimit) 20240b58c328SManuel Klimek return {0, false}; 202593699f4dSManuel Klimek // ColumnWidth was already accounted into State.Column before calling 202693699f4dSManuel Klimek // breakProtrudingToken. 202793699f4dSManuel Klimek unsigned StartColumn = State.Column - Current.ColumnWidth; 202877866143SManuel Klimek unsigned NewBreakPenalty = Current.isStringLiteral() 202977866143SManuel Klimek ? Style.PenaltyBreakString 203077866143SManuel Klimek : Style.PenaltyBreakComment; 20310b58c328SManuel Klimek // Stores whether we intentionally decide to let a line exceed the column 20320b58c328SManuel Klimek // limit. 20330b58c328SManuel Klimek bool Exceeded = false; 203493699f4dSManuel Klimek // Stores whether we introduce a break anywhere in the token. 203577866143SManuel Klimek bool BreakInserted = Token->introducesBreakBeforeToken(); 203677866143SManuel Klimek // Store whether we inserted a new line break at the end of the previous 203777866143SManuel Klimek // logical line. 203877866143SManuel Klimek bool NewBreakBefore = false; 203991834227SKrasimir Georgiev // We use a conservative reflowing strategy. Reflow starts after a line is 204091834227SKrasimir Georgiev // broken or the corresponding whitespace compressed. Reflow ends as soon as a 204191834227SKrasimir Georgiev // line that doesn't get reflown with the previous line is reached. 204293699f4dSManuel Klimek bool Reflow = false; 204393699f4dSManuel Klimek // Keep track of where we are in the token: 204493699f4dSManuel Klimek // Where we are in the content of the current logical line. 204522d7e6b0SKrasimir Georgiev unsigned TailOffset = 0; 204693699f4dSManuel Klimek // The column number we're currently at. 204793699f4dSManuel Klimek unsigned ContentStartColumn = 204893699f4dSManuel Klimek Token->getContentStartColumn(0, /*Break=*/false); 204993699f4dSManuel Klimek // The number of columns left in the current logical line after TailOffset. 205093699f4dSManuel Klimek unsigned RemainingTokenColumns = 205193699f4dSManuel Klimek Token->getRemainingLength(0, TailOffset, ContentStartColumn); 205293699f4dSManuel Klimek // Adapt the start of the token, for example indent. 205393699f4dSManuel Klimek if (!DryRun) 205493699f4dSManuel Klimek Token->adaptStartOfLine(0, Whitespaces); 205593699f4dSManuel Klimek 20566a5c95bdSKrasimir Georgiev unsigned ContentIndent = 0; 205793699f4dSManuel Klimek unsigned Penalty = 0; 20583538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << "Breaking protruding token at column " 20593538b39eSNicola Zaghen << StartColumn << ".\n"); 2060de0328aeSDaniel Jasper for (unsigned LineIndex = 0, EndIndex = Token->getLineCount(); 2061de0328aeSDaniel Jasper LineIndex != EndIndex; ++LineIndex) { 20623538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 20633538b39eSNicola Zaghen << " Line: " << LineIndex << " (Reflow: " << Reflow << ")\n"); 206477866143SManuel Klimek NewBreakBefore = false; 206593699f4dSManuel Klimek // If we did reflow the previous line, we'll try reflowing again. Otherwise 206693699f4dSManuel Klimek // we'll start reflowing if the current line is broken or whitespace is 206793699f4dSManuel Klimek // compressed. 206893699f4dSManuel Klimek bool TryReflow = Reflow; 206993699f4dSManuel Klimek // Break the current token until we can fit the rest of the line. 207093699f4dSManuel Klimek while (ContentStartColumn + RemainingTokenColumns > ColumnLimit) { 20713538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Over limit, need: " 207293699f4dSManuel Klimek << (ContentStartColumn + RemainingTokenColumns) 207393699f4dSManuel Klimek << ", space: " << ColumnLimit 207493699f4dSManuel Klimek << ", reflown prefix: " << ContentStartColumn 207593699f4dSManuel Klimek << ", offset in line: " << TailOffset << "\n"); 207693699f4dSManuel Klimek // If the current token doesn't fit, find the latest possible split in the 207793699f4dSManuel Klimek // current line so that breaking at it will be under the column limit. 207893699f4dSManuel Klimek // FIXME: Use the earliest possible split while reflowing to correctly 207993699f4dSManuel Klimek // compress whitespace within a line. 208093699f4dSManuel Klimek BreakableToken::Split Split = 208193699f4dSManuel Klimek Token->getSplit(LineIndex, TailOffset, ColumnLimit, 208293699f4dSManuel Klimek ContentStartColumn, CommentPragmasRegex); 2083de0328aeSDaniel Jasper if (Split.first == StringRef::npos) { 208493699f4dSManuel Klimek // No break opportunity - update the penalty and continue with the next 208593699f4dSManuel Klimek // logical line. 2086de0328aeSDaniel Jasper if (LineIndex < EndIndex - 1) 2087123ca80bSKrasimir Georgiev // The last line's penalty is handled in addNextStateToQueue() or when 2088123ca80bSKrasimir Georgiev // calling replaceWhitespaceAfterLastLine below. 2089de0328aeSDaniel Jasper Penalty += Style.PenaltyExcessCharacter * 209093699f4dSManuel Klimek (ContentStartColumn + RemainingTokenColumns - ColumnLimit); 20913538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " No break opportunity.\n"); 2092de0328aeSDaniel Jasper break; 2093de0328aeSDaniel Jasper } 2094de0328aeSDaniel Jasper assert(Split.first != 0); 2095875395f9SAlexander Kornienko 209693699f4dSManuel Klimek if (Token->supportsReflow()) { 209793699f4dSManuel Klimek // Check whether the next natural split point after the current one can 209893699f4dSManuel Klimek // still fit the line, either because we can compress away whitespace, 209993699f4dSManuel Klimek // or because the penalty the excess characters introduce is lower than 210093699f4dSManuel Klimek // the break penalty. 210193699f4dSManuel Klimek // We only do this for tokens that support reflowing, and thus allow us 210293699f4dSManuel Klimek // to change the whitespace arbitrarily (e.g. comments). 210393699f4dSManuel Klimek // Other tokens, like string literals, can be broken on arbitrary 210493699f4dSManuel Klimek // positions. 210577866143SManuel Klimek 210693699f4dSManuel Klimek // First, compute the columns from TailOffset to the next possible split 210793699f4dSManuel Klimek // position. 210893699f4dSManuel Klimek // For example: 210993699f4dSManuel Klimek // ColumnLimit: | 211093699f4dSManuel Klimek // // Some text that breaks 211193699f4dSManuel Klimek // ^ tail offset 211293699f4dSManuel Klimek // ^-- split 211393699f4dSManuel Klimek // ^-------- to split columns 211493699f4dSManuel Klimek // ^--- next split 211593699f4dSManuel Klimek // ^--------------- to next split columns 211693699f4dSManuel Klimek unsigned ToSplitColumns = Token->getRangeLength( 211793699f4dSManuel Klimek LineIndex, TailOffset, Split.first, ContentStartColumn); 21183538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " ToSplit: " << ToSplitColumns << "\n"); 211993699f4dSManuel Klimek 212093699f4dSManuel Klimek BreakableToken::Split NextSplit = Token->getSplit( 212193699f4dSManuel Klimek LineIndex, TailOffset + Split.first + Split.second, ColumnLimit, 212293699f4dSManuel Klimek ContentStartColumn + ToSplitColumns + 1, CommentPragmasRegex); 212393699f4dSManuel Klimek // Compute the columns necessary to fit the next non-breakable sequence 212493699f4dSManuel Klimek // into the current line. 212593699f4dSManuel Klimek unsigned ToNextSplitColumns = 0; 212693699f4dSManuel Klimek if (NextSplit.first == StringRef::npos) { 212793699f4dSManuel Klimek ToNextSplitColumns = Token->getRemainingLength(LineIndex, TailOffset, 212893699f4dSManuel Klimek ContentStartColumn); 212993699f4dSManuel Klimek } else { 213093699f4dSManuel Klimek ToNextSplitColumns = Token->getRangeLength( 213193699f4dSManuel Klimek LineIndex, TailOffset, 213293699f4dSManuel Klimek Split.first + Split.second + NextSplit.first, ContentStartColumn); 213393699f4dSManuel Klimek } 213493699f4dSManuel Klimek // Compress the whitespace between the break and the start of the next 213593699f4dSManuel Klimek // unbreakable sequence. 213693699f4dSManuel Klimek ToNextSplitColumns = 213793699f4dSManuel Klimek Token->getLengthAfterCompression(ToNextSplitColumns, Split); 21383538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 21393538b39eSNicola Zaghen << " ContentStartColumn: " << ContentStartColumn << "\n"); 21403538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 21413538b39eSNicola Zaghen << " ToNextSplit: " << ToNextSplitColumns << "\n"); 214293699f4dSManuel Klimek // If the whitespace compression makes us fit, continue on the current 214393699f4dSManuel Klimek // line. 214493699f4dSManuel Klimek bool ContinueOnLine = 214593699f4dSManuel Klimek ContentStartColumn + ToNextSplitColumns <= ColumnLimit; 214693699f4dSManuel Klimek unsigned ExcessCharactersPenalty = 0; 21470b58c328SManuel Klimek if (!ContinueOnLine && !Strict) { 214893699f4dSManuel Klimek // Similarly, if the excess characters' penalty is lower than the 214993699f4dSManuel Klimek // penalty of introducing a new break, continue on the current line. 215093699f4dSManuel Klimek ExcessCharactersPenalty = 215193699f4dSManuel Klimek (ContentStartColumn + ToNextSplitColumns - ColumnLimit) * 215277866143SManuel Klimek Style.PenaltyExcessCharacter; 21533538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 215493699f4dSManuel Klimek << " Penalty excess: " << ExcessCharactersPenalty 215593699f4dSManuel Klimek << "\n break : " << NewBreakPenalty << "\n"); 21560b58c328SManuel Klimek if (ExcessCharactersPenalty < NewBreakPenalty) { 21570b58c328SManuel Klimek Exceeded = true; 215893699f4dSManuel Klimek ContinueOnLine = true; 2159875395f9SAlexander Kornienko } 21600b58c328SManuel Klimek } 216193699f4dSManuel Klimek if (ContinueOnLine) { 21623538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Continuing on line...\n"); 216393699f4dSManuel Klimek // The current line fits after compressing the whitespace - reflow 216493699f4dSManuel Klimek // the next line into it if possible. 216593699f4dSManuel Klimek TryReflow = true; 216693699f4dSManuel Klimek if (!DryRun) 216793699f4dSManuel Klimek Token->compressWhitespace(LineIndex, TailOffset, Split, 216893699f4dSManuel Klimek Whitespaces); 216993699f4dSManuel Klimek // When we continue on the same line, leave one space between content. 217093699f4dSManuel Klimek ContentStartColumn += ToSplitColumns + 1; 217193699f4dSManuel Klimek Penalty += ExcessCharactersPenalty; 217293699f4dSManuel Klimek TailOffset += Split.first + Split.second; 217393699f4dSManuel Klimek RemainingTokenColumns = Token->getRemainingLength( 217493699f4dSManuel Klimek LineIndex, TailOffset, ContentStartColumn); 217593699f4dSManuel Klimek continue; 217693699f4dSManuel Klimek } 217793699f4dSManuel Klimek } 21783538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Breaking...\n"); 21796a5c95bdSKrasimir Georgiev // Update the ContentIndent only if the current line was not reflown with 21806a5c95bdSKrasimir Georgiev // the previous line, since in that case the previous line should still 21816a5c95bdSKrasimir Georgiev // determine the ContentIndent. Also never intent the last line. 21826a5c95bdSKrasimir Georgiev if (!Reflow) 21836a5c95bdSKrasimir Georgiev ContentIndent = Token->getContentIndent(LineIndex); 21846a5c95bdSKrasimir Georgiev LLVM_DEBUG(llvm::dbgs() 21856a5c95bdSKrasimir Georgiev << " ContentIndent: " << ContentIndent << "\n"); 21866a5c95bdSKrasimir Georgiev ContentStartColumn = ContentIndent + Token->getContentStartColumn( 21876a5c95bdSKrasimir Georgiev LineIndex, /*Break=*/true); 21886a5c95bdSKrasimir Georgiev 218993699f4dSManuel Klimek unsigned NewRemainingTokenColumns = Token->getRemainingLength( 219093699f4dSManuel Klimek LineIndex, TailOffset + Split.first + Split.second, 219193699f4dSManuel Klimek ContentStartColumn); 21926a5c95bdSKrasimir Georgiev if (NewRemainingTokenColumns == 0) { 21936a5c95bdSKrasimir Georgiev // No content to indent. 21946a5c95bdSKrasimir Georgiev ContentIndent = 0; 21956a5c95bdSKrasimir Georgiev ContentStartColumn = 21966a5c95bdSKrasimir Georgiev Token->getContentStartColumn(LineIndex, /*Break=*/true); 21976a5c95bdSKrasimir Georgiev NewRemainingTokenColumns = Token->getRemainingLength( 21986a5c95bdSKrasimir Georgiev LineIndex, TailOffset + Split.first + Split.second, 21996a5c95bdSKrasimir Georgiev ContentStartColumn); 22006a5c95bdSKrasimir Georgiev } 220191834227SKrasimir Georgiev 220264a42b8dSAlexander Kornienko // When breaking before a tab character, it may be moved by a few columns, 220364a42b8dSAlexander Kornienko // but will still be expanded to the next tab stop, so we don't save any 220464a42b8dSAlexander Kornienko // columns. 220584bf5e32SManuel Klimek if (NewRemainingTokenColumns >= RemainingTokenColumns) { 220677866143SManuel Klimek // FIXME: Do we need to adjust the penalty? 220764a42b8dSAlexander Kornienko break; 220893699f4dSManuel Klimek } 220977866143SManuel Klimek 22103538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Breaking at: " << TailOffset + Split.first 221193699f4dSManuel Klimek << ", " << Split.second << "\n"); 2212de0328aeSDaniel Jasper if (!DryRun) 22136a5c95bdSKrasimir Georgiev Token->insertBreak(LineIndex, TailOffset, Split, ContentIndent, 22146a5c95bdSKrasimir Georgiev Whitespaces); 221577866143SManuel Klimek 221693699f4dSManuel Klimek Penalty += NewBreakPenalty; 2217de0328aeSDaniel Jasper TailOffset += Split.first + Split.second; 2218de0328aeSDaniel Jasper RemainingTokenColumns = NewRemainingTokenColumns; 2219de0328aeSDaniel Jasper BreakInserted = true; 222077866143SManuel Klimek NewBreakBefore = true; 2221de0328aeSDaniel Jasper } 222293699f4dSManuel Klimek // In case there's another line, prepare the state for the start of the next 222393699f4dSManuel Klimek // line. 222493699f4dSManuel Klimek if (LineIndex + 1 != EndIndex) { 222593699f4dSManuel Klimek unsigned NextLineIndex = LineIndex + 1; 222693699f4dSManuel Klimek if (NewBreakBefore) 222793699f4dSManuel Klimek // After breaking a line, try to reflow the next line into the current 222893699f4dSManuel Klimek // one once RemainingTokenColumns fits. 222993699f4dSManuel Klimek TryReflow = true; 223093699f4dSManuel Klimek if (TryReflow) { 223193699f4dSManuel Klimek // We decided that we want to try reflowing the next line into the 223293699f4dSManuel Klimek // current one. 223393699f4dSManuel Klimek // We will now adjust the state as if the reflow is successful (in 223493699f4dSManuel Klimek // preparation for the next line), and see whether that works. If we 223593699f4dSManuel Klimek // decide that we cannot reflow, we will later reset the state to the 223693699f4dSManuel Klimek // start of the next line. 223793699f4dSManuel Klimek Reflow = false; 223893699f4dSManuel Klimek // As we did not continue breaking the line, RemainingTokenColumns is 223993699f4dSManuel Klimek // known to fit after ContentStartColumn. Adapt ContentStartColumn to 224093699f4dSManuel Klimek // the position at which we want to format the next line if we do 224193699f4dSManuel Klimek // actually reflow. 224293699f4dSManuel Klimek // When we reflow, we need to add a space between the end of the current 224393699f4dSManuel Klimek // line and the next line's start column. 224493699f4dSManuel Klimek ContentStartColumn += RemainingTokenColumns + 1; 224593699f4dSManuel Klimek // Get the split that we need to reflow next logical line into the end 224693699f4dSManuel Klimek // of the current one; the split will include any leading whitespace of 224793699f4dSManuel Klimek // the next logical line. 224893699f4dSManuel Klimek BreakableToken::Split SplitBeforeNext = 224993699f4dSManuel Klimek Token->getReflowSplit(NextLineIndex, CommentPragmasRegex); 22503538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 22513538b39eSNicola Zaghen << " Size of reflown text: " << ContentStartColumn 225293699f4dSManuel Klimek << "\n Potential reflow split: "); 225393699f4dSManuel Klimek if (SplitBeforeNext.first != StringRef::npos) { 22543538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << SplitBeforeNext.first << ", " 225593699f4dSManuel Klimek << SplitBeforeNext.second << "\n"); 225693699f4dSManuel Klimek TailOffset = SplitBeforeNext.first + SplitBeforeNext.second; 225793699f4dSManuel Klimek // If the rest of the next line fits into the current line below the 225893699f4dSManuel Klimek // column limit, we can safely reflow. 225993699f4dSManuel Klimek RemainingTokenColumns = Token->getRemainingLength( 226093699f4dSManuel Klimek NextLineIndex, TailOffset, ContentStartColumn); 226193699f4dSManuel Klimek Reflow = true; 226293699f4dSManuel Klimek if (ContentStartColumn + RemainingTokenColumns > ColumnLimit) { 22633538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 22643538b39eSNicola Zaghen << " Over limit after reflow, need: " 226593699f4dSManuel Klimek << (ContentStartColumn + RemainingTokenColumns) 226693699f4dSManuel Klimek << ", space: " << ColumnLimit 226793699f4dSManuel Klimek << ", reflown prefix: " << ContentStartColumn 226893699f4dSManuel Klimek << ", offset in line: " << TailOffset << "\n"); 226993699f4dSManuel Klimek // If the whole next line does not fit, try to find a point in 227093699f4dSManuel Klimek // the next line at which we can break so that attaching the part 227193699f4dSManuel Klimek // of the next line to that break point onto the current line is 227293699f4dSManuel Klimek // below the column limit. 227393699f4dSManuel Klimek BreakableToken::Split Split = 227493699f4dSManuel Klimek Token->getSplit(NextLineIndex, TailOffset, ColumnLimit, 227593699f4dSManuel Klimek ContentStartColumn, CommentPragmasRegex); 227693699f4dSManuel Klimek if (Split.first == StringRef::npos) { 22773538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Did not find later break\n"); 227893699f4dSManuel Klimek Reflow = false; 227993699f4dSManuel Klimek } else { 228093699f4dSManuel Klimek // Check whether the first split point gets us below the column 228193699f4dSManuel Klimek // limit. Note that we will execute this split below as part of 228293699f4dSManuel Klimek // the normal token breaking and reflow logic within the line. 228393699f4dSManuel Klimek unsigned ToSplitColumns = Token->getRangeLength( 228493699f4dSManuel Klimek NextLineIndex, TailOffset, Split.first, ContentStartColumn); 228593699f4dSManuel Klimek if (ContentStartColumn + ToSplitColumns > ColumnLimit) { 22863538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Next split protrudes, need: " 228793699f4dSManuel Klimek << (ContentStartColumn + ToSplitColumns) 228893699f4dSManuel Klimek << ", space: " << ColumnLimit); 228993699f4dSManuel Klimek unsigned ExcessCharactersPenalty = 229093699f4dSManuel Klimek (ContentStartColumn + ToSplitColumns - ColumnLimit) * 229193699f4dSManuel Klimek Style.PenaltyExcessCharacter; 229293699f4dSManuel Klimek if (NewBreakPenalty < ExcessCharactersPenalty) { 229393699f4dSManuel Klimek Reflow = false; 229493699f4dSManuel Klimek } 229593699f4dSManuel Klimek } 229693699f4dSManuel Klimek } 229793699f4dSManuel Klimek } 229893699f4dSManuel Klimek } else { 22993538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << "not found.\n"); 230093699f4dSManuel Klimek } 230193699f4dSManuel Klimek } 230293699f4dSManuel Klimek if (!Reflow) { 230393699f4dSManuel Klimek // If we didn't reflow into the next line, the only space to consider is 230493699f4dSManuel Klimek // the next logical line. Reset our state to match the start of the next 230593699f4dSManuel Klimek // line. 230693699f4dSManuel Klimek TailOffset = 0; 230793699f4dSManuel Klimek ContentStartColumn = 230893699f4dSManuel Klimek Token->getContentStartColumn(NextLineIndex, /*Break=*/false); 230993699f4dSManuel Klimek RemainingTokenColumns = Token->getRemainingLength( 231093699f4dSManuel Klimek NextLineIndex, TailOffset, ContentStartColumn); 231193699f4dSManuel Klimek // Adapt the start of the token, for example indent. 231293699f4dSManuel Klimek if (!DryRun) 231393699f4dSManuel Klimek Token->adaptStartOfLine(NextLineIndex, Whitespaces); 231493699f4dSManuel Klimek } else { 231593699f4dSManuel Klimek // If we found a reflow split and have added a new break before the next 231693699f4dSManuel Klimek // line, we are going to remove the line break at the start of the next 231793699f4dSManuel Klimek // logical line. For example, here we'll add a new line break after 231893699f4dSManuel Klimek // 'text', and subsequently delete the line break between 'that' and 231993699f4dSManuel Klimek // 'reflows'. 232093699f4dSManuel Klimek // // some text that 232193699f4dSManuel Klimek // // reflows 232293699f4dSManuel Klimek // -> 232393699f4dSManuel Klimek // // some text 232493699f4dSManuel Klimek // // that reflows 232593699f4dSManuel Klimek // When adding the line break, we also added the penalty for it, so we 232693699f4dSManuel Klimek // need to subtract that penalty again when we remove the line break due 232793699f4dSManuel Klimek // to reflowing. 232893699f4dSManuel Klimek if (NewBreakBefore) { 232993699f4dSManuel Klimek assert(Penalty >= NewBreakPenalty); 233093699f4dSManuel Klimek Penalty -= NewBreakPenalty; 233193699f4dSManuel Klimek } 233293699f4dSManuel Klimek if (!DryRun) 233393699f4dSManuel Klimek Token->reflow(NextLineIndex, Whitespaces); 233493699f4dSManuel Klimek } 233593699f4dSManuel Klimek } 2336de0328aeSDaniel Jasper } 2337de0328aeSDaniel Jasper 23383b86534aSKrasimir Georgiev BreakableToken::Split SplitAfterLastLine = 233993699f4dSManuel Klimek Token->getSplitAfterLastLine(TailOffset); 234022d7e6b0SKrasimir Georgiev if (SplitAfterLastLine.first != StringRef::npos) { 23413538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << "Replacing whitespace after last line.\n"); 2342123ca80bSKrasimir Georgiev 2343123ca80bSKrasimir Georgiev // We add the last line's penalty here, since that line is going to be split 2344123ca80bSKrasimir Georgiev // now. 2345123ca80bSKrasimir Georgiev Penalty += Style.PenaltyExcessCharacter * 2346123ca80bSKrasimir Georgiev (ContentStartColumn + RemainingTokenColumns - ColumnLimit); 2347123ca80bSKrasimir Georgiev 234822d7e6b0SKrasimir Georgiev if (!DryRun) 234922d7e6b0SKrasimir Georgiev Token->replaceWhitespaceAfterLastLine(TailOffset, SplitAfterLastLine, 235022d7e6b0SKrasimir Georgiev Whitespaces); 235193699f4dSManuel Klimek ContentStartColumn = 235293699f4dSManuel Klimek Token->getContentStartColumn(Token->getLineCount() - 1, /*Break=*/true); 235393699f4dSManuel Klimek RemainingTokenColumns = Token->getRemainingLength( 235493699f4dSManuel Klimek Token->getLineCount() - 1, 235593699f4dSManuel Klimek TailOffset + SplitAfterLastLine.first + SplitAfterLastLine.second, 235693699f4dSManuel Klimek ContentStartColumn); 235722d7e6b0SKrasimir Georgiev } 235822d7e6b0SKrasimir Georgiev 235993699f4dSManuel Klimek State.Column = ContentStartColumn + RemainingTokenColumns - 236093699f4dSManuel Klimek Current.UnbreakableTailLength; 2361de0328aeSDaniel Jasper 2362de0328aeSDaniel Jasper if (BreakInserted) { 2363de0328aeSDaniel Jasper // If we break the token inside a parameter list, we need to break before 2364de0328aeSDaniel Jasper // the next parameter on all levels, so that the next parameter is clearly 2365de0328aeSDaniel Jasper // visible. Line comments already introduce a break. 2366a98b7b01SDaniel Jasper if (Current.isNot(TT_LineComment)) { 2367de0328aeSDaniel Jasper for (unsigned i = 0, e = State.Stack.size(); i != e; ++i) 2368de0328aeSDaniel Jasper State.Stack[i].BreakBeforeParameter = true; 2369de0328aeSDaniel Jasper } 2370de0328aeSDaniel Jasper 237135599fdfSKrasimir Georgiev if (Current.is(TT_BlockComment)) 237235599fdfSKrasimir Georgiev State.NoContinuation = true; 237335599fdfSKrasimir Georgiev 2374de0328aeSDaniel Jasper State.Stack.back().LastSpace = StartColumn; 2375de0328aeSDaniel Jasper } 237691834227SKrasimir Georgiev 237791834227SKrasimir Georgiev Token->updateNextToken(State); 237891834227SKrasimir Georgiev 23790b58c328SManuel Klimek return {Penalty, Exceeded}; 2380de0328aeSDaniel Jasper } 2381de0328aeSDaniel Jasper 23829fe0e8daSDaniel Jasper unsigned ContinuationIndenter::getColumnLimit(const LineState &State) const { 2383de0328aeSDaniel Jasper // In preprocessor directives reserve two chars for trailing " \" 23849fe0e8daSDaniel Jasper return Style.ColumnLimit - (State.Line->InPPDirective ? 2 : 0); 2385de0328aeSDaniel Jasper } 2386de0328aeSDaniel Jasper 2387c39b56feSDaniel Jasper bool ContinuationIndenter::nextIsMultilineString(const LineState &State) { 2388f438cb76SDaniel Jasper const FormatToken &Current = *State.NextToken; 2389a98b7b01SDaniel Jasper if (!Current.isStringLiteral() || Current.is(TT_ImplicitStringLiteral)) 2390f438cb76SDaniel Jasper return false; 2391d7b837e7SAlexander Kornienko // We never consider raw string literals "multiline" for the purpose of 2392c39b56feSDaniel Jasper // AlwaysBreakBeforeMultilineStrings implementation as they are special-cased 2393c39b56feSDaniel Jasper // (see TokenAnnotator::mustBreakBefore(). 2394d7b837e7SAlexander Kornienko if (Current.TokenText.startswith("R\"")) 2395d7b837e7SAlexander Kornienko return false; 239639856b71SAlexander Kornienko if (Current.IsMultiline) 2397d7b837e7SAlexander Kornienko return true; 2398f438cb76SDaniel Jasper if (Current.getNextNonComment() && 239904b6a081SDaniel Jasper Current.getNextNonComment()->isStringLiteral()) 2400f438cb76SDaniel Jasper return true; // Implicit concatenation. 2401374e6de8SKrasimir Georgiev if (Style.ColumnLimit != 0 && Style.BreakStringLiterals && 24026fd5d646SDaniel Jasper State.Column + Current.ColumnWidth + Current.UnbreakableTailLength > 2403f438cb76SDaniel Jasper Style.ColumnLimit) 2404f438cb76SDaniel Jasper return true; // String will be split. 2405d7b837e7SAlexander Kornienko return false; 2406f438cb76SDaniel Jasper } 2407f438cb76SDaniel Jasper 2408de0328aeSDaniel Jasper } // namespace format 2409de0328aeSDaniel Jasper } // namespace clang 2410