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) && 126*bebf7bdfSowenca Style.BreakConstructorInitializers == FormatStyle::BCIS_BeforeComma) { 127ec01cd6eSDaniel Jasper return true; 128*bebf7bdfSowenca } 129ff747be4SKrasimir Georgiev if (Style.Language == FormatStyle::LK_Proto && Current.is(TT_SelectorName)) 130ff747be4SKrasimir Georgiev return true; 131ec01cd6eSDaniel Jasper return Previous.is(tok::comma) && !Current.isTrailingComment() && 1320ef8ee19SAndi-Bogdan Postelnicu ((Previous.isNot(TT_CtorInitializerComma) || 133a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers != 134a6b6d51bSFrancois Ferrand FormatStyle::BCIS_BeforeComma) && 1350ef8ee19SAndi-Bogdan Postelnicu (Previous.isNot(TT_InheritanceComma) || 1366bb103f9SFrancois Ferrand Style.BreakInheritanceList != FormatStyle::BILS_BeforeComma)); 137ec01cd6eSDaniel Jasper } 138ec01cd6eSDaniel Jasper 13926b144ccSKrasimir Georgiev static bool opensProtoMessageField(const FormatToken &LessTok, 14026b144ccSKrasimir Georgiev const FormatStyle &Style) { 14126b144ccSKrasimir Georgiev if (LessTok.isNot(tok::less)) 14226b144ccSKrasimir Georgiev return false; 14326b144ccSKrasimir Georgiev return Style.Language == FormatStyle::LK_TextProto || 14426b144ccSKrasimir Georgiev (Style.Language == FormatStyle::LK_Proto && 14526b144ccSKrasimir Georgiev (LessTok.NestingLevel > 0 || 14626b144ccSKrasimir Georgiev (LessTok.Previous && LessTok.Previous->is(tok::equal)))); 14726b144ccSKrasimir Georgiev } 14826b144ccSKrasimir Georgiev 1499ad83fe7SKrasimir Georgiev // Returns the delimiter of a raw string literal, or None if TokenText is not 1509ad83fe7SKrasimir Georgiev // the text of a raw string literal. The delimiter could be the empty string. 1519ad83fe7SKrasimir Georgiev // For example, the delimiter of R"deli(cont)deli" is deli. 1529ad83fe7SKrasimir Georgiev static llvm::Optional<StringRef> getRawStringDelimiter(StringRef TokenText) { 1539ad83fe7SKrasimir Georgiev if (TokenText.size() < 5 // The smallest raw string possible is 'R"()"'. 154*bebf7bdfSowenca || !TokenText.startswith("R\"") || !TokenText.endswith("\"")) { 1559ad83fe7SKrasimir Georgiev return None; 156*bebf7bdfSowenca } 1579ad83fe7SKrasimir Georgiev 1589ad83fe7SKrasimir Georgiev // A raw string starts with 'R"<delimiter>(' and delimiter is ascii and has 1599ad83fe7SKrasimir Georgiev // size at most 16 by the standard, so the first '(' must be among the first 1609ad83fe7SKrasimir Georgiev // 19 bytes. 1619ad83fe7SKrasimir Georgiev size_t LParenPos = TokenText.substr(0, 19).find_first_of('('); 1629ad83fe7SKrasimir Georgiev if (LParenPos == StringRef::npos) 1639ad83fe7SKrasimir Georgiev return None; 1649ad83fe7SKrasimir Georgiev StringRef Delimiter = TokenText.substr(2, LParenPos - 2); 1659ad83fe7SKrasimir Georgiev 1669ad83fe7SKrasimir Georgiev // Check that the string ends in ')Delimiter"'. 1679ad83fe7SKrasimir Georgiev size_t RParenPos = TokenText.size() - Delimiter.size() - 2; 1689ad83fe7SKrasimir Georgiev if (TokenText[RParenPos] != ')') 1699ad83fe7SKrasimir Georgiev return None; 1709ad83fe7SKrasimir Georgiev if (!TokenText.substr(RParenPos + 1).startswith(Delimiter)) 1719ad83fe7SKrasimir Georgiev return None; 1729ad83fe7SKrasimir Georgiev return Delimiter; 1739ad83fe7SKrasimir Georgiev } 1749ad83fe7SKrasimir Georgiev 175412ed095SKrasimir Georgiev // Returns the canonical delimiter for \p Language, or the empty string if no 176412ed095SKrasimir Georgiev // canonical delimiter is specified. 177412ed095SKrasimir Georgiev static StringRef 178412ed095SKrasimir Georgiev getCanonicalRawStringDelimiter(const FormatStyle &Style, 179412ed095SKrasimir Georgiev FormatStyle::LanguageKind Language) { 180d079995dSMarek Kurdej for (const auto &Format : Style.RawStringFormats) 181412ed095SKrasimir Georgiev if (Format.Language == Language) 182412ed095SKrasimir Georgiev return StringRef(Format.CanonicalDelimiter); 183412ed095SKrasimir Georgiev return ""; 184412ed095SKrasimir Georgiev } 185412ed095SKrasimir Georgiev 1869ad83fe7SKrasimir Georgiev RawStringFormatStyleManager::RawStringFormatStyleManager( 1879ad83fe7SKrasimir Georgiev const FormatStyle &CodeStyle) { 1889ad83fe7SKrasimir Georgiev for (const auto &RawStringFormat : CodeStyle.RawStringFormats) { 1894527f13aSKrasimir Georgiev llvm::Optional<FormatStyle> LanguageStyle = 1904527f13aSKrasimir Georgiev CodeStyle.GetLanguageStyle(RawStringFormat.Language); 1914527f13aSKrasimir Georgiev if (!LanguageStyle) { 1924527f13aSKrasimir Georgiev FormatStyle PredefinedStyle; 1939ad83fe7SKrasimir Georgiev if (!getPredefinedStyle(RawStringFormat.BasedOnStyle, 1944527f13aSKrasimir Georgiev RawStringFormat.Language, &PredefinedStyle)) { 1954527f13aSKrasimir Georgiev PredefinedStyle = getLLVMStyle(); 1964527f13aSKrasimir Georgiev PredefinedStyle.Language = RawStringFormat.Language; 1979ad83fe7SKrasimir Georgiev } 1984527f13aSKrasimir Georgiev LanguageStyle = PredefinedStyle; 1994527f13aSKrasimir Georgiev } 2004527f13aSKrasimir Georgiev LanguageStyle->ColumnLimit = CodeStyle.ColumnLimit; 201630c7360SMarek Kurdej for (StringRef Delimiter : RawStringFormat.Delimiters) 2024527f13aSKrasimir Georgiev DelimiterStyle.insert({Delimiter, *LanguageStyle}); 203630c7360SMarek Kurdej for (StringRef EnclosingFunction : RawStringFormat.EnclosingFunctions) 2042537e220SKrasimir Georgiev EnclosingFunctionStyle.insert({EnclosingFunction, *LanguageStyle}); 2052537e220SKrasimir Georgiev } 2069ad83fe7SKrasimir Georgiev } 2079ad83fe7SKrasimir Georgiev 2089ad83fe7SKrasimir Georgiev llvm::Optional<FormatStyle> 2092537e220SKrasimir Georgiev RawStringFormatStyleManager::getDelimiterStyle(StringRef Delimiter) const { 2109ad83fe7SKrasimir Georgiev auto It = DelimiterStyle.find(Delimiter); 2119ad83fe7SKrasimir Georgiev if (It == DelimiterStyle.end()) 2129ad83fe7SKrasimir Georgiev return None; 2139ad83fe7SKrasimir Georgiev return It->second; 2149ad83fe7SKrasimir Georgiev } 2159ad83fe7SKrasimir Georgiev 2162537e220SKrasimir Georgiev llvm::Optional<FormatStyle> 2172537e220SKrasimir Georgiev RawStringFormatStyleManager::getEnclosingFunctionStyle( 2182537e220SKrasimir Georgiev StringRef EnclosingFunction) const { 2192537e220SKrasimir Georgiev auto It = EnclosingFunctionStyle.find(EnclosingFunction); 2202537e220SKrasimir Georgiev if (It == EnclosingFunctionStyle.end()) 2212537e220SKrasimir Georgiev return None; 2222537e220SKrasimir Georgiev return It->second; 2232537e220SKrasimir Georgiev } 2242537e220SKrasimir Georgiev 225de0328aeSDaniel Jasper ContinuationIndenter::ContinuationIndenter(const FormatStyle &Style, 226d0ec0d62SDaniel Jasper const AdditionalKeywords &Keywords, 227635423e6SEric Liu const SourceManager &SourceMgr, 228de0328aeSDaniel Jasper WhitespaceManager &Whitespaces, 229de0328aeSDaniel Jasper encoding::Encoding Encoding, 230de0328aeSDaniel Jasper bool BinPackInconclusiveFunctions) 231d0ec0d62SDaniel Jasper : Style(Style), Keywords(Keywords), SourceMgr(SourceMgr), 232d0ec0d62SDaniel Jasper Whitespaces(Whitespaces), Encoding(Encoding), 233ce9161a5SAlexander Kornienko BinPackInconclusiveFunctions(BinPackInconclusiveFunctions), 2349ad83fe7SKrasimir Georgiev CommentPragmasRegex(Style.CommentPragmas), RawStringFormats(Style) {} 235de0328aeSDaniel Jasper 2369fe0e8daSDaniel Jasper LineState ContinuationIndenter::getInitialState(unsigned FirstIndent, 2379ad83fe7SKrasimir Georgiev unsigned FirstStartColumn, 2381c5d9df8SDaniel Jasper const AnnotatedLine *Line, 2391c5d9df8SDaniel Jasper bool DryRun) { 240de0328aeSDaniel Jasper LineState State; 2419fe0e8daSDaniel Jasper State.FirstIndent = FirstIndent; 2429ad83fe7SKrasimir Georgiev if (FirstStartColumn && Line->First->NewlinesBefore == 0) 2439ad83fe7SKrasimir Georgiev State.Column = FirstStartColumn; 2449ad83fe7SKrasimir Georgiev else 245de0328aeSDaniel Jasper State.Column = FirstIndent; 246ad47c907SKrasimir Georgiev // With preprocessor directive indentation, the line starts on column 0 247ad47c907SKrasimir Georgiev // since it's indented after the hash, but FirstIndent is set to the 248ad47c907SKrasimir Georgiev // preprocessor indent. 249ad47c907SKrasimir Georgiev if (Style.IndentPPDirectives == FormatStyle::PPDIS_AfterHash && 250ad47c907SKrasimir Georgiev (Line->Type == LT_PreprocessorDirective || 251*bebf7bdfSowenca Line->Type == LT_ImportStatement)) { 252ad47c907SKrasimir Georgiev State.Column = 0; 253*bebf7bdfSowenca } 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)); 25935599fdfSKrasimir Georgiev State.NoContinuation = false; 260de0328aeSDaniel Jasper State.StartOfStringLiteral = 0; 26105cd5868SDaniel Jasper State.StartOfLineLevel = 0; 26205cd5868SDaniel Jasper State.LowestLevelOnLine = 0; 263de0328aeSDaniel Jasper State.IgnoreStackForComparison = false; 264de0328aeSDaniel Jasper 26526b144ccSKrasimir Georgiev if (Style.Language == FormatStyle::LK_TextProto) { 26626b144ccSKrasimir Georgiev // We need this in order to deal with the bin packing of text fields at 26726b144ccSKrasimir Georgiev // global scope. 2688da319feSBjörn Schäpers auto &CurrentState = State.Stack.back(); 2698da319feSBjörn Schäpers CurrentState.AvoidBinPacking = true; 2708da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = true; 2718da319feSBjörn Schäpers CurrentState.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; 2828da319feSBjörn Schäpers const auto &CurrentState = State.Stack.back(); 283de0328aeSDaniel Jasper assert(&Previous == Current.Previous); 2848da319feSBjörn Schäpers if (!Current.CanBreakBefore && !(CurrentState.BreakBeforeClosingBrace && 285*bebf7bdfSowenca Current.closesBlockOrBlockTypeList(Style))) { 286de0328aeSDaniel Jasper return false; 287*bebf7bdfSowenca } 288de0328aeSDaniel Jasper // The opening "{" of a braced list has to be on the same line as the first 289de0328aeSDaniel Jasper // element if it is nested in another braced init list or function call. 290de0328aeSDaniel Jasper if (!Current.MustBreakBefore && Previous.is(tok::l_brace) && 291f5acd11dSBruno Ricci Previous.isNot(TT_DictLiteral) && Previous.is(BK_BracedInit) && 292b05a81deSDaniel Jasper Previous.Previous && 293*bebf7bdfSowenca Previous.Previous->isOneOf(tok::l_brace, tok::l_paren, tok::comma)) { 294de0328aeSDaniel Jasper return false; 295*bebf7bdfSowenca } 296de0328aeSDaniel Jasper // This prevents breaks like: 297de0328aeSDaniel Jasper // ... 298de0328aeSDaniel Jasper // SomeParameter, OtherParameter).DoSomething( 299de0328aeSDaniel Jasper // ... 300de0328aeSDaniel Jasper // As they hide "DoSomething" and are generally bad for readability. 3018f59ae53SDaniel Jasper if (Previous.opensScope() && Previous.isNot(tok::l_brace) && 302fcfac10cSDaniel Jasper State.LowestLevelOnLine < State.StartOfLineLevel && 303*bebf7bdfSowenca State.LowestLevelOnLine < Current.NestingLevel) { 304de0328aeSDaniel Jasper return false; 305*bebf7bdfSowenca } 3068da319feSBjörn Schäpers if (Current.isMemberAccess() && CurrentState.ContainsUnwrappedBuilder) 3074c6e0059SDaniel Jasper return false; 308114a2bc9SDaniel Jasper 309114a2bc9SDaniel Jasper // Don't create a 'hanging' indent if there are multiple blocks in a single 310114a2bc9SDaniel Jasper // statement. 3114b444495SDaniel Jasper if (Previous.is(tok::l_brace) && State.Stack.size() > 1 && 3124b444495SDaniel Jasper State.Stack[State.Stack.size() - 2].NestedBlockInlined && 313*bebf7bdfSowenca State.Stack[State.Stack.size() - 2].HasMultipleNestedBlocks) { 314114a2bc9SDaniel Jasper return false; 315*bebf7bdfSowenca } 316114a2bc9SDaniel Jasper 317e068ac77SDaniel Jasper // Don't break after very short return types (e.g. "void") as that is often 318e068ac77SDaniel Jasper // unexpected. 319448592eeSZachary Turner if (Current.is(TT_FunctionDeclarationName) && State.Column < 6) { 320448592eeSZachary Turner if (Style.AlwaysBreakAfterReturnType == FormatStyle::RTBS_None) 321e068ac77SDaniel Jasper return false; 322448592eeSZachary Turner } 323e068ac77SDaniel Jasper 324240527caSDaniel Jasper // If binary operators are moved to the next line (including commas for some 325240527caSDaniel Jasper // styles of constructor initializers), that's always ok. 326240527caSDaniel Jasper if (!Current.isOneOf(TT_BinaryOperator, tok::comma) && 327*bebf7bdfSowenca CurrentState.NoLineBreakInOperand) { 328240527caSDaniel Jasper return false; 329*bebf7bdfSowenca } 330240527caSDaniel Jasper 331e7c4b590SJacek Olesiak if (Previous.is(tok::l_square) && Previous.is(TT_ObjCMethodExpr)) 332e7c4b590SJacek Olesiak return false; 333e7c4b590SJacek Olesiak 3348da319feSBjörn Schäpers return !CurrentState.NoLineBreak; 335de0328aeSDaniel Jasper } 336de0328aeSDaniel Jasper 337de0328aeSDaniel Jasper bool ContinuationIndenter::mustBreak(const LineState &State) { 338de0328aeSDaniel Jasper const FormatToken &Current = *State.NextToken; 339de0328aeSDaniel Jasper const FormatToken &Previous = *Current.Previous; 3408da319feSBjörn Schäpers const auto &CurrentState = State.Stack.back(); 341fa0118e6SWawha if (Style.BraceWrapping.BeforeLambdaBody && Current.CanBreakBefore && 3425b8ffb41Smydeveloperday Current.is(TT_LambdaLBrace) && Previous.isNot(TT_LineComment)) { 343fa0118e6SWawha auto LambdaBodyLength = getLengthToMatchingParen(Current, State.Stack); 3447d5062c6SMarek Kurdej return LambdaBodyLength > getColumnLimit(State); 345fa0118e6SWawha } 346a98b7b01SDaniel Jasper if (Current.MustBreakBefore || Current.is(TT_InlineASMColon)) 347de0328aeSDaniel Jasper return true; 3488da319feSBjörn Schäpers if (CurrentState.BreakBeforeClosingBrace && 349*bebf7bdfSowenca Current.closesBlockOrBlockTypeList(Style)) { 350de0328aeSDaniel Jasper return true; 351*bebf7bdfSowenca } 3528da319feSBjörn Schäpers if (CurrentState.BreakBeforeClosingParen && Current.is(tok::r_paren)) 353966f24e5SCameron Mulhern return true; 354fb7f5c08SJacek Olesiak if (Style.Language == FormatStyle::LK_ObjC && 35528ecd7f0SKanglei Fang Style.ObjCBreakBeforeNestedBlockParam && 356fb7f5c08SJacek Olesiak Current.ObjCSelectorNameParts > 1 && 357*bebf7bdfSowenca Current.startsSequence(TT_SelectorName, tok::colon, tok::caret)) { 358fb7f5c08SJacek Olesiak return true; 359*bebf7bdfSowenca } 360d1b412aeSJonathan Coe // Avoid producing inconsistent states by requiring breaks where they are not 361d1b412aeSJonathan Coe // permitted for C# generic type constraints. 3628da319feSBjörn Schäpers if (CurrentState.IsCSharpGenericTypeConstraint && 363*bebf7bdfSowenca Previous.isNot(TT_CSharpGenericTypeConstraintComma)) { 364d1b412aeSJonathan Coe return false; 365*bebf7bdfSowenca } 366ec01cd6eSDaniel Jasper if ((startsNextParameter(Current, Style) || Previous.is(tok::semi) || 36700693b08SDaniel Jasper (Previous.is(TT_TemplateCloser) && Current.is(TT_StartOfName) && 3681dbc2105SDaniel Jasper Style.isCpp() && 36906ca0fc6SDaniel Jasper // FIXME: This is a temporary workaround for the case where clang-format 37006ca0fc6SDaniel Jasper // sets BreakBeforeParameter to avoid bin packing and this creates a 37106ca0fc6SDaniel Jasper // completely unnecessary line break after a template type that isn't 37206ca0fc6SDaniel Jasper // line-wrapped. 37306ca0fc6SDaniel Jasper (Previous.NestingLevel == 1 || Style.BinPackParameters)) || 3743e0dcc27SDaniel Jasper (Style.BreakBeforeTernaryOperators && Current.is(TT_ConditionalExpr) && 3753e0dcc27SDaniel Jasper Previous.isNot(tok::question)) || 376165b29e2SDaniel Jasper (!Style.BreakBeforeTernaryOperators && 3773e0dcc27SDaniel Jasper Previous.is(TT_ConditionalExpr))) && 3788da319feSBjörn Schäpers CurrentState.BreakBeforeParameter && !Current.isTrailingComment() && 379*bebf7bdfSowenca !Current.isOneOf(tok::r_paren, tok::r_brace)) { 380de0328aeSDaniel Jasper return true; 381*bebf7bdfSowenca } 3828da319feSBjörn Schäpers if (CurrentState.IsChainedConditional && 3834db94094SFrancois Ferrand ((Style.BreakBeforeTernaryOperators && Current.is(TT_ConditionalExpr) && 3844db94094SFrancois Ferrand Current.is(tok::colon)) || 3854db94094SFrancois Ferrand (!Style.BreakBeforeTernaryOperators && Previous.is(TT_ConditionalExpr) && 386*bebf7bdfSowenca Previous.is(tok::colon)))) { 3874db94094SFrancois Ferrand return true; 388*bebf7bdfSowenca } 389a98b7b01SDaniel Jasper if (((Previous.is(TT_DictLiteral) && Previous.is(tok::l_brace)) || 390ccff4d1eSDaniel Jasper (Previous.is(TT_ArrayInitializerLSquare) && 39126b144ccSKrasimir Georgiev Previous.ParameterCount > 1) || 39226b144ccSKrasimir Georgiev opensProtoMessageField(Previous, Style)) && 393db8804b3SDaniel Jasper Style.ColumnLimit > 0 && 3948b4bc76aSKrasimir Georgiev getLengthToMatchingParen(Previous, State.Stack) + State.Column - 1 > 395*bebf7bdfSowenca getColumnLimit(State)) { 396d489dd34SDaniel Jasper return true; 397*bebf7bdfSowenca } 398a6b6d51bSFrancois Ferrand 399a6b6d51bSFrancois Ferrand const FormatToken &BreakConstructorInitializersToken = 400a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon 401a6b6d51bSFrancois Ferrand ? Previous 402a6b6d51bSFrancois Ferrand : Current; 403a6b6d51bSFrancois Ferrand if (BreakConstructorInitializersToken.is(TT_CtorInitializerColon) && 404a6b6d51bSFrancois Ferrand (State.Column + State.Line->Last->TotalLength - Previous.TotalLength > 4057b259cdaSDaniel Jasper getColumnLimit(State) || 4068da319feSBjörn Schäpers CurrentState.BreakBeforeParameter) && 407a6b6d51bSFrancois Ferrand (Style.AllowShortFunctionsOnASingleLine != FormatStyle::SFS_All || 408a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers != FormatStyle::BCIS_BeforeColon || 409*bebf7bdfSowenca Style.ColumnLimit != 0)) { 4105d2587daSDaniel Jasper return true; 411*bebf7bdfSowenca } 412a6b6d51bSFrancois Ferrand 413fd36f0b5SDaniel Jasper if (Current.is(TT_ObjCMethodExpr) && !Previous.is(TT_SelectorName) && 414*bebf7bdfSowenca State.Line->startsWith(TT_ObjCMethodSpecifier)) { 415fd36f0b5SDaniel Jasper return true; 416*bebf7bdfSowenca } 417da5cca4fSKrasimir Georgiev if (Current.is(TT_SelectorName) && !Previous.is(tok::at) && 4188da319feSBjörn Schäpers CurrentState.ObjCSelectorNameFound && CurrentState.BreakBeforeParameter && 419244022a3SSamuel Giddins (Style.ObjCBreakBeforeNestedBlockParam || 420*bebf7bdfSowenca !Current.startsSequence(TT_SelectorName, tok::colon, tok::caret))) { 4212746a308SDaniel Jasper return true; 422*bebf7bdfSowenca } 423de0328aeSDaniel Jasper 4242aaedd3aSDaniel Jasper unsigned NewLineColumn = getNewLineColumn(State); 425a3cd2164SDaniel Jasper if (Current.isMemberAccess() && Style.ColumnLimit != 0 && 42628024560SDaniel Jasper State.Column + getLengthToNextOperator(Current) > Style.ColumnLimit && 42728024560SDaniel Jasper (State.Column > NewLineColumn || 428*bebf7bdfSowenca Current.NestingLevel < State.StartOfLineLevel)) { 42928024560SDaniel Jasper return true; 430*bebf7bdfSowenca } 43128024560SDaniel Jasper 432e61f9f99SDaniel Jasper if (startsSegmentOfBuilderTypeCall(Current) && 4338da319feSBjörn Schäpers (CurrentState.CallContinuation != 0 || 4348da319feSBjörn Schäpers CurrentState.BreakBeforeParameter) && 43551c868e9SDaniel Jasper // JavaScript is treated different here as there is a frequent pattern: 43651c868e9SDaniel Jasper // SomeFunction(function() { 43751c868e9SDaniel Jasper // ... 43851c868e9SDaniel Jasper // }.bind(...)); 43951c868e9SDaniel Jasper // FIXME: We should find a more generic solution to this problem. 440142e79b8Smydeveloperday !(State.Column <= NewLineColumn && Style.isJavaScript()) && 441*bebf7bdfSowenca !(Previous.closesScopeAfterBlock() && State.Column <= NewLineColumn)) { 442e61f9f99SDaniel Jasper return true; 443*bebf7bdfSowenca } 444e61f9f99SDaniel Jasper 44558e6fe5bSFrancois Ferrand // If the template declaration spans multiple lines, force wrap before the 44658e6fe5bSFrancois Ferrand // function/class declaration 4478da319feSBjörn Schäpers if (Previous.ClosesTemplateDeclaration && CurrentState.BreakBeforeParameter && 448*bebf7bdfSowenca Current.CanBreakBefore) { 44958e6fe5bSFrancois Ferrand return true; 450*bebf7bdfSowenca } 45158e6fe5bSFrancois Ferrand 452072ae7c1Smydeveloperday if (!State.Line->First->is(tok::kw_enum) && State.Column <= NewLineColumn) 4535d2587daSDaniel Jasper return false; 454173504e4SDaniel Jasper 4552aaedd3aSDaniel Jasper if (Style.AlwaysBreakBeforeMultilineStrings && 4561bf729ceSDaniel Jasper (NewLineColumn == State.FirstIndent + Style.ContinuationIndentWidth || 4579fb676a2SDaniel Jasper Previous.is(tok::comma) || Current.NestingLevel < 2) && 458cbb726d0SPaul Hoad !Previous.isOneOf(tok::kw_return, tok::lessless, tok::at, 459cbb726d0SPaul Hoad Keywords.kw_dollar) && 4602aaedd3aSDaniel Jasper !Previous.isOneOf(TT_InlineASMColon, TT_ConditionalExpr) && 461*bebf7bdfSowenca nextIsMultilineString(State)) { 4622aaedd3aSDaniel Jasper return true; 463*bebf7bdfSowenca } 4642aaedd3aSDaniel Jasper 465173504e4SDaniel Jasper // Using CanBreakBefore here and below takes care of the decision whether the 466173504e4SDaniel Jasper // current style uses wrapping before or after operators for the given 467173504e4SDaniel Jasper // operator. 468173504e4SDaniel Jasper if (Previous.is(TT_BinaryOperator) && Current.CanBreakBefore) { 4699b139923SBjörn Schäpers const auto PreviousPrecedence = Previous.getPrecedence(); 4709b139923SBjörn Schäpers if (PreviousPrecedence != prec::Assignment && 4719b139923SBjörn Schäpers CurrentState.BreakBeforeParameter && !Current.isTrailingComment()) { 4729b139923SBjörn Schäpers const bool LHSIsBinaryExpr = 4739b139923SBjörn Schäpers Previous.Previous && Previous.Previous->EndsBinaryExpression; 4749b139923SBjörn Schäpers if (LHSIsBinaryExpr) 4759b139923SBjörn Schäpers return true; 476de0328aeSDaniel Jasper // If we need to break somewhere inside the LHS of a binary expression, we 477de0328aeSDaniel Jasper // should also break after the operator. Otherwise, the formatting would 478de0328aeSDaniel Jasper // hide the operator precedence, e.g. in: 479de0328aeSDaniel Jasper // if (aaaaaaaaaaaaaa == 480de0328aeSDaniel Jasper // bbbbbbbbbbbbbb && c) {.. 481de0328aeSDaniel Jasper // For comparisons, we only apply this rule, if the LHS is a binary 482de0328aeSDaniel Jasper // expression itself as otherwise, the line breaks seem superfluous. 483de0328aeSDaniel Jasper // We need special cases for ">>" which we have split into two ">" while 484de0328aeSDaniel Jasper // lexing in order to make template parsing easier. 4859b139923SBjörn Schäpers const bool IsComparison = 4869b139923SBjörn Schäpers (PreviousPrecedence == prec::Relational || 4879b139923SBjörn Schäpers PreviousPrecedence == prec::Equality || 4889b139923SBjörn Schäpers PreviousPrecedence == prec::Spaceship) && 489de0328aeSDaniel Jasper Previous.Previous && 490a98b7b01SDaniel Jasper Previous.Previous->isNot(TT_BinaryOperator); // For >>. 4919b139923SBjörn Schäpers if (!IsComparison) 492de0328aeSDaniel Jasper return true; 4939b139923SBjörn Schäpers } 494173504e4SDaniel Jasper } else if (Current.is(TT_BinaryOperator) && Current.CanBreakBefore && 4958da319feSBjörn Schäpers CurrentState.BreakBeforeParameter) { 4963219e43cSDaniel Jasper return true; 497de0328aeSDaniel Jasper } 498de0328aeSDaniel Jasper 499de0328aeSDaniel Jasper // Same as above, but for the first "<<" operator. 500a98b7b01SDaniel Jasper if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator) && 501*bebf7bdfSowenca CurrentState.BreakBeforeParameter && CurrentState.FirstLessLess == 0) { 502de0328aeSDaniel Jasper return true; 503*bebf7bdfSowenca } 504de0328aeSDaniel Jasper 505211e1329SDaniel Jasper if (Current.NestingLevel == 0 && !Current.isTrailingComment()) { 5069aab0db1SBjörn Schäpers // Always break after "template <...>"(*) and leading annotations. This is 5079aab0db1SBjörn Schäpers // only for cases where the entire line does not fit on a single line as a 508f090f031SDaniel Jasper // different LineFormatter would be used otherwise. 5099aab0db1SBjörn Schäpers // *: Except when another option interferes with that, like concepts. 5109aab0db1SBjörn Schäpers if (Previous.ClosesTemplateDeclaration) { 5119aab0db1SBjörn Schäpers if (Current.is(tok::kw_concept)) { 5129aab0db1SBjörn Schäpers switch (Style.BreakBeforeConceptDeclarations) { 5139aab0db1SBjörn Schäpers case FormatStyle::BBCDS_Allowed: 5149aab0db1SBjörn Schäpers break; 5159aab0db1SBjörn Schäpers case FormatStyle::BBCDS_Always: 5169aab0db1SBjörn Schäpers return true; 5179aab0db1SBjörn Schäpers case FormatStyle::BBCDS_Never: 5189aab0db1SBjörn Schäpers return false; 5199aab0db1SBjörn Schäpers } 5209aab0db1SBjörn Schäpers } 5219aab0db1SBjörn Schäpers if (Current.is(TT_RequiresClause)) { 5229aab0db1SBjörn Schäpers switch (Style.RequiresClausePosition) { 5239aab0db1SBjörn Schäpers case FormatStyle::RCPS_SingleLine: 5249aab0db1SBjörn Schäpers case FormatStyle::RCPS_WithPreceding: 5259aab0db1SBjörn Schäpers return false; 5269aab0db1SBjörn Schäpers default: 5279aab0db1SBjörn Schäpers return true; 5289aab0db1SBjörn Schäpers } 5299aab0db1SBjörn Schäpers } 53058e6fe5bSFrancois Ferrand return Style.AlwaysBreakTemplateDeclarations != FormatStyle::BTDS_No; 5319aab0db1SBjörn Schäpers } 53247a9eb21SArmen Khachkinaev if (Previous.is(TT_FunctionAnnotationRParen) && 533*bebf7bdfSowenca State.Line->Type != LT_PreprocessorDirective) { 534f090f031SDaniel Jasper return true; 535*bebf7bdfSowenca } 536beb03938SNico Weber if (Previous.is(TT_LeadingJavaAnnotation) && Current.isNot(tok::l_paren) && 537*bebf7bdfSowenca Current.isNot(TT_LeadingJavaAnnotation)) { 538211e1329SDaniel Jasper return true; 539211e1329SDaniel Jasper } 540*bebf7bdfSowenca } 541de0328aeSDaniel Jasper 542142e79b8Smydeveloperday if (Style.isJavaScript() && Previous.is(tok::r_paren) && 543142e79b8Smydeveloperday Previous.is(TT_JavaAnnotation)) { 544befb9dc3SKrasimir Georgiev // Break after the closing parenthesis of TypeScript decorators before 545befb9dc3SKrasimir Georgiev // functions, getters and setters. 546befb9dc3SKrasimir Georgiev static const llvm::StringSet<> BreakBeforeDecoratedTokens = {"get", "set", 547befb9dc3SKrasimir Georgiev "function"}; 5488d1c8545SBjörn Schäpers if (BreakBeforeDecoratedTokens.contains(Current.TokenText)) 549f3671a68SKrasimir Georgiev return true; 550f3671a68SKrasimir Georgiev } 551f3671a68SKrasimir Georgiev 5524355e7f0SDaniel Jasper // If the return type spans multiple lines, wrap before the function name. 553a2f963bbSPaul Hoad if (((Current.is(TT_FunctionDeclarationName) && 554a2f963bbSPaul Hoad // Don't break before a C# function when no break after return type 555a2f963bbSPaul Hoad (!Style.isCSharp() || 5564f4f2783SKrasimir Georgiev Style.AlwaysBreakAfterReturnType != FormatStyle::RTBS_None) && 5574f4f2783SKrasimir Georgiev // Don't always break between a JavaScript `function` and the function 5584f4f2783SKrasimir Georgiev // name. 559142e79b8Smydeveloperday !Style.isJavaScript()) || 560525579d4SBirunthan Mohanathas (Current.is(tok::kw_operator) && !Previous.is(tok::coloncolon))) && 561*bebf7bdfSowenca !Previous.is(tok::kw_template) && CurrentState.BreakBeforeParameter) { 562de0328aeSDaniel Jasper return true; 563*bebf7bdfSowenca } 5644355e7f0SDaniel Jasper 5659697281eSDaniel Jasper // The following could be precomputed as they do not depend on the state. 5669697281eSDaniel Jasper // However, as they should take effect only if the UnwrappedLine does not fit 5679697281eSDaniel Jasper // into the ColumnLimit, they are checked here in the ContinuationIndenter. 568f5acd11dSBruno Ricci if (Style.ColumnLimit != 0 && Previous.is(BK_Block) && 569*bebf7bdfSowenca Previous.is(tok::l_brace) && 570*bebf7bdfSowenca !Current.isOneOf(tok::r_brace, tok::comment)) { 5719697281eSDaniel Jasper return true; 572*bebf7bdfSowenca } 5739697281eSDaniel Jasper 5740a589416SDaniel Jasper if (Current.is(tok::lessless) && 5750a589416SDaniel Jasper ((Previous.is(tok::identifier) && Previous.TokenText == "endl") || 5760a589416SDaniel Jasper (Previous.Tok.isLiteral() && (Previous.TokenText.endswith("\\n\"") || 577*bebf7bdfSowenca Previous.TokenText == "\'\\n\'")))) { 57869963129SDaniel Jasper return true; 579*bebf7bdfSowenca } 58069963129SDaniel Jasper 58135599fdfSKrasimir Georgiev if (Previous.is(TT_BlockComment) && Previous.IsMultiline) 58235599fdfSKrasimir Georgiev return true; 58335599fdfSKrasimir Georgiev 58435599fdfSKrasimir Georgiev if (State.NoContinuation) 58535599fdfSKrasimir Georgiev return true; 58635599fdfSKrasimir Georgiev 587de0328aeSDaniel Jasper return false; 588de0328aeSDaniel Jasper } 589de0328aeSDaniel Jasper 590de0328aeSDaniel Jasper unsigned ContinuationIndenter::addTokenToState(LineState &State, bool Newline, 5918de9ed05SDaniel Jasper bool DryRun, 5928de9ed05SDaniel Jasper unsigned ExtraSpaces) { 593de0328aeSDaniel Jasper const FormatToken &Current = *State.NextToken; 59481793bd2SMarek Kurdej assert(State.NextToken->Previous); 59581793bd2SMarek Kurdej const FormatToken &Previous = *State.NextToken->Previous; 596de0328aeSDaniel Jasper 597819788daSManuel Klimek assert(!State.Stack.empty()); 59835599fdfSKrasimir Georgiev State.NoContinuation = false; 59935599fdfSKrasimir Georgiev 600a98b7b01SDaniel Jasper if ((Current.is(TT_ImplicitStringLiteral) && 60181793bd2SMarek Kurdej (Previous.Tok.getIdentifierInfo() == nullptr || 60281793bd2SMarek Kurdej Previous.Tok.getIdentifierInfo()->getPPKeywordID() == 6039885784dSDaniel Jasper tok::pp_not_keyword))) { 60430526e79SDaniel Jasper unsigned EndColumn = 60530526e79SDaniel Jasper SourceMgr.getSpellingColumnNumber(Current.WhitespaceRange.getEnd()); 60630526e79SDaniel Jasper if (Current.LastNewlineOffset != 0) { 60730526e79SDaniel Jasper // If there is a newline within this token, the final column will solely 60830526e79SDaniel Jasper // determined by the current end column. 60930526e79SDaniel Jasper State.Column = EndColumn; 61030526e79SDaniel Jasper } else { 61130526e79SDaniel Jasper unsigned StartColumn = 61230526e79SDaniel Jasper SourceMgr.getSpellingColumnNumber(Current.WhitespaceRange.getBegin()); 61330526e79SDaniel Jasper assert(EndColumn >= StartColumn); 61430526e79SDaniel Jasper State.Column += EndColumn - StartColumn; 61530526e79SDaniel Jasper } 616240dfda3SDaniel Jasper moveStateToNextToken(State, DryRun, /*Newline=*/false); 617de0328aeSDaniel Jasper return 0; 618de0328aeSDaniel Jasper } 619de0328aeSDaniel Jasper 6201f80396dSAlexander Kornienko unsigned Penalty = 0; 6211f80396dSAlexander Kornienko if (Newline) 6221f80396dSAlexander Kornienko Penalty = addTokenOnNewLine(State, DryRun); 6231f80396dSAlexander Kornienko else 62448437ce5SDaniel Jasper addTokenOnCurrentLine(State, DryRun, ExtraSpaces); 6251f80396dSAlexander Kornienko 6261f80396dSAlexander Kornienko return moveStateToNextToken(State, DryRun, Newline) + Penalty; 6271f80396dSAlexander Kornienko } 6281f80396dSAlexander Kornienko 62948437ce5SDaniel Jasper void ContinuationIndenter::addTokenOnCurrentLine(LineState &State, bool DryRun, 6301f80396dSAlexander Kornienko unsigned ExtraSpaces) { 63171814b44SManuel Klimek FormatToken &Current = *State.NextToken; 63281793bd2SMarek Kurdej assert(State.NextToken->Previous); 6331f80396dSAlexander Kornienko const FormatToken &Previous = *State.NextToken->Previous; 6348da319feSBjörn Schäpers auto &CurrentState = State.Stack.back(); 63581793bd2SMarek Kurdej 6361f80396dSAlexander Kornienko if (Current.is(tok::equal) && 63705cd5868SDaniel Jasper (State.Line->First->is(tok::kw_for) || Current.NestingLevel == 0) && 6388da319feSBjörn Schäpers CurrentState.VariablePos == 0) { 6398da319feSBjörn Schäpers CurrentState.VariablePos = State.Column; 6401f80396dSAlexander Kornienko // Move over * and & if they are bound to the variable name. 6411f80396dSAlexander Kornienko const FormatToken *Tok = &Previous; 6428da319feSBjörn Schäpers while (Tok && CurrentState.VariablePos >= Tok->ColumnWidth) { 6438da319feSBjörn Schäpers CurrentState.VariablePos -= Tok->ColumnWidth; 6441f80396dSAlexander Kornienko if (Tok->SpacesRequiredBefore != 0) 6451f80396dSAlexander Kornienko break; 6461f80396dSAlexander Kornienko Tok = Tok->Previous; 6471f80396dSAlexander Kornienko } 6481f80396dSAlexander Kornienko if (Previous.PartOfMultiVariableDeclStmt) 6498da319feSBjörn Schäpers CurrentState.LastSpace = CurrentState.VariablePos; 6501f80396dSAlexander Kornienko } 6511f80396dSAlexander Kornienko 6521f80396dSAlexander Kornienko unsigned Spaces = Current.SpacesRequiredBefore + ExtraSpaces; 6531f80396dSAlexander Kornienko 654ad47c907SKrasimir Georgiev // Indent preprocessor directives after the hash if required. 655ad47c907SKrasimir Georgiev int PPColumnCorrection = 0; 656ad47c907SKrasimir Georgiev if (Style.IndentPPDirectives == FormatStyle::PPDIS_AfterHash && 657ad47c907SKrasimir Georgiev Previous.is(tok::hash) && State.FirstIndent > 0 && 658ad47c907SKrasimir Georgiev (State.Line->Type == LT_PreprocessorDirective || 659ad47c907SKrasimir Georgiev State.Line->Type == LT_ImportStatement)) { 660ad47c907SKrasimir Georgiev Spaces += State.FirstIndent; 661ad47c907SKrasimir Georgiev 662ad47c907SKrasimir Georgiev // For preprocessor indent with tabs, State.Column will be 1 because of the 663ad47c907SKrasimir Georgiev // hash. This causes second-level indents onward to have an extra space 664ad47c907SKrasimir Georgiev // after the tabs. We avoid this misalignment by subtracting 1 from the 665ad47c907SKrasimir Georgiev // column value passed to replaceWhitespace(). 666ad47c907SKrasimir Georgiev if (Style.UseTab != FormatStyle::UT_Never) 667ad47c907SKrasimir Georgiev PPColumnCorrection = -1; 668ad47c907SKrasimir Georgiev } 669ad47c907SKrasimir Georgiev 670*bebf7bdfSowenca if (!DryRun) { 6717d42f3f7SDaniel Jasper Whitespaces.replaceWhitespace(Current, /*Newlines=*/0, Spaces, 672ad47c907SKrasimir Georgiev State.Column + Spaces + PPColumnCorrection); 673*bebf7bdfSowenca } 6741f80396dSAlexander Kornienko 6750ef8ee19SAndi-Bogdan Postelnicu // If "BreakBeforeInheritanceComma" mode, don't break within the inheritance 6760ef8ee19SAndi-Bogdan Postelnicu // declaration unless there is multiple inheritance. 6776bb103f9SFrancois Ferrand if (Style.BreakInheritanceList == FormatStyle::BILS_BeforeComma && 678*bebf7bdfSowenca Current.is(TT_InheritanceColon)) { 6798da319feSBjörn Schäpers CurrentState.NoLineBreak = true; 680*bebf7bdfSowenca } 6816bb103f9SFrancois Ferrand if (Style.BreakInheritanceList == FormatStyle::BILS_AfterColon && 682*bebf7bdfSowenca Previous.is(TT_InheritanceColon)) { 6838da319feSBjörn Schäpers CurrentState.NoLineBreak = true; 684*bebf7bdfSowenca } 6850ef8ee19SAndi-Bogdan Postelnicu 6868da319feSBjörn Schäpers if (Current.is(TT_SelectorName) && !CurrentState.ObjCSelectorNameFound) { 6878da319feSBjörn Schäpers unsigned MinIndent = std::max( 6888da319feSBjörn Schäpers State.FirstIndent + Style.ContinuationIndentWidth, CurrentState.Indent); 68906a26957SDaniel Jasper unsigned FirstColonPos = State.Column + Spaces + Current.ColumnWidth; 690b88b25feSDaniel Jasper if (Current.LongestObjCSelectorName == 0) 6918da319feSBjörn Schäpers CurrentState.AlignColons = false; 69206a26957SDaniel Jasper else if (MinIndent + Current.LongestObjCSelectorName > FirstColonPos) 6938da319feSBjörn Schäpers CurrentState.ColonPos = MinIndent + Current.LongestObjCSelectorName; 6941f80396dSAlexander Kornienko else 6958da319feSBjörn Schäpers CurrentState.ColonPos = FirstColonPos; 6961f80396dSAlexander Kornienko } 6971f80396dSAlexander Kornienko 698966f24e5SCameron Mulhern // In "AlwaysBreak" or "BlockIndent" mode, enforce wrapping directly after the 699966f24e5SCameron Mulhern // parenthesis by disallowing any further line breaks if there is no line 700966f24e5SCameron Mulhern // break after the opening parenthesis. Don't break if it doesn't conserve 701966f24e5SCameron Mulhern // columns. 702966f24e5SCameron Mulhern if ((Style.AlignAfterOpenBracket == FormatStyle::BAS_AlwaysBreak || 703966f24e5SCameron Mulhern Style.AlignAfterOpenBracket == FormatStyle::BAS_BlockIndent) && 704d40ddb9dSOwen Pan (Previous.isOneOf(tok::l_paren, TT_TemplateOpener, tok::l_square) || 705f5acd11dSBruno Ricci (Previous.is(tok::l_brace) && Previous.isNot(BK_Block) && 706d40ddb9dSOwen Pan Style.Cpp11BracedListStyle)) && 707b618a985SDaniel Jasper State.Column > getNewLineColumn(State) && 70889628f64SManuel Klimek (!Previous.Previous || !Previous.Previous->isOneOf( 70989628f64SManuel Klimek tok::kw_for, tok::kw_while, tok::kw_switch)) && 710710f8493SDaniel Jasper // Don't do this for simple (no expressions) one-argument function calls 711710f8493SDaniel Jasper // as that feels like needlessly wasting whitespace, e.g.: 712710f8493SDaniel Jasper // 713710f8493SDaniel Jasper // caaaaaaaaaaaall( 714710f8493SDaniel Jasper // caaaaaaaaaaaall( 715710f8493SDaniel Jasper // caaaaaaaaaaaall( 716710f8493SDaniel Jasper // caaaaaaaaaaaaaaaaaaaaaaall(aaaaaaaaaaaaaa, aaaaaaaaa)))); 717710f8493SDaniel Jasper Current.FakeLParens.size() > 0 && 718*bebf7bdfSowenca Current.FakeLParens.back() > prec::Unknown) { 7198da319feSBjörn Schäpers CurrentState.NoLineBreak = true; 720*bebf7bdfSowenca } 72198e0b12dSDaniel Jasper if (Previous.is(TT_TemplateString) && Previous.opensScope()) 7228da319feSBjörn Schäpers CurrentState.NoLineBreak = true; 7236501f7e8SDaniel Jasper 7246501f7e8SDaniel Jasper if (Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign && 7258da319feSBjörn Schäpers !CurrentState.IsCSharpGenericTypeConstraint && Previous.opensScope() && 7268da319feSBjörn Schäpers Previous.isNot(TT_ObjCMethodExpr) && Previous.isNot(TT_RequiresClause) && 727f5acd11dSBruno Ricci (Current.isNot(TT_LineComment) || Previous.is(BK_BracedInit))) { 7288da319feSBjörn Schäpers CurrentState.Indent = State.Column + Spaces; 7298da319feSBjörn Schäpers CurrentState.IsAligned = true; 730e8111502Smydeveloperday } 7318da319feSBjörn Schäpers if (CurrentState.AvoidBinPacking && startsNextParameter(Current, Style)) 7328da319feSBjörn Schäpers CurrentState.NoLineBreak = true; 733775954beSDaniel Jasper if (startsSegmentOfBuilderTypeCall(Current) && 734*bebf7bdfSowenca State.Column > getNewLineColumn(State)) { 7358da319feSBjörn Schäpers CurrentState.ContainsUnwrappedBuilder = true; 736*bebf7bdfSowenca } 7371f80396dSAlexander Kornienko 7386f2b88a3SDaniel Jasper if (Current.is(TT_LambdaArrow) && Style.Language == FormatStyle::LK_Java) 7398da319feSBjörn Schäpers CurrentState.NoLineBreak = true; 740d6f17d83SDaniel Jasper if (Current.isMemberAccess() && Previous.is(tok::r_paren) && 741d6f17d83SDaniel Jasper (Previous.MatchingParen && 742*bebf7bdfSowenca (Previous.TotalLength - Previous.MatchingParen->TotalLength > 10))) { 743d6f17d83SDaniel Jasper // If there is a function call with long parameters, break before trailing 744d6f17d83SDaniel Jasper // calls. This prevents things like: 745d6f17d83SDaniel Jasper // EXPECT_CALL(SomeLongParameter).Times( 746d6f17d83SDaniel Jasper // 2); 747d6f17d83SDaniel Jasper // We don't want to do this for short parameters as they can just be 748d6f17d83SDaniel Jasper // indexes. 7498da319feSBjörn Schäpers CurrentState.NoLineBreak = true; 750*bebf7bdfSowenca } 751d6f17d83SDaniel Jasper 752240527caSDaniel Jasper // Don't allow the RHS of an operator to be split over multiple lines unless 753240527caSDaniel Jasper // there is a line-break right after the operator. 754240527caSDaniel Jasper // Exclude relational operators, as there, it is always more desirable to 755240527caSDaniel Jasper // have the LHS 'left' of the RHS. 756240527caSDaniel Jasper const FormatToken *P = Current.getPreviousNonComment(); 757240527caSDaniel Jasper if (!Current.is(tok::comment) && P && 758240527caSDaniel Jasper (P->isOneOf(TT_BinaryOperator, tok::comma) || 759240527caSDaniel Jasper (P->is(TT_ConditionalExpr) && P->is(tok::colon))) && 760240527caSDaniel Jasper !P->isOneOf(TT_OverloadedOperator, TT_CtorInitializerComma) && 761240527caSDaniel Jasper P->getPrecedence() != prec::Assignment && 762c70f1d63SRichard Smith P->getPrecedence() != prec::Relational && 763c70f1d63SRichard Smith P->getPrecedence() != prec::Spaceship) { 764240527caSDaniel Jasper bool BreakBeforeOperator = 765b127039cSDaniel Jasper P->MustBreakBefore || P->is(tok::lessless) || 766240527caSDaniel Jasper (P->is(TT_BinaryOperator) && 767240527caSDaniel Jasper Style.BreakBeforeBinaryOperators != FormatStyle::BOS_None) || 768240527caSDaniel Jasper (P->is(TT_ConditionalExpr) && Style.BreakBeforeTernaryOperators); 76921f7dea5SDaniel Jasper // Don't do this if there are only two operands. In these cases, there is 77021f7dea5SDaniel Jasper // always a nice vertical separation between them and the extra line break 77121f7dea5SDaniel Jasper // does not help. 77221f7dea5SDaniel Jasper bool HasTwoOperands = 77321f7dea5SDaniel Jasper P->OperatorIndex == 0 && !P->NextOperator && !P->is(TT_ConditionalExpr); 7740ee04e6eSFrancois Ferrand if ((!BreakBeforeOperator && 7750ee04e6eSFrancois Ferrand !(HasTwoOperands && 7760ee04e6eSFrancois Ferrand Style.AlignOperands != FormatStyle::OAS_DontAlign)) || 777*bebf7bdfSowenca (!CurrentState.LastOperatorWrapped && BreakBeforeOperator)) { 7788da319feSBjörn Schäpers CurrentState.NoLineBreakInOperand = true; 779240527caSDaniel Jasper } 780*bebf7bdfSowenca } 781240527caSDaniel Jasper 7821f80396dSAlexander Kornienko State.Column += Spaces; 7838acf822bSDaniel Jasper if (Current.isNot(tok::comment) && Previous.is(tok::l_paren) && 78411a0ac66SDaniel Jasper Previous.Previous && 7851361a4c2SNico Weber (Previous.Previous->is(tok::kw_for) || Previous.Previous->isIf())) { 7861f80396dSAlexander Kornienko // Treat the condition inside an if as if it was a second function 7876633ab8aSDaniel Jasper // parameter, i.e. let nested calls have a continuation indent. 7888da319feSBjörn Schäpers CurrentState.LastSpace = State.Column; 7898da319feSBjörn Schäpers CurrentState.NestedBlockIndent = State.Column; 79011a0ac66SDaniel Jasper } else if (!Current.isOneOf(tok::comment, tok::caret) && 791804a276fSDaniel Jasper ((Previous.is(tok::comma) && 792804a276fSDaniel Jasper !Previous.is(TT_OverloadedOperator)) || 79311a0ac66SDaniel Jasper (Previous.is(tok::colon) && Previous.is(TT_ObjCMethodExpr)))) { 7948da319feSBjörn Schäpers CurrentState.LastSpace = State.Column; 795a6b6d51bSFrancois Ferrand } else if (Previous.is(TT_CtorInitializerColon) && 796a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers == 797a6b6d51bSFrancois Ferrand FormatStyle::BCIS_AfterColon) { 7988da319feSBjörn Schäpers CurrentState.Indent = State.Column; 7998da319feSBjörn Schäpers CurrentState.LastSpace = State.Column; 80011a0ac66SDaniel Jasper } else if ((Previous.isOneOf(TT_BinaryOperator, TT_ConditionalExpr, 801a98b7b01SDaniel Jasper TT_CtorInitializerColon)) && 8020e61784aSDaniel Jasper ((Previous.getPrecedence() != prec::Assignment && 8030e61784aSDaniel Jasper (Previous.isNot(tok::lessless) || Previous.OperatorIndex != 0 || 80400492f96SDaniel Jasper Previous.NextOperator)) || 80511a0ac66SDaniel Jasper Current.StartsBinaryExpression)) { 806602a727aSDaniel Jasper // Indent relative to the RHS of the expression unless this is a simple 807602a727aSDaniel Jasper // assignment without binary expression on the RHS. Also indent relative to 808602a727aSDaniel Jasper // unary operators and the colons of constructor initializers. 8094636debcSJonas Toth if (Style.BreakBeforeBinaryOperators == FormatStyle::BOS_None) 8108da319feSBjörn Schäpers CurrentState.LastSpace = State.Column; 81111a0ac66SDaniel Jasper } else if (Previous.is(TT_InheritanceColon)) { 8128da319feSBjörn Schäpers CurrentState.Indent = State.Column; 8138da319feSBjörn Schäpers CurrentState.LastSpace = State.Column; 81478e2a3c6SJonathan Coe } else if (Current.is(TT_CSharpGenericTypeConstraintColon)) { 8158da319feSBjörn Schäpers CurrentState.ColonPos = State.Column; 816f9a5e404SDaniel Jasper } else if (Previous.opensScope()) { 8171f80396dSAlexander Kornienko // If a function has a trailing call, indent all parameters from the 8181f80396dSAlexander Kornienko // opening parenthesis. This avoids confusing indents like: 8191f80396dSAlexander Kornienko // OuterFunction(InnerFunctionCall( // break 8201f80396dSAlexander Kornienko // ParameterToInnerFunction)) // break 8211f80396dSAlexander Kornienko // .SecondInnerFunctionCall(); 8221f80396dSAlexander Kornienko if (Previous.MatchingParen) { 8231f80396dSAlexander Kornienko const FormatToken *Next = Previous.MatchingParen->getNextNonComment(); 8247ebb00a2SBjörn Schäpers if (Next && Next->isMemberAccess() && State.Stack.size() > 1 && 825*bebf7bdfSowenca State.Stack[State.Stack.size() - 2].CallContinuation == 0) { 8268da319feSBjörn Schäpers CurrentState.LastSpace = State.Column; 8271f80396dSAlexander Kornienko } 8281f80396dSAlexander Kornienko } 8297ebb00a2SBjörn Schäpers } 830*bebf7bdfSowenca } 8311f80396dSAlexander Kornienko 8321f80396dSAlexander Kornienko unsigned ContinuationIndenter::addTokenOnNewLine(LineState &State, 8331f80396dSAlexander Kornienko bool DryRun) { 83471814b44SManuel Klimek FormatToken &Current = *State.NextToken; 83581793bd2SMarek Kurdej assert(State.NextToken->Previous); 8361f80396dSAlexander Kornienko const FormatToken &Previous = *State.NextToken->Previous; 8378da319feSBjörn Schäpers auto &CurrentState = State.Stack.back(); 8389f388d04SDaniel Jasper 8391f80396dSAlexander Kornienko // Extra penalty that needs to be added because of the way certain line 8401f80396dSAlexander Kornienko // breaks are chosen. 8411f80396dSAlexander Kornienko unsigned Penalty = 0; 8421f80396dSAlexander Kornienko 843a0407740SDaniel Jasper const FormatToken *PreviousNonComment = Current.getPreviousNonComment(); 844a0407740SDaniel Jasper const FormatToken *NextNonComment = Previous.getNextNonComment(); 845a0407740SDaniel Jasper if (!NextNonComment) 846a0407740SDaniel Jasper NextNonComment = &Current; 84705cd5868SDaniel Jasper // The first line break on any NestingLevel causes an extra penalty in order 8488de9ed05SDaniel Jasper // prefer similar line breaks. 8498da319feSBjörn Schäpers if (!CurrentState.ContainsLineBreak) 8508de9ed05SDaniel Jasper Penalty += 15; 8518da319feSBjörn Schäpers CurrentState.ContainsLineBreak = true; 8528de9ed05SDaniel Jasper 8538de9ed05SDaniel Jasper Penalty += State.NextToken->SplitPenalty; 8548de9ed05SDaniel Jasper 855de0328aeSDaniel Jasper // Breaking before the first "<<" is generally not desirable if the LHS is 8567aacf468SDaniel Jasper // short. Also always add the penalty if the LHS is split over multiple lines 8572b7556ecSDaniel Jasper // to avoid unnecessary line breaks that just work around this penalty. 8588da319feSBjörn Schäpers if (NextNonComment->is(tok::lessless) && CurrentState.FirstLessLess == 0 && 859004177eeSDaniel Jasper (State.Column <= Style.ColumnLimit / 3 || 860*bebf7bdfSowenca CurrentState.BreakBeforeParameter)) { 8618de9ed05SDaniel Jasper Penalty += Style.PenaltyBreakFirstLessLess; 862*bebf7bdfSowenca } 863de0328aeSDaniel Jasper 8649f388d04SDaniel Jasper State.Column = getNewLineColumn(State); 865ed3f3957SDaniel Jasper 8661e4d6d1cSMark Nauwelaerts // Add Penalty proportional to amount of whitespace away from FirstColumn 8671e4d6d1cSMark Nauwelaerts // This tends to penalize several lines that are far-right indented, 8681e4d6d1cSMark Nauwelaerts // and prefers a line-break prior to such a block, e.g: 8691e4d6d1cSMark Nauwelaerts // 8701e4d6d1cSMark Nauwelaerts // Constructor() : 8711e4d6d1cSMark Nauwelaerts // member(value), looooooooooooooooong_member( 8721e4d6d1cSMark Nauwelaerts // looooooooooong_call(param_1, param_2, param_3)) 8731e4d6d1cSMark Nauwelaerts // would then become 8741e4d6d1cSMark Nauwelaerts // Constructor() : 8751e4d6d1cSMark Nauwelaerts // member(value), 8761e4d6d1cSMark Nauwelaerts // looooooooooooooooong_member( 8771e4d6d1cSMark Nauwelaerts // looooooooooong_call(param_1, param_2, param_3)) 878*bebf7bdfSowenca if (State.Column > State.FirstIndent) { 8791e4d6d1cSMark Nauwelaerts Penalty += 8801e4d6d1cSMark Nauwelaerts Style.PenaltyIndentedWhitespace * (State.Column - State.FirstIndent); 881*bebf7bdfSowenca } 8821e4d6d1cSMark Nauwelaerts 883ed3f3957SDaniel Jasper // Indent nested blocks relative to this column, unless in a very specific 884ed3f3957SDaniel Jasper // JavaScript special case where: 885ed3f3957SDaniel Jasper // 886ed3f3957SDaniel Jasper // var loooooong_name = 887ed3f3957SDaniel Jasper // function() { 888ed3f3957SDaniel Jasper // // code 889ed3f3957SDaniel Jasper // } 890ed3f3957SDaniel Jasper // 89187448c55SDaniel Jasper // is common and should be formatted like a free-standing function. The same 89287448c55SDaniel Jasper // goes for wrapping before the lambda return type arrow. 89387448c55SDaniel Jasper if (!Current.is(TT_LambdaArrow) && 894142e79b8Smydeveloperday (!Style.isJavaScript() || Current.NestingLevel != 0 || 895142e79b8Smydeveloperday !PreviousNonComment || !PreviousNonComment->is(tok::equal) || 896*bebf7bdfSowenca !Current.isOneOf(Keywords.kw_async, Keywords.kw_function))) { 8978da319feSBjörn Schäpers CurrentState.NestedBlockIndent = State.Column; 898*bebf7bdfSowenca } 899ed3f3957SDaniel Jasper 9009f388d04SDaniel Jasper if (NextNonComment->isMemberAccess()) { 9018da319feSBjörn Schäpers if (CurrentState.CallContinuation == 0) 9028da319feSBjörn Schäpers CurrentState.CallContinuation = State.Column; 903a98b7b01SDaniel Jasper } else if (NextNonComment->is(TT_SelectorName)) { 9048da319feSBjörn Schäpers if (!CurrentState.ObjCSelectorNameFound) { 905a0407740SDaniel Jasper if (NextNonComment->LongestObjCSelectorName == 0) { 9068da319feSBjörn Schäpers CurrentState.AlignColons = false; 907b88b25feSDaniel Jasper } else { 9088da319feSBjörn Schäpers CurrentState.ColonPos = 909416348efSBen Hamilton (shouldIndentWrappedSelectorName(Style, State.Line->Type) 9108da319feSBjörn Schäpers ? std::max(CurrentState.Indent, 911a2a4d9c0SDaniel Jasper State.FirstIndent + Style.ContinuationIndentWidth) 9128da319feSBjörn Schäpers : CurrentState.Indent) + 91338d80134SFrancois Ferrand std::max(NextNonComment->LongestObjCSelectorName, 91438d80134SFrancois Ferrand NextNonComment->ColumnWidth); 915b88b25feSDaniel Jasper } 9168da319feSBjörn Schäpers } else if (CurrentState.AlignColons && 9178da319feSBjörn Schäpers CurrentState.ColonPos <= NextNonComment->ColumnWidth) { 9188da319feSBjörn Schäpers CurrentState.ColonPos = State.Column + NextNonComment->ColumnWidth; 919de0328aeSDaniel Jasper } 9201fd6f1f8SDaniel Jasper } else if (PreviousNonComment && PreviousNonComment->is(tok::colon) && 921a98b7b01SDaniel Jasper PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) { 922b88b25feSDaniel Jasper // FIXME: This is hacky, find a better way. The problem is that in an ObjC 923b88b25feSDaniel Jasper // method expression, the block should be aligned to the line starting it, 924b88b25feSDaniel Jasper // e.g.: 925b88b25feSDaniel Jasper // [aaaaaaaaaaaaaaa aaaaaaaaa: \\ break for some reason 926b88b25feSDaniel Jasper // ^(int *i) { 927b88b25feSDaniel Jasper // // ... 928b88b25feSDaniel Jasper // }]; 92905cd5868SDaniel Jasper // Thus, we set LastSpace of the next higher NestingLevel, to which we move 930b88b25feSDaniel Jasper // when we consume all of the "}"'s FakeRParens at the "{". 931*bebf7bdfSowenca if (State.Stack.size() > 1) { 9329f388d04SDaniel Jasper State.Stack[State.Stack.size() - 2].LastSpace = 9338da319feSBjörn Schäpers std::max(CurrentState.LastSpace, CurrentState.Indent) + 9349f388d04SDaniel Jasper Style.ContinuationIndentWidth; 935de0328aeSDaniel Jasper } 936*bebf7bdfSowenca } 937de0328aeSDaniel Jasper 93835e4122fSDaniel Jasper if ((PreviousNonComment && 93935e4122fSDaniel Jasper PreviousNonComment->isOneOf(tok::comma, tok::semi) && 9408da319feSBjörn Schäpers !CurrentState.AvoidBinPacking) || 941*bebf7bdfSowenca Previous.is(TT_BinaryOperator)) { 9428da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = false; 943*bebf7bdfSowenca } 9440fb19de0SKrasimir Georgiev if (PreviousNonComment && 9459aab0db1SBjörn Schäpers (PreviousNonComment->isOneOf(TT_TemplateCloser, TT_JavaAnnotation) || 9469aab0db1SBjörn Schäpers PreviousNonComment->ClosesRequiresClause) && 947*bebf7bdfSowenca Current.NestingLevel == 0) { 9488da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = false; 949*bebf7bdfSowenca } 950a0407740SDaniel Jasper if (NextNonComment->is(tok::question) || 951*bebf7bdfSowenca (PreviousNonComment && PreviousNonComment->is(tok::question))) { 9528da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = true; 953*bebf7bdfSowenca } 9545962fa80SDaniel Jasper if (Current.is(TT_BinaryOperator) && Current.CanBreakBefore) 9558da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = false; 956de0328aeSDaniel Jasper 957de0328aeSDaniel Jasper if (!DryRun) { 958a004b3f5SMartin Probst unsigned MaxEmptyLinesToKeep = Style.MaxEmptyLinesToKeep + 1; 959a004b3f5SMartin Probst if (Current.is(tok::r_brace) && Current.MatchingParen && 960a004b3f5SMartin Probst // Only strip trailing empty lines for l_braces that have children, i.e. 961a004b3f5SMartin Probst // for function expressions (lambdas, arrows, etc). 962a004b3f5SMartin Probst !Current.MatchingParen->Children.empty()) { 963a004b3f5SMartin Probst // lambdas and arrow functions are expressions, thus their r_brace is not 964a004b3f5SMartin Probst // on its own line, and thus not covered by UnwrappedLineFormatter's logic 965a004b3f5SMartin Probst // about removing empty lines on closing blocks. Special case them here. 966a004b3f5SMartin Probst MaxEmptyLinesToKeep = 1; 967a004b3f5SMartin Probst } 9685bcf99b4SPaul Hoad unsigned Newlines = 9695bcf99b4SPaul Hoad std::max(1u, std::min(Current.NewlinesBefore, MaxEmptyLinesToKeep)); 970a1c30937SKrasimir Georgiev bool ContinuePPDirective = 971a1c30937SKrasimir Georgiev State.Line->InPPDirective && State.Line->Type != LT_ImportStatement; 9727d42f3f7SDaniel Jasper Whitespaces.replaceWhitespace(Current, Newlines, State.Column, State.Column, 9738da319feSBjörn Schäpers CurrentState.IsAligned, ContinuePPDirective); 974de0328aeSDaniel Jasper } 975de0328aeSDaniel Jasper 976de0328aeSDaniel Jasper if (!Current.isTrailingComment()) 9778da319feSBjörn Schäpers CurrentState.LastSpace = State.Column; 978*bebf7bdfSowenca if (Current.is(tok::lessless)) { 979602a727aSDaniel Jasper // If we are breaking before a "<<", we always want to indent relative to 980602a727aSDaniel Jasper // RHS. This is necessary only for "<<", as we special-case it and don't 981602a727aSDaniel Jasper // always indent relative to the RHS. 9828da319feSBjörn Schäpers CurrentState.LastSpace += 3; // 3 -> width of "<< ". 983*bebf7bdfSowenca } 984602a727aSDaniel Jasper 98505cd5868SDaniel Jasper State.StartOfLineLevel = Current.NestingLevel; 98605cd5868SDaniel Jasper State.LowestLevelOnLine = Current.NestingLevel; 987de0328aeSDaniel Jasper 988de0328aeSDaniel Jasper // Any break on this level means that the parent level has been broken 989de0328aeSDaniel Jasper // and we need to avoid bin packing there. 9904b444495SDaniel Jasper bool NestedBlockSpecialCase = 99170c98671Smydeveloperday (!Style.isCpp() && Current.is(tok::r_brace) && State.Stack.size() > 1 && 99270c98671Smydeveloperday State.Stack[State.Stack.size() - 2].NestedBlockInlined) || 99370c98671Smydeveloperday (Style.Language == FormatStyle::LK_ObjC && Current.is(tok::r_brace) && 99470c98671Smydeveloperday State.Stack.size() > 1 && !Style.ObjCBreakBeforeNestedBlockParam); 9959aab0db1SBjörn Schäpers // Do not force parameter break for statements with requires expressions. 9969aab0db1SBjörn Schäpers NestedBlockSpecialCase = 9979aab0db1SBjörn Schäpers NestedBlockSpecialCase || 9989aab0db1SBjörn Schäpers (Current.MatchingParen && 9999aab0db1SBjörn Schäpers Current.MatchingParen->is(TT_RequiresExpressionLBrace)); 10001699eca1SDaniel Jasper if (!NestedBlockSpecialCase) 10019dad527fSMarek Kurdej for (ParenState &PState : llvm::drop_end(State.Stack)) 10029dad527fSMarek Kurdej PState.BreakBeforeParameter = true; 1003b16b969dSDaniel Jasper 10049e5ede0fSDaniel Jasper if (PreviousNonComment && 1005a6b6d51bSFrancois Ferrand !PreviousNonComment->isOneOf(tok::comma, tok::colon, tok::semi) && 10069aab0db1SBjörn Schäpers ((PreviousNonComment->isNot(TT_TemplateCloser) && 10079aab0db1SBjörn Schäpers !PreviousNonComment->ClosesRequiresClause) || 10086c0ee17bSDaniel Jasper Current.NestingLevel != 0) && 100947bbda09SDaniel Jasper !PreviousNonComment->isOneOf( 101047bbda09SDaniel Jasper TT_BinaryOperator, TT_FunctionAnnotationRParen, TT_JavaAnnotation, 1011a98b7b01SDaniel Jasper TT_LeadingJavaAnnotation) && 1012*bebf7bdfSowenca Current.isNot(TT_BinaryOperator) && !PreviousNonComment->opensScope()) { 10138da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = true; 1014*bebf7bdfSowenca } 1015de0328aeSDaniel Jasper 10161db6c386SDaniel Jasper // If we break after { or the [ of an array initializer, we should also break 10171db6c386SDaniel Jasper // before the corresponding } or ]. 101890818057SDaniel Jasper if (PreviousNonComment && 101998e0b12dSDaniel Jasper (PreviousNonComment->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) || 1020*bebf7bdfSowenca opensProtoMessageField(*PreviousNonComment, Style))) { 10218da319feSBjörn Schäpers CurrentState.BreakBeforeClosingBrace = true; 1022*bebf7bdfSowenca } 1023de0328aeSDaniel Jasper 1024*bebf7bdfSowenca if (PreviousNonComment && PreviousNonComment->is(tok::l_paren)) { 10258da319feSBjörn Schäpers CurrentState.BreakBeforeClosingParen = 1026966f24e5SCameron Mulhern Style.AlignAfterOpenBracket == FormatStyle::BAS_BlockIndent; 1027*bebf7bdfSowenca } 1028966f24e5SCameron Mulhern 10298da319feSBjörn Schäpers if (CurrentState.AvoidBinPacking) { 1030c6deae45SPaul Hoad // If we are breaking after '(', '{', '<', or this is the break after a ':' 1031c6deae45SPaul Hoad // to start a member initializater list in a constructor, this should not 1032c6deae45SPaul Hoad // be considered bin packing unless the relevant AllowAll option is false or 1033c6deae45SPaul Hoad // this is a dict/object literal. 1034c6deae45SPaul Hoad bool PreviousIsBreakingCtorInitializerColon = 1035c6deae45SPaul Hoad Previous.is(TT_CtorInitializerColon) && 1036c6deae45SPaul Hoad Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon; 1037c6deae45SPaul Hoad if (!(Previous.isOneOf(tok::l_paren, tok::l_brace, TT_BinaryOperator) || 1038c6deae45SPaul Hoad PreviousIsBreakingCtorInitializerColon) || 1039de0328aeSDaniel Jasper (!Style.AllowAllParametersOfDeclarationOnNextLine && 10402a958321SDaniel Jasper State.Line->MustBeDeclaration) || 1041c6deae45SPaul Hoad (!Style.AllowAllArgumentsOnNextLine && 1042c6deae45SPaul Hoad !State.Line->MustBeDeclaration) || 10438a780a2fSowenca (Style.PackConstructorInitializers != FormatStyle::PCIS_NextLine && 1044c6deae45SPaul Hoad PreviousIsBreakingCtorInitializerColon) || 1045*bebf7bdfSowenca Previous.is(TT_DictLiteral)) { 10468da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = true; 1047*bebf7bdfSowenca } 1048c6deae45SPaul Hoad 1049c6deae45SPaul Hoad // If we are breaking after a ':' to start a member initializer list, 1050c6deae45SPaul Hoad // and we allow all arguments on the next line, we should not break 1051c6deae45SPaul Hoad // before the next parameter. 1052c6deae45SPaul Hoad if (PreviousIsBreakingCtorInitializerColon && 1053*bebf7bdfSowenca Style.PackConstructorInitializers == FormatStyle::PCIS_NextLine) { 10548da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = false; 1055de0328aeSDaniel Jasper } 1056*bebf7bdfSowenca } 1057de0328aeSDaniel Jasper 10581f80396dSAlexander Kornienko return Penalty; 1059de0328aeSDaniel Jasper } 1060de0328aeSDaniel Jasper 10619f388d04SDaniel Jasper unsigned ContinuationIndenter::getNewLineColumn(const LineState &State) { 10625d2587daSDaniel Jasper if (!State.NextToken || !State.NextToken->Previous) 10635d2587daSDaniel Jasper return 0; 106478e2a3c6SJonathan Coe 10659f388d04SDaniel Jasper FormatToken &Current = *State.NextToken; 10668da319feSBjörn Schäpers const auto &CurrentState = State.Stack.back(); 106778e2a3c6SJonathan Coe 10688da319feSBjörn Schäpers if (CurrentState.IsCSharpGenericTypeConstraint && 1069*bebf7bdfSowenca Current.isNot(TT_CSharpGenericTypeConstraint)) { 10708da319feSBjörn Schäpers return CurrentState.ColonPos + 2; 1071*bebf7bdfSowenca } 107278e2a3c6SJonathan Coe 10734281c5aeSDaniel Jasper const FormatToken &Previous = *Current.Previous; 10749f388d04SDaniel Jasper // If we are continuing an expression, we want to use the continuation indent. 10759f388d04SDaniel Jasper unsigned ContinuationIndent = 10768da319feSBjörn Schäpers std::max(CurrentState.LastSpace, CurrentState.Indent) + 10779f388d04SDaniel Jasper Style.ContinuationIndentWidth; 10789f388d04SDaniel Jasper const FormatToken *PreviousNonComment = Current.getPreviousNonComment(); 10799f388d04SDaniel Jasper const FormatToken *NextNonComment = Previous.getNextNonComment(); 10809f388d04SDaniel Jasper if (!NextNonComment) 10819f388d04SDaniel Jasper NextNonComment = &Current; 108250b4bd7cSDaniel Jasper 108350b4bd7cSDaniel Jasper // Java specific bits. 1084d0ec0d62SDaniel Jasper if (Style.Language == FormatStyle::LK_Java && 1085*bebf7bdfSowenca Current.isOneOf(Keywords.kw_implements, Keywords.kw_extends)) { 10868da319feSBjörn Schäpers return std::max(CurrentState.LastSpace, 10878da319feSBjörn Schäpers CurrentState.Indent + Style.ContinuationIndentWidth); 1088*bebf7bdfSowenca } 108950b4bd7cSDaniel Jasper 1090a506ed25SPaul Hoad if (Style.BreakBeforeBraces == FormatStyle::BS_Whitesmiths && 1091*bebf7bdfSowenca State.Line->First->is(tok::kw_enum)) { 1092a506ed25SPaul Hoad return (Style.IndentWidth * State.Line->First->IndentLevel) + 1093a506ed25SPaul Hoad Style.IndentWidth; 1094*bebf7bdfSowenca } 1095a506ed25SPaul Hoad 1096f5acd11dSBruno Ricci if (NextNonComment->is(tok::l_brace) && NextNonComment->is(BK_Block)) 10978da319feSBjörn Schäpers return Current.NestingLevel == 0 ? State.FirstIndent : CurrentState.Indent; 1098ff747be4SKrasimir Georgiev if ((Current.isOneOf(tok::r_brace, tok::r_square) || 109926b144ccSKrasimir Georgiev (Current.is(tok::greater) && 110026b144ccSKrasimir Georgiev (Style.Language == FormatStyle::LK_Proto || 110126b144ccSKrasimir Georgiev Style.Language == FormatStyle::LK_TextProto))) && 1102ff747be4SKrasimir Georgiev State.Stack.size() > 1) { 1103bd73bcf8SDaniel Jasper if (Current.closesBlockOrBlockTypeList(Style)) 110411a0ac66SDaniel Jasper return State.Stack[State.Stack.size() - 2].NestedBlockIndent; 1105f5acd11dSBruno Ricci if (Current.MatchingParen && Current.MatchingParen->is(BK_BracedInit)) 11069f388d04SDaniel Jasper return State.Stack[State.Stack.size() - 2].LastSpace; 11079f388d04SDaniel Jasper return State.FirstIndent; 11089f388d04SDaniel Jasper } 1109e4f95e8eSOwen Pan // Indent a closing parenthesis at the previous level if followed by a semi, 1110e4f95e8eSOwen Pan // const, or opening brace. This allows indentations such as: 1111b2f06eaeSMartin Probst // foo( 1112b2f06eaeSMartin Probst // a, 1113b2f06eaeSMartin Probst // ); 1114e4f95e8eSOwen Pan // int Foo::getter( 1115e4f95e8eSOwen Pan // // 1116e4f95e8eSOwen Pan // ) const { 1117e4f95e8eSOwen Pan // return foo; 1118e4f95e8eSOwen Pan // } 1119b2f06eaeSMartin Probst // function foo( 1120b2f06eaeSMartin Probst // a, 1121b2f06eaeSMartin Probst // ) { 1122b2f06eaeSMartin Probst // code(); // 1123b2f06eaeSMartin Probst // } 1124b2f06eaeSMartin Probst if (Current.is(tok::r_paren) && State.Stack.size() > 1 && 1125e4f95e8eSOwen Pan (!Current.Next || 1126*bebf7bdfSowenca Current.Next->isOneOf(tok::semi, tok::kw_const, tok::l_brace))) { 11272c1cdae2SMartin Probst return State.Stack[State.Stack.size() - 2].LastSpace; 1128*bebf7bdfSowenca } 1129966f24e5SCameron Mulhern if (Style.AlignAfterOpenBracket == FormatStyle::BAS_BlockIndent && 1130*bebf7bdfSowenca Current.is(tok::r_paren) && State.Stack.size() > 1) { 1131966f24e5SCameron Mulhern return State.Stack[State.Stack.size() - 2].LastSpace; 1132*bebf7bdfSowenca } 113398e0b12dSDaniel Jasper if (NextNonComment->is(TT_TemplateString) && NextNonComment->closesScope()) 113498e0b12dSDaniel Jasper return State.Stack[State.Stack.size() - 2].LastSpace; 1135783bac6bSDaniel Jasper if (Current.is(tok::identifier) && Current.Next && 1136ddb19248SKrasimir Georgiev (Current.Next->is(TT_DictLiteral) || 1137ddb19248SKrasimir Georgiev ((Style.Language == FormatStyle::LK_Proto || 1138ddb19248SKrasimir Georgiev Style.Language == FormatStyle::LK_TextProto) && 1139*bebf7bdfSowenca Current.Next->isOneOf(tok::less, tok::l_brace)))) { 11408da319feSBjörn Schäpers return CurrentState.Indent; 1141*bebf7bdfSowenca } 11420928553eSDaniel Jasper if (NextNonComment->is(TT_ObjCStringLiteral) && 1143*bebf7bdfSowenca State.StartOfStringLiteral != 0) { 11440928553eSDaniel Jasper return State.StartOfStringLiteral - 1; 1145*bebf7bdfSowenca } 1146d4fa2e63SAlexander Kornienko if (NextNonComment->isStringLiteral() && State.StartOfStringLiteral != 0) 1147d4fa2e63SAlexander Kornienko return State.StartOfStringLiteral; 11488da319feSBjörn Schäpers if (NextNonComment->is(tok::lessless) && CurrentState.FirstLessLess != 0) 11498da319feSBjörn Schäpers return CurrentState.FirstLessLess; 11509f388d04SDaniel Jasper if (NextNonComment->isMemberAccess()) { 11518da319feSBjörn Schäpers if (CurrentState.CallContinuation == 0) 11529f388d04SDaniel Jasper return ContinuationIndent; 11538da319feSBjörn Schäpers return CurrentState.CallContinuation; 11549f388d04SDaniel Jasper } 11558da319feSBjörn Schäpers if (CurrentState.QuestionColumn != 0 && 1156c0d606a5SDaniel Jasper ((NextNonComment->is(tok::colon) && 1157a98b7b01SDaniel Jasper NextNonComment->is(TT_ConditionalExpr)) || 11584db94094SFrancois Ferrand Previous.is(TT_ConditionalExpr))) { 11594db94094SFrancois Ferrand if (((NextNonComment->is(tok::colon) && NextNonComment->Next && 11604db94094SFrancois Ferrand !NextNonComment->Next->FakeLParens.empty() && 11614db94094SFrancois Ferrand NextNonComment->Next->FakeLParens.back() == prec::Conditional) || 11624db94094SFrancois Ferrand (Previous.is(tok::colon) && !Current.FakeLParens.empty() && 11634db94094SFrancois Ferrand Current.FakeLParens.back() == prec::Conditional)) && 11648da319feSBjörn Schäpers !CurrentState.IsWrappedConditional) { 11654db94094SFrancois Ferrand // NOTE: we may tweak this slightly: 11664db94094SFrancois Ferrand // * not remove the 'lead' ContinuationIndentWidth 11674db94094SFrancois Ferrand // * always un-indent by the operator when 11684db94094SFrancois Ferrand // BreakBeforeTernaryOperators=true 11698da319feSBjörn Schäpers unsigned Indent = CurrentState.Indent; 1170d079995dSMarek Kurdej if (Style.AlignOperands != FormatStyle::OAS_DontAlign) 11710fad648bSKrasimir Georgiev Indent -= Style.ContinuationIndentWidth; 11728da319feSBjörn Schäpers if (Style.BreakBeforeTernaryOperators && CurrentState.UnindentOperator) 11730ee04e6eSFrancois Ferrand Indent -= 2; 11744db94094SFrancois Ferrand return Indent; 11754db94094SFrancois Ferrand } 11768da319feSBjörn Schäpers return CurrentState.QuestionColumn; 11774db94094SFrancois Ferrand } 11788da319feSBjörn Schäpers if (Previous.is(tok::comma) && CurrentState.VariablePos != 0) 11798da319feSBjörn Schäpers return CurrentState.VariablePos; 11809aab0db1SBjörn Schäpers if (Current.is(TT_RequiresClause)) { 11819aab0db1SBjörn Schäpers if (Style.IndentRequiresClause) 11828da319feSBjörn Schäpers return CurrentState.Indent + Style.IndentWidth; 11839aab0db1SBjörn Schäpers switch (Style.RequiresClausePosition) { 11849aab0db1SBjörn Schäpers case FormatStyle::RCPS_OwnLine: 11859aab0db1SBjörn Schäpers case FormatStyle::RCPS_WithFollowing: 11868da319feSBjörn Schäpers return CurrentState.Indent; 11879aab0db1SBjörn Schäpers default: 11889aab0db1SBjörn Schäpers break; 11899aab0db1SBjörn Schäpers } 11909aab0db1SBjörn Schäpers } 1191e9ab42dfSDaniel Jasper if ((PreviousNonComment && 1192e9ab42dfSDaniel Jasper (PreviousNonComment->ClosesTemplateDeclaration || 11939aab0db1SBjörn Schäpers PreviousNonComment->ClosesRequiresClause || 119447bbda09SDaniel Jasper PreviousNonComment->isOneOf( 1195b060ad87SBen Hamilton TT_AttributeParen, TT_AttributeSquare, TT_FunctionAnnotationRParen, 1196b060ad87SBen Hamilton TT_JavaAnnotation, TT_LeadingJavaAnnotation))) || 1197c75e1effSDaniel Jasper (!Style.IndentWrappedFunctionNames && 1198*bebf7bdfSowenca NextNonComment->isOneOf(tok::kw_operator, TT_FunctionDeclarationName))) { 11998da319feSBjörn Schäpers return std::max(CurrentState.LastSpace, CurrentState.Indent); 1200*bebf7bdfSowenca } 1201a98b7b01SDaniel Jasper if (NextNonComment->is(TT_SelectorName)) { 12028da319feSBjörn Schäpers if (!CurrentState.ObjCSelectorNameFound) { 12038da319feSBjörn Schäpers unsigned MinIndent = CurrentState.Indent; 1204*bebf7bdfSowenca if (shouldIndentWrappedSelectorName(Style, State.Line->Type)) { 12054dc1cdc5SBen Hamilton MinIndent = std::max(MinIndent, 12064dc1cdc5SBen Hamilton State.FirstIndent + Style.ContinuationIndentWidth); 1207*bebf7bdfSowenca } 12084dc1cdc5SBen Hamilton // If LongestObjCSelectorName is 0, we are indenting the first 12094dc1cdc5SBen Hamilton // part of an ObjC selector (or a selector component which is 12104dc1cdc5SBen Hamilton // not colon-aligned due to block formatting). 12114dc1cdc5SBen Hamilton // 12124dc1cdc5SBen Hamilton // Otherwise, we are indenting a subsequent part of an ObjC 12134dc1cdc5SBen Hamilton // selector which should be colon-aligned to the longest 12144dc1cdc5SBen Hamilton // component of the ObjC selector. 12154dc1cdc5SBen Hamilton // 12164dc1cdc5SBen Hamilton // In either case, we want to respect Style.IndentWrappedFunctionNames. 12174dc1cdc5SBen Hamilton return MinIndent + 121838d80134SFrancois Ferrand std::max(NextNonComment->LongestObjCSelectorName, 121938d80134SFrancois Ferrand NextNonComment->ColumnWidth) - 12209f388d04SDaniel Jasper NextNonComment->ColumnWidth; 12219f388d04SDaniel Jasper } 12228da319feSBjörn Schäpers if (!CurrentState.AlignColons) 12238da319feSBjörn Schäpers return CurrentState.Indent; 12248da319feSBjörn Schäpers if (CurrentState.ColonPos > NextNonComment->ColumnWidth) 12258da319feSBjörn Schäpers return CurrentState.ColonPos - NextNonComment->ColumnWidth; 12268da319feSBjörn Schäpers return CurrentState.Indent; 12279f388d04SDaniel Jasper } 1228fd36f0b5SDaniel Jasper if (NextNonComment->is(tok::colon) && NextNonComment->is(TT_ObjCMethodExpr)) 12298da319feSBjörn Schäpers return CurrentState.ColonPos; 1230a98b7b01SDaniel Jasper if (NextNonComment->is(TT_ArraySubscriptLSquare)) { 1231*bebf7bdfSowenca if (CurrentState.StartOfArraySubscripts != 0) { 12328da319feSBjörn Schäpers return CurrentState.StartOfArraySubscripts; 1233*bebf7bdfSowenca } else if (Style.isCSharp()) { // C# allows `["key"] = value` inside object 12340c28a093SJonathan Coe // initializers. 12358da319feSBjörn Schäpers return CurrentState.Indent; 1236*bebf7bdfSowenca } 12379f388d04SDaniel Jasper return ContinuationIndent; 12389f388d04SDaniel Jasper } 12399c95013eSDaniel Jasper 12409c95013eSDaniel Jasper // This ensure that we correctly format ObjC methods calls without inputs, 12419c95013eSDaniel Jasper // i.e. where the last element isn't selector like: [callee method]; 12429c95013eSDaniel Jasper if (NextNonComment->is(tok::identifier) && NextNonComment->FakeRParens == 0 && 1243*bebf7bdfSowenca NextNonComment->Next && NextNonComment->Next->is(TT_ObjCMethodExpr)) { 12448da319feSBjörn Schäpers return CurrentState.Indent; 1245*bebf7bdfSowenca } 12469c95013eSDaniel Jasper 1247b754a747SDaniel Jasper if (NextNonComment->isOneOf(TT_StartOfName, TT_PointerOrReference) || 1248*bebf7bdfSowenca Previous.isOneOf(tok::coloncolon, tok::equal, TT_JsTypeColon)) { 12499f388d04SDaniel Jasper return ContinuationIndent; 1250*bebf7bdfSowenca } 12519f388d04SDaniel Jasper if (PreviousNonComment && PreviousNonComment->is(tok::colon) && 1252*bebf7bdfSowenca PreviousNonComment->isOneOf(TT_ObjCMethodExpr, TT_DictLiteral)) { 12539f388d04SDaniel Jasper return ContinuationIndent; 1254*bebf7bdfSowenca } 1255a98b7b01SDaniel Jasper if (NextNonComment->is(TT_CtorInitializerComma)) 12568da319feSBjörn Schäpers return CurrentState.Indent; 1257a6b6d51bSFrancois Ferrand if (PreviousNonComment && PreviousNonComment->is(TT_CtorInitializerColon) && 1258*bebf7bdfSowenca Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon) { 12598da319feSBjörn Schäpers return CurrentState.Indent; 1260*bebf7bdfSowenca } 12616bb103f9SFrancois Ferrand if (PreviousNonComment && PreviousNonComment->is(TT_InheritanceColon) && 1262*bebf7bdfSowenca Style.BreakInheritanceList == FormatStyle::BILS_AfterColon) { 12638da319feSBjörn Schäpers return CurrentState.Indent; 1264*bebf7bdfSowenca } 12650ef8ee19SAndi-Bogdan Postelnicu if (NextNonComment->isOneOf(TT_CtorInitializerColon, TT_InheritanceColon, 1266*bebf7bdfSowenca TT_InheritanceComma)) { 12670ef8ee19SAndi-Bogdan Postelnicu return State.FirstIndent + Style.ConstructorInitializerIndentWidth; 1268*bebf7bdfSowenca } 1269316ab38eSDaniel Jasper if (Previous.is(tok::r_paren) && !Current.isBinaryOperator() && 1270*bebf7bdfSowenca !Current.isOneOf(tok::colon, tok::comment)) { 1271316ab38eSDaniel Jasper return ContinuationIndent; 1272*bebf7bdfSowenca } 12734e290648SKrasimir Georgiev if (Current.is(TT_ProtoExtensionLSquare)) 12748da319feSBjörn Schäpers return CurrentState.Indent; 1275*bebf7bdfSowenca if (Current.isBinaryOperator() && CurrentState.UnindentOperator) { 12768da319feSBjörn Schäpers return CurrentState.Indent - Current.Tok.getLength() - 12770ee04e6eSFrancois Ferrand Current.SpacesRequiredBefore; 1278*bebf7bdfSowenca } 12790ee04e6eSFrancois Ferrand if (Current.isOneOf(tok::comment, TT_BlockComment, TT_LineComment) && 1280*bebf7bdfSowenca NextNonComment->isBinaryOperator() && CurrentState.UnindentOperator) { 12818da319feSBjörn Schäpers return CurrentState.Indent - NextNonComment->Tok.getLength() - 12820ee04e6eSFrancois Ferrand NextNonComment->SpacesRequiredBefore; 1283*bebf7bdfSowenca } 12848da319feSBjörn Schäpers if (CurrentState.Indent == State.FirstIndent && PreviousNonComment && 1285*bebf7bdfSowenca !PreviousNonComment->isOneOf(tok::r_brace, TT_CtorInitializerComma)) { 12869f388d04SDaniel Jasper // Ensure that we fall back to the continuation indent width instead of 12879f388d04SDaniel Jasper // just flushing continuations left. 12888da319feSBjörn Schäpers return CurrentState.Indent + Style.ContinuationIndentWidth; 1289*bebf7bdfSowenca } 12908da319feSBjörn Schäpers return CurrentState.Indent; 12919f388d04SDaniel Jasper } 12929f388d04SDaniel Jasper 1293fa0118e6SWawha static bool hasNestedBlockInlined(const FormatToken *Previous, 1294fa0118e6SWawha const FormatToken &Current, 1295fa0118e6SWawha const FormatStyle &Style) { 1296fa0118e6SWawha if (Previous->isNot(tok::l_paren)) 1297fa0118e6SWawha return true; 1298fa0118e6SWawha if (Previous->ParameterCount > 1) 1299fa0118e6SWawha return true; 1300fa0118e6SWawha 1301fa0118e6SWawha // Also a nested block if contains a lambda inside function with 1 parameter 13027d5062c6SMarek Kurdej return Style.BraceWrapping.BeforeLambdaBody && Current.is(TT_LambdaLSquare); 1303fa0118e6SWawha } 1304fa0118e6SWawha 1305de0328aeSDaniel Jasper unsigned ContinuationIndenter::moveStateToNextToken(LineState &State, 1306de0328aeSDaniel Jasper bool DryRun, bool Newline) { 1307de0328aeSDaniel Jasper assert(State.Stack.size()); 130860553be5SDaniel Jasper const FormatToken &Current = *State.NextToken; 13098da319feSBjörn Schäpers auto &CurrentState = State.Stack.back(); 1310de0328aeSDaniel Jasper 131178e2a3c6SJonathan Coe if (Current.is(TT_CSharpGenericTypeConstraint)) 13128da319feSBjörn Schäpers CurrentState.IsCSharpGenericTypeConstraint = true; 1313240527caSDaniel Jasper if (Current.isOneOf(tok::comma, TT_BinaryOperator)) 13148da319feSBjörn Schäpers CurrentState.NoLineBreakInOperand = false; 131578e2a3c6SJonathan Coe if (Current.isOneOf(TT_InheritanceColon, TT_CSharpGenericTypeConstraintColon)) 13168da319feSBjörn Schäpers CurrentState.AvoidBinPacking = true; 1317a98b7b01SDaniel Jasper if (Current.is(tok::lessless) && Current.isNot(TT_OverloadedOperator)) { 13188da319feSBjörn Schäpers if (CurrentState.FirstLessLess == 0) 13198da319feSBjörn Schäpers CurrentState.FirstLessLess = State.Column; 1320c0d606a5SDaniel Jasper else 13218da319feSBjörn Schäpers CurrentState.LastOperatorWrapped = Newline; 1322c0d606a5SDaniel Jasper } 1323240527caSDaniel Jasper if (Current.is(TT_BinaryOperator) && Current.isNot(tok::lessless)) 13248da319feSBjörn Schäpers CurrentState.LastOperatorWrapped = Newline; 1325240527caSDaniel Jasper if (Current.is(TT_ConditionalExpr) && Current.Previous && 1326*bebf7bdfSowenca !Current.Previous->is(TT_ConditionalExpr)) { 13278da319feSBjörn Schäpers CurrentState.LastOperatorWrapped = Newline; 1328*bebf7bdfSowenca } 1329a98b7b01SDaniel Jasper if (Current.is(TT_ArraySubscriptLSquare) && 1330*bebf7bdfSowenca CurrentState.StartOfArraySubscripts == 0) { 13318da319feSBjörn Schäpers CurrentState.StartOfArraySubscripts = State.Column; 1332*bebf7bdfSowenca } 133323f27850SMarek Kurdej 133423f27850SMarek Kurdej auto IsWrappedConditional = [](const FormatToken &Tok) { 133523f27850SMarek Kurdej if (!(Tok.is(TT_ConditionalExpr) && Tok.is(tok::question))) 133623f27850SMarek Kurdej return false; 133723f27850SMarek Kurdej if (Tok.MustBreakBefore) 133823f27850SMarek Kurdej return true; 133923f27850SMarek Kurdej 134023f27850SMarek Kurdej const FormatToken *Next = Tok.getNextNonComment(); 134123f27850SMarek Kurdej return Next && Next->MustBreakBefore; 134223f27850SMarek Kurdej }; 134323f27850SMarek Kurdej if (IsWrappedConditional(Current)) 13448da319feSBjörn Schäpers CurrentState.IsWrappedConditional = true; 134545860facSDaniel Jasper if (Style.BreakBeforeTernaryOperators && Current.is(tok::question)) 13468da319feSBjörn Schäpers CurrentState.QuestionColumn = State.Column; 134745860facSDaniel Jasper if (!Style.BreakBeforeTernaryOperators && Current.isNot(tok::colon)) { 134845860facSDaniel Jasper const FormatToken *Previous = Current.Previous; 134945860facSDaniel Jasper while (Previous && Previous->isTrailingComment()) 135045860facSDaniel Jasper Previous = Previous->Previous; 135145860facSDaniel Jasper if (Previous && Previous->is(tok::question)) 13528da319feSBjörn Schäpers CurrentState.QuestionColumn = State.Column; 135345860facSDaniel Jasper } 1354cab46171SDaniel Jasper if (!Current.opensScope() && !Current.closesScope() && 1355*bebf7bdfSowenca !Current.is(TT_PointerOrReference)) { 1356de0328aeSDaniel Jasper State.LowestLevelOnLine = 135705cd5868SDaniel Jasper std::min(State.LowestLevelOnLine, Current.NestingLevel); 1358*bebf7bdfSowenca } 13594c6e0059SDaniel Jasper if (Current.isMemberAccess()) 13608da319feSBjörn Schäpers CurrentState.StartOfFunctionCall = !Current.NextOperator ? 0 : State.Column; 1361f51ce777SJacek Olesiak if (Current.is(TT_SelectorName)) 13628da319feSBjörn Schäpers CurrentState.ObjCSelectorNameFound = true; 1363a6b6d51bSFrancois Ferrand if (Current.is(TT_CtorInitializerColon) && 1364a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers != FormatStyle::BCIS_AfterColon) { 1365de0328aeSDaniel Jasper // Indent 2 from the column, so: 1366de0328aeSDaniel Jasper // SomeClass::SomeClass() 1367de0328aeSDaniel Jasper // : First(...), ... 1368de0328aeSDaniel Jasper // Next(...) 1369de0328aeSDaniel Jasper // ^ line up here. 13708da319feSBjörn Schäpers CurrentState.Indent = State.Column + (Style.BreakConstructorInitializers == 13718da319feSBjörn Schäpers FormatStyle::BCIS_BeforeComma 137289628f64SManuel Klimek ? 0 137389628f64SManuel Klimek : 2); 13748da319feSBjörn Schäpers CurrentState.NestedBlockIndent = CurrentState.Indent; 13758a780a2fSowenca if (Style.PackConstructorInitializers > FormatStyle::PCIS_BinPack) { 13768da319feSBjörn Schäpers CurrentState.AvoidBinPacking = true; 13778da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = 13788a780a2fSowenca Style.PackConstructorInitializers != FormatStyle::PCIS_NextLine; 1379c6deae45SPaul Hoad } else { 13808da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = false; 1381de0328aeSDaniel Jasper } 1382c6deae45SPaul Hoad } 1383a6b6d51bSFrancois Ferrand if (Current.is(TT_CtorInitializerColon) && 1384a6b6d51bSFrancois Ferrand Style.BreakConstructorInitializers == FormatStyle::BCIS_AfterColon) { 13858da319feSBjörn Schäpers CurrentState.Indent = 1386a6b6d51bSFrancois Ferrand State.FirstIndent + Style.ConstructorInitializerIndentWidth; 13878da319feSBjörn Schäpers CurrentState.NestedBlockIndent = CurrentState.Indent; 13888a780a2fSowenca if (Style.PackConstructorInitializers > FormatStyle::PCIS_BinPack) 13898da319feSBjörn Schäpers CurrentState.AvoidBinPacking = true; 1390a6b6d51bSFrancois Ferrand } 1391*bebf7bdfSowenca if (Current.is(TT_InheritanceColon)) { 13928da319feSBjörn Schäpers CurrentState.Indent = 13936bb103f9SFrancois Ferrand State.FirstIndent + Style.ConstructorInitializerIndentWidth; 1394*bebf7bdfSowenca } 1395c4144ea4SDaniel Jasper if (Current.isOneOf(TT_BinaryOperator, TT_ConditionalExpr) && Newline) 13968da319feSBjörn Schäpers CurrentState.NestedBlockIndent = State.Column + Current.ColumnWidth + 1; 1397d9b319e3SDaniel Jasper if (Current.isOneOf(TT_LambdaLSquare, TT_LambdaArrow)) 13988da319feSBjörn Schäpers CurrentState.LastSpace = State.Column; 13999aab0db1SBjörn Schäpers if (Current.is(TT_RequiresExpression)) 14008da319feSBjörn Schäpers CurrentState.NestedBlockIndent = State.Column; 1401de0328aeSDaniel Jasper 1402de0328aeSDaniel Jasper // Insert scopes created by fake parenthesis. 1403de0328aeSDaniel Jasper const FormatToken *Previous = Current.getPreviousNonComment(); 1404b16b969dSDaniel Jasper 1405b16b969dSDaniel Jasper // Add special behavior to support a format commonly used for JavaScript 1406b16b969dSDaniel Jasper // closures: 1407b16b969dSDaniel Jasper // SomeFunction(function() { 1408b16b969dSDaniel Jasper // foo(); 1409b16b969dSDaniel Jasper // bar(); 1410b16b969dSDaniel Jasper // }, a, b, c); 14119aab0db1SBjörn Schäpers if (Current.isNot(tok::comment) && !Current.ClosesRequiresClause && 14129aab0db1SBjörn Schäpers Previous && Previous->isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) && 14135528caceSKrasimir Georgiev !Previous->is(TT_DictLiteral) && State.Stack.size() > 1 && 14148da319feSBjörn Schäpers !CurrentState.HasMultipleNestedBlocks) { 14151699eca1SDaniel Jasper if (State.Stack[State.Stack.size() - 2].NestedBlockInlined && Newline) 14169dad527fSMarek Kurdej for (ParenState &PState : llvm::drop_end(State.Stack)) 14179dad527fSMarek Kurdej PState.NoLineBreak = true; 14184b444495SDaniel Jasper State.Stack[State.Stack.size() - 2].NestedBlockInlined = false; 1419b16b969dSDaniel Jasper } 1420d48d1f8eSBjörn Schäpers if (Previous && (Previous->isOneOf(TT_BinaryOperator, TT_ConditionalExpr) || 1421d48d1f8eSBjörn Schäpers (Previous->isOneOf(tok::l_paren, tok::comma, tok::colon) && 1422d48d1f8eSBjörn Schäpers !Previous->isOneOf(TT_DictLiteral, TT_ObjCMethodExpr)))) { 14238da319feSBjörn Schäpers CurrentState.NestedBlockInlined = 1424fa0118e6SWawha !Newline && hasNestedBlockInlined(Previous, Current, Style); 1425b16b969dSDaniel Jasper } 1426b16b969dSDaniel Jasper 142760553be5SDaniel Jasper moveStatePastFakeLParens(State, Newline); 142860553be5SDaniel Jasper moveStatePastScopeCloser(State); 14298da319feSBjörn Schäpers // Do not use CurrentState here, since the two functions before may change the 14308da319feSBjörn Schäpers // Stack. 143145ab559eSManuel Klimek bool AllowBreak = !State.Stack.back().NoLineBreak && 1432bc05ebaaSKrasimir Georgiev !State.Stack.back().NoLineBreakInOperand; 1433c06f6da3SDaniel Jasper moveStatePastScopeOpener(State, Newline); 143460553be5SDaniel Jasper moveStatePastFakeRParens(State); 143560553be5SDaniel Jasper 14360928553eSDaniel Jasper if (Current.is(TT_ObjCStringLiteral) && State.StartOfStringLiteral == 0) 14370928553eSDaniel Jasper State.StartOfStringLiteral = State.Column + 1; 1438*bebf7bdfSowenca if (Current.is(TT_CSharpStringLiteral) && State.StartOfStringLiteral == 0) { 1439cbb726d0SPaul Hoad State.StartOfStringLiteral = State.Column + 1; 1440*bebf7bdfSowenca } else if (Current.isStringLiteral() && State.StartOfStringLiteral == 0) { 1441d4fa2e63SAlexander Kornienko State.StartOfStringLiteral = State.Column; 1442*bebf7bdfSowenca } else if (!Current.isOneOf(tok::comment, tok::identifier, tok::hash) && 1443*bebf7bdfSowenca !Current.isStringLiteral()) { 144460553be5SDaniel Jasper State.StartOfStringLiteral = 0; 1445*bebf7bdfSowenca } 144660553be5SDaniel Jasper 144760553be5SDaniel Jasper State.Column += Current.ColumnWidth; 144860553be5SDaniel Jasper State.NextToken = State.NextToken->Next; 144945ab559eSManuel Klimek 1450bda8482aSKrasimir Georgiev unsigned Penalty = 1451bda8482aSKrasimir Georgiev handleEndOfLine(Current, State, DryRun, AllowBreak, Newline); 145260553be5SDaniel Jasper 145360553be5SDaniel Jasper if (Current.Role) 145460553be5SDaniel Jasper Current.Role->formatFromToken(State, this, DryRun); 145560553be5SDaniel Jasper // If the previous has a special role, let it consume tokens as appropriate. 145660553be5SDaniel Jasper // It is necessary to start at the previous token for the only implemented 145760553be5SDaniel Jasper // role (comma separated list). That way, the decision whether or not to break 145860553be5SDaniel Jasper // after the "{" is already done and both options are tried and evaluated. 145960553be5SDaniel Jasper // FIXME: This is ugly, find a better way. 146060553be5SDaniel Jasper if (Previous && Previous->Role) 146160553be5SDaniel Jasper Penalty += Previous->Role->formatAfterToken(State, this, DryRun); 146260553be5SDaniel Jasper 146360553be5SDaniel Jasper return Penalty; 146460553be5SDaniel Jasper } 146560553be5SDaniel Jasper 146660553be5SDaniel Jasper void ContinuationIndenter::moveStatePastFakeLParens(LineState &State, 146760553be5SDaniel Jasper bool Newline) { 146860553be5SDaniel Jasper const FormatToken &Current = *State.NextToken; 14694041f16bSBjörn Schäpers if (Current.FakeLParens.empty()) 14704041f16bSBjörn Schäpers return; 14714041f16bSBjörn Schäpers 147260553be5SDaniel Jasper const FormatToken *Previous = Current.getPreviousNonComment(); 147360553be5SDaniel Jasper 1474de0328aeSDaniel Jasper // Don't add extra indentation for the first fake parenthesis after 14759aab0db1SBjörn Schäpers // 'return', assignments, opening <({[, or requires clauses. The indentation 14769aab0db1SBjörn Schäpers // for these cases is special cased. 1477de0328aeSDaniel Jasper bool SkipFirstExtraIndent = 14789aab0db1SBjörn Schäpers Previous && 14799aab0db1SBjörn Schäpers (Previous->opensScope() || 14809aab0db1SBjörn Schäpers Previous->isOneOf(tok::semi, tok::kw_return, TT_RequiresClause) || 14813219e43cSDaniel Jasper (Previous->getPrecedence() == prec::Assignment && 14820ee04e6eSFrancois Ferrand Style.AlignOperands != FormatStyle::OAS_DontAlign) || 14839aab0db1SBjörn Schäpers Previous->is(TT_ObjCMethodExpr)); 148488fa4bfeSBjörn Schäpers for (const auto &PrecedenceLevel : llvm::reverse(Current.FakeLParens)) { 14858da319feSBjörn Schäpers const auto &CurrentState = State.Stack.back(); 14868da319feSBjörn Schäpers ParenState NewParenState = CurrentState; 14878b4bc76aSKrasimir Georgiev NewParenState.Tok = nullptr; 1488de0328aeSDaniel Jasper NewParenState.ContainsLineBreak = false; 148904bbda99SDaniel Jasper NewParenState.LastOperatorWrapped = true; 14904db94094SFrancois Ferrand NewParenState.IsChainedConditional = false; 14914db94094SFrancois Ferrand NewParenState.IsWrappedConditional = false; 14920ee04e6eSFrancois Ferrand NewParenState.UnindentOperator = false; 1493240527caSDaniel Jasper NewParenState.NoLineBreak = 14948da319feSBjörn Schäpers NewParenState.NoLineBreak || CurrentState.NoLineBreakInOperand; 1495eabede6dSDaniel Jasper 1496988e7e4fSDaniel Jasper // Don't propagate AvoidBinPacking into subexpressions of arg/param lists. 149788fa4bfeSBjörn Schäpers if (PrecedenceLevel > prec::Comma) 1498988e7e4fSDaniel Jasper NewParenState.AvoidBinPacking = false; 1499988e7e4fSDaniel Jasper 15003aa9a6a1SDaniel Jasper // Indent from 'LastSpace' unless these are fake parentheses encapsulating 15013aa9a6a1SDaniel Jasper // a builder type call after 'return' or, if the alignment after opening 15023aa9a6a1SDaniel Jasper // brackets is disabled. 15034281c5aeSDaniel Jasper if (!Current.isTrailingComment() && 15040ee04e6eSFrancois Ferrand (Style.AlignOperands != FormatStyle::OAS_DontAlign || 150588fa4bfeSBjörn Schäpers PrecedenceLevel < prec::Assignment) && 15066cab6784SDaniel Jasper (!Previous || Previous->isNot(tok::kw_return) || 150788fa4bfeSBjörn Schäpers (Style.Language != FormatStyle::LK_Java && PrecedenceLevel > 0)) && 15086501f7e8SDaniel Jasper (Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign || 150988fa4bfeSBjörn Schäpers PrecedenceLevel != prec::Comma || Current.NestingLevel == 0)) { 15108da319feSBjörn Schäpers NewParenState.Indent = std::max( 15118da319feSBjörn Schäpers std::max(State.Column, NewParenState.Indent), CurrentState.LastSpace); 15120ee04e6eSFrancois Ferrand } 15130ee04e6eSFrancois Ferrand 1514b18c63e8SMaximilian Fickert if (Previous && 15150ee04e6eSFrancois Ferrand (Previous->getPrecedence() == prec::Assignment || 15169aab0db1SBjörn Schäpers Previous->isOneOf(tok::kw_return, TT_RequiresClause) || 151788fa4bfeSBjörn Schäpers (PrecedenceLevel == prec::Conditional && Previous->is(tok::question) && 15180ee04e6eSFrancois Ferrand Previous->is(TT_ConditionalExpr))) && 1519b18c63e8SMaximilian Fickert !Newline) { 1520b18c63e8SMaximilian Fickert // If BreakBeforeBinaryOperators is set, un-indent a bit to account for 1521b18c63e8SMaximilian Fickert // the operator and keep the operands aligned 1522b18c63e8SMaximilian Fickert if (Style.AlignOperands == FormatStyle::OAS_AlignAfterOperator) 15230ee04e6eSFrancois Ferrand NewParenState.UnindentOperator = true; 1524b18c63e8SMaximilian Fickert // Mark indentation as alignment if the expression is aligned. 1525b18c63e8SMaximilian Fickert if (Style.AlignOperands != FormatStyle::OAS_DontAlign) 1526b18c63e8SMaximilian Fickert NewParenState.IsAligned = true; 1527b18c63e8SMaximilian Fickert } 1528eabede6dSDaniel Jasper 15299fe0e8daSDaniel Jasper // Do not indent relative to the fake parentheses inserted for "." or "->". 15309fe0e8daSDaniel Jasper // This is a special case to make the following to statements consistent: 15319fe0e8daSDaniel Jasper // OuterFunction(InnerFunctionCall( // break 15329fe0e8daSDaniel Jasper // ParameterToInnerFunction)); 15339fe0e8daSDaniel Jasper // OuterFunction(SomeObject.InnerFunctionCall( // break 15349fe0e8daSDaniel Jasper // ParameterToInnerFunction)); 153588fa4bfeSBjörn Schäpers if (PrecedenceLevel > prec::Unknown) 15369fe0e8daSDaniel Jasper NewParenState.LastSpace = std::max(NewParenState.LastSpace, State.Column); 153788fa4bfeSBjörn Schäpers if (PrecedenceLevel != prec::Conditional && !Current.is(TT_UnaryOperator) && 1538*bebf7bdfSowenca Style.AlignAfterOpenBracket != FormatStyle::BAS_DontAlign) { 153996964359SDaniel Jasper NewParenState.StartOfFunctionCall = State.Column; 1540*bebf7bdfSowenca } 1541de0328aeSDaniel Jasper 15424db94094SFrancois Ferrand // Indent conditional expressions, unless they are chained "else-if" 15434db94094SFrancois Ferrand // conditionals. Never indent expression where the 'operator' is ',', ';' or 15444db94094SFrancois Ferrand // an assignment (i.e. *I <= prec::Assignment) as those have different 15454db94094SFrancois Ferrand // indentation rules. Indent other expression, unless the indentation needs 15464db94094SFrancois Ferrand // to be skipped. 154788fa4bfeSBjörn Schäpers if (PrecedenceLevel == prec::Conditional && Previous && 154888fa4bfeSBjörn Schäpers Previous->is(tok::colon) && Previous->is(TT_ConditionalExpr) && 154988fa4bfeSBjörn Schäpers &PrecedenceLevel == &Current.FakeLParens.back() && 15508da319feSBjörn Schäpers !CurrentState.IsWrappedConditional) { 15514db94094SFrancois Ferrand NewParenState.IsChainedConditional = true; 15520ee04e6eSFrancois Ferrand NewParenState.UnindentOperator = State.Stack.back().UnindentOperator; 155388fa4bfeSBjörn Schäpers } else if (PrecedenceLevel == prec::Conditional || 155488fa4bfeSBjörn Schäpers (!SkipFirstExtraIndent && PrecedenceLevel > prec::Assignment && 15554db94094SFrancois Ferrand !Current.isTrailingComment())) { 15566633ab8aSDaniel Jasper NewParenState.Indent += Style.ContinuationIndentWidth; 15574db94094SFrancois Ferrand } 155888fa4bfeSBjörn Schäpers if ((Previous && !Previous->opensScope()) || PrecedenceLevel != prec::Comma) 1559de0328aeSDaniel Jasper NewParenState.BreakBeforeParameter = false; 1560de0328aeSDaniel Jasper State.Stack.push_back(NewParenState); 1561de0328aeSDaniel Jasper SkipFirstExtraIndent = false; 1562de0328aeSDaniel Jasper } 156360553be5SDaniel Jasper } 1564de0328aeSDaniel Jasper 156511a0ac66SDaniel Jasper void ContinuationIndenter::moveStatePastFakeRParens(LineState &State) { 156611a0ac66SDaniel Jasper for (unsigned i = 0, e = State.NextToken->FakeRParens; i != e; ++i) { 156760553be5SDaniel Jasper unsigned VariablePos = State.Stack.back().VariablePos; 1568819788daSManuel Klimek if (State.Stack.size() == 1) { 1569819788daSManuel Klimek // Do not pop the last element. 1570819788daSManuel Klimek break; 1571819788daSManuel Klimek } 15729fe0e8daSDaniel Jasper State.Stack.pop_back(); 157360553be5SDaniel Jasper State.Stack.back().VariablePos = VariablePos; 1574819788daSManuel Klimek } 15759aab0db1SBjörn Schäpers 15769aab0db1SBjörn Schäpers if (State.NextToken->ClosesRequiresClause && Style.IndentRequiresClause) { 15779aab0db1SBjörn Schäpers // Remove the indentation of the requires clauses (which is not in Indent, 15789aab0db1SBjörn Schäpers // but in LastSpace). 15799aab0db1SBjörn Schäpers State.Stack.back().LastSpace -= Style.IndentWidth; 15809aab0db1SBjörn Schäpers } 158160553be5SDaniel Jasper } 1582335ff266SDaniel Jasper 158360553be5SDaniel Jasper void ContinuationIndenter::moveStatePastScopeOpener(LineState &State, 158460553be5SDaniel Jasper bool Newline) { 158560553be5SDaniel Jasper const FormatToken &Current = *State.NextToken; 158660553be5SDaniel Jasper if (!Current.opensScope()) 158760553be5SDaniel Jasper return; 158860553be5SDaniel Jasper 15898da319feSBjörn Schäpers const auto &CurrentState = State.Stack.back(); 15908da319feSBjörn Schäpers 159178e2a3c6SJonathan Coe // Don't allow '<' or '(' in C# generic type constraints to start new scopes. 159278e2a3c6SJonathan Coe if (Current.isOneOf(tok::less, tok::l_paren) && 1593*bebf7bdfSowenca CurrentState.IsCSharpGenericTypeConstraint) { 159478e2a3c6SJonathan Coe return; 1595*bebf7bdfSowenca } 159678e2a3c6SJonathan Coe 1597f5acd11dSBruno Ricci if (Current.MatchingParen && Current.is(BK_Block)) { 159860553be5SDaniel Jasper moveStateToNewBlock(State); 159960553be5SDaniel Jasper return; 160060553be5SDaniel Jasper } 160160553be5SDaniel Jasper 160260553be5SDaniel Jasper unsigned NewIndent; 16038da319feSBjörn Schäpers unsigned LastSpace = CurrentState.LastSpace; 160460553be5SDaniel Jasper bool AvoidBinPacking; 160560553be5SDaniel Jasper bool BreakBeforeParameter = false; 16068da319feSBjörn Schäpers unsigned NestedBlockIndent = std::max(CurrentState.StartOfFunctionCall, 16078da319feSBjörn Schäpers CurrentState.NestedBlockIndent); 1608ff747be4SKrasimir Georgiev if (Current.isOneOf(tok::l_brace, TT_ArrayInitializerLSquare) || 160926b144ccSKrasimir Georgiev opensProtoMessageField(Current, Style)) { 1610bd73bcf8SDaniel Jasper if (Current.opensBlockOrBlockTypeList(Style)) { 161138423276SMartin Probst NewIndent = Style.IndentWidth + 16128da319feSBjörn Schäpers std::min(State.Column, CurrentState.NestedBlockIndent); 1613b8f61682SDaniel Jasper } else { 16148da319feSBjörn Schäpers NewIndent = CurrentState.LastSpace + Style.ContinuationIndentWidth; 1615e2e0387fSAlexander Kornienko } 1616de0328aeSDaniel Jasper const FormatToken *NextNoComment = Current.getNextNonComment(); 161750780ce1SDaniel Jasper bool EndsInComma = Current.MatchingParen && 161850780ce1SDaniel Jasper Current.MatchingParen->Previous && 161950780ce1SDaniel Jasper Current.MatchingParen->Previous->is(tok::comma); 162089628f64SManuel Klimek AvoidBinPacking = EndsInComma || Current.is(TT_DictLiteral) || 162126b144ccSKrasimir Georgiev Style.Language == FormatStyle::LK_Proto || 162226b144ccSKrasimir Georgiev Style.Language == FormatStyle::LK_TextProto || 162326b144ccSKrasimir Georgiev !Style.BinPackArguments || 16245f07f443SFrancois Ferrand (NextNoComment && 16255f07f443SFrancois Ferrand NextNoComment->isOneOf(TT_DesignatedInitializerPeriod, 16265f07f443SFrancois Ferrand TT_DesignatedInitializerLSquare)); 1627d2130f51SFrancois Ferrand BreakBeforeParameter = EndsInComma; 1628ea40cee2SDaniel Jasper if (Current.ParameterCount > 1) 1629ea40cee2SDaniel Jasper NestedBlockIndent = std::max(NestedBlockIndent, State.Column + 1); 1630de0328aeSDaniel Jasper } else { 16318da319feSBjörn Schäpers NewIndent = 16328da319feSBjörn Schäpers Style.ContinuationIndentWidth + 16338da319feSBjörn Schäpers std::max(CurrentState.LastSpace, CurrentState.StartOfFunctionCall); 1634de7ca75cSDaniel Jasper 1635de7ca75cSDaniel Jasper // Ensure that different different brackets force relative alignment, e.g.: 1636de7ca75cSDaniel Jasper // void SomeFunction(vector< // break 1637de7ca75cSDaniel Jasper // int> v); 1638de7ca75cSDaniel Jasper // FIXME: We likely want to do this for more combinations of brackets. 16393f11941dSDaniel Jasper if (Current.is(tok::less) && Current.ParentBracket == tok::l_paren) { 16408da319feSBjörn Schäpers NewIndent = std::max(NewIndent, CurrentState.Indent); 16418da319feSBjörn Schäpers LastSpace = std::max(LastSpace, CurrentState.Indent); 1642de7ca75cSDaniel Jasper } 1643de7ca75cSDaniel Jasper 16442c1cdae2SMartin Probst bool EndsInComma = 16452c1cdae2SMartin Probst Current.MatchingParen && 16462c1cdae2SMartin Probst Current.MatchingParen->getPreviousNonComment() && 16472c1cdae2SMartin Probst Current.MatchingParen->getPreviousNonComment()->is(tok::comma); 16482c1cdae2SMartin Probst 16494dc658caSBen Hamilton // If ObjCBinPackProtocolList is unspecified, fall back to BinPackParameters 16504dc658caSBen Hamilton // for backwards compatibility. 16514dc658caSBen Hamilton bool ObjCBinPackProtocolList = 16524dc658caSBen Hamilton (Style.ObjCBinPackProtocolList == FormatStyle::BPS_Auto && 16534dc658caSBen Hamilton Style.BinPackParameters) || 16544dc658caSBen Hamilton Style.ObjCBinPackProtocolList == FormatStyle::BPS_Always; 16554dc658caSBen Hamilton 16564dc658caSBen Hamilton bool BinPackDeclaration = 16574dc658caSBen Hamilton (State.Line->Type != LT_ObjCDecl && Style.BinPackParameters) || 16584dc658caSBen Hamilton (State.Line->Type == LT_ObjCDecl && ObjCBinPackProtocolList); 16594dc658caSBen Hamilton 166018210d7dSDaniel Jasper AvoidBinPacking = 16618da319feSBjörn Schäpers (CurrentState.IsCSharpGenericTypeConstraint) || 1662142e79b8Smydeveloperday (Style.isJavaScript() && EndsInComma) || 16634dc658caSBen Hamilton (State.Line->MustBeDeclaration && !BinPackDeclaration) || 166418210d7dSDaniel Jasper (!State.Line->MustBeDeclaration && !Style.BinPackArguments) || 1665de0328aeSDaniel Jasper (Style.ExperimentalAutoDetectBinPacking && 1666f5acd11dSBruno Ricci (Current.is(PPK_OnePerLine) || 1667f5acd11dSBruno Ricci (!BinPackInconclusiveFunctions && Current.is(PPK_Inconclusive)))); 16682c1cdae2SMartin Probst 166970c98671Smydeveloperday if (Current.is(TT_ObjCMethodExpr) && Current.MatchingParen && 167070c98671Smydeveloperday Style.ObjCBreakBeforeNestedBlockParam) { 1671289afc07SDaniel Jasper if (Style.ColumnLimit) { 16721db6c386SDaniel Jasper // If this '[' opens an ObjC call, determine whether all parameters fit 16731db6c386SDaniel Jasper // into one line and put one per line if they don't. 16748b4bc76aSKrasimir Georgiev if (getLengthToMatchingParen(Current, State.Stack) + State.Column > 1675*bebf7bdfSowenca getColumnLimit(State)) { 16761db6c386SDaniel Jasper BreakBeforeParameter = true; 1677*bebf7bdfSowenca } 1678289afc07SDaniel Jasper } else { 1679289afc07SDaniel Jasper // For ColumnLimit = 0, we have to figure out whether there is or has to 1680289afc07SDaniel Jasper // be a line break within this call. 1681289afc07SDaniel Jasper for (const FormatToken *Tok = &Current; 1682289afc07SDaniel Jasper Tok && Tok != Current.MatchingParen; Tok = Tok->Next) { 1683289afc07SDaniel Jasper if (Tok->MustBreakBefore || 1684289afc07SDaniel Jasper (Tok->CanBreakBefore && Tok->NewlinesBefore > 0)) { 1685289afc07SDaniel Jasper BreakBeforeParameter = true; 1686289afc07SDaniel Jasper break; 1687289afc07SDaniel Jasper } 1688289afc07SDaniel Jasper } 1689289afc07SDaniel Jasper } 1690289afc07SDaniel Jasper } 16912c1cdae2SMartin Probst 1692142e79b8Smydeveloperday if (Style.isJavaScript() && EndsInComma) 16932c1cdae2SMartin Probst BreakBeforeParameter = true; 1694de0328aeSDaniel Jasper } 1695bd73bcf8SDaniel Jasper // Generally inherit NoLineBreak from the current scope to nested scope. 1696bd73bcf8SDaniel Jasper // However, don't do this for non-empty nested blocks, dict literals and 1697bd73bcf8SDaniel Jasper // array literals as these follow different indentation rules. 1698bd73bcf8SDaniel Jasper bool NoLineBreak = 1699bd73bcf8SDaniel Jasper Current.Children.empty() && 1700bd73bcf8SDaniel Jasper !Current.isOneOf(TT_DictLiteral, TT_ArrayInitializerLSquare) && 17018da319feSBjörn Schäpers (CurrentState.NoLineBreak || CurrentState.NoLineBreakInOperand || 1702bd73bcf8SDaniel Jasper (Current.is(TT_TemplateOpener) && 17038da319feSBjörn Schäpers CurrentState.ContainsUnwrappedBuilder)); 17047d42f3f7SDaniel Jasper State.Stack.push_back( 17058b4bc76aSKrasimir Georgiev ParenState(&Current, NewIndent, LastSpace, AvoidBinPacking, NoLineBreak)); 17068da319feSBjörn Schäpers auto &NewState = State.Stack.back(); 17078da319feSBjörn Schäpers NewState.NestedBlockIndent = NestedBlockIndent; 17088da319feSBjörn Schäpers NewState.BreakBeforeParameter = BreakBeforeParameter; 17098da319feSBjörn Schäpers NewState.HasMultipleNestedBlocks = (Current.BlockParameterCount > 1); 1710fa0118e6SWawha 1711f594e3d2Smydeveloperday if (Style.BraceWrapping.BeforeLambdaBody && Current.Next != nullptr && 1712fee4a971SMarek Kurdej Current.is(tok::l_paren)) { 1713fa0118e6SWawha // Search for any parameter that is a lambda 1714fa0118e6SWawha FormatToken const *next = Current.Next; 1715fa0118e6SWawha while (next != nullptr) { 1716fa0118e6SWawha if (next->is(TT_LambdaLSquare)) { 17178da319feSBjörn Schäpers NewState.HasMultipleNestedBlocks = true; 1718fa0118e6SWawha break; 1719fa0118e6SWawha } 1720fa0118e6SWawha next = next->Next; 1721fa0118e6SWawha } 1722fa0118e6SWawha } 1723fa0118e6SWawha 17248da319feSBjörn Schäpers NewState.IsInsideObjCArrayLiteral = Current.is(TT_ArrayInitializerLSquare) && 17258da319feSBjörn Schäpers Current.Previous && 172609051f29SBen Hamilton Current.Previous->is(tok::at); 1727de0328aeSDaniel Jasper } 1728de0328aeSDaniel Jasper 172960553be5SDaniel Jasper void ContinuationIndenter::moveStatePastScopeCloser(LineState &State) { 173060553be5SDaniel Jasper const FormatToken &Current = *State.NextToken; 173160553be5SDaniel Jasper if (!Current.closesScope()) 173260553be5SDaniel Jasper return; 173360553be5SDaniel Jasper 1734de0328aeSDaniel Jasper // If we encounter a closing ), ], } or >, we can remove a level from our 1735de0328aeSDaniel Jasper // stacks. 173696df37a6SDaniel Jasper if (State.Stack.size() > 1 && 1737c06f6da3SDaniel Jasper (Current.isOneOf(tok::r_paren, tok::r_square, TT_TemplateString) || 17389fe0e8daSDaniel Jasper (Current.is(tok::r_brace) && State.NextToken != State.Line->First) || 1739a79d62d2SKrasimir Georgiev State.NextToken->is(TT_TemplateCloser) || 1740*bebf7bdfSowenca (Current.is(tok::greater) && Current.is(TT_DictLiteral)))) { 1741de0328aeSDaniel Jasper State.Stack.pop_back(); 1742*bebf7bdfSowenca } 1743335ff266SDaniel Jasper 17448da319feSBjörn Schäpers auto &CurrentState = State.Stack.back(); 17458da319feSBjörn Schäpers 17462dc3dac9SJacek Olesiak // Reevaluate whether ObjC message arguments fit into one line. 17472dc3dac9SJacek Olesiak // If a receiver spans multiple lines, e.g.: 17482dc3dac9SJacek Olesiak // [[object block:^{ 17492dc3dac9SJacek Olesiak // return 42; 17502dc3dac9SJacek Olesiak // }] a:42 b:42]; 17512dc3dac9SJacek Olesiak // BreakBeforeParameter is calculated based on an incorrect assumption 17522dc3dac9SJacek Olesiak // (it is checked whether the whole expression fits into one line without 17532dc3dac9SJacek Olesiak // considering a line break inside a message receiver). 1754f1191705SNico Weber // We check whether arguments fit after receiver scope closer (into the same 17552dc3dac9SJacek Olesiak // line). 17568da319feSBjörn Schäpers if (CurrentState.BreakBeforeParameter && Current.MatchingParen && 17572dc3dac9SJacek Olesiak Current.MatchingParen->Previous) { 17582dc3dac9SJacek Olesiak const FormatToken &CurrentScopeOpener = *Current.MatchingParen->Previous; 17592dc3dac9SJacek Olesiak if (CurrentScopeOpener.is(TT_ObjCMethodExpr) && 17602dc3dac9SJacek Olesiak CurrentScopeOpener.MatchingParen) { 17612dc3dac9SJacek Olesiak int NecessarySpaceInLine = 17622dc3dac9SJacek Olesiak getLengthToMatchingParen(CurrentScopeOpener, State.Stack) + 17632dc3dac9SJacek Olesiak CurrentScopeOpener.TotalLength - Current.TotalLength - 1; 17642dc3dac9SJacek Olesiak if (State.Column + Current.ColumnWidth + NecessarySpaceInLine <= 1765*bebf7bdfSowenca Style.ColumnLimit) { 17668da319feSBjörn Schäpers CurrentState.BreakBeforeParameter = false; 17672dc3dac9SJacek Olesiak } 17682dc3dac9SJacek Olesiak } 1769*bebf7bdfSowenca } 17702dc3dac9SJacek Olesiak 1771de0328aeSDaniel Jasper if (Current.is(tok::r_square)) { 1772de0328aeSDaniel Jasper // If this ends the array subscript expr, reset the corresponding value. 1773de0328aeSDaniel Jasper const FormatToken *NextNonComment = Current.getNextNonComment(); 1774de0328aeSDaniel Jasper if (NextNonComment && NextNonComment->isNot(tok::l_square)) 17758da319feSBjörn Schäpers CurrentState.StartOfArraySubscripts = 0; 1776de0328aeSDaniel Jasper } 177760553be5SDaniel Jasper } 1778de0328aeSDaniel Jasper 177960553be5SDaniel Jasper void ContinuationIndenter::moveStateToNewBlock(LineState &State) { 178011a0ac66SDaniel Jasper unsigned NestedBlockIndent = State.Stack.back().NestedBlockIndent; 178150d634b3SDaniel Jasper // ObjC block sometimes follow special indentation rules. 1782a98b7b01SDaniel Jasper unsigned NewIndent = 178311a0ac66SDaniel Jasper NestedBlockIndent + (State.NextToken->is(TT_ObjCBlockLBrace) 178450d634b3SDaniel Jasper ? Style.ObjCBlockIndentWidth 178560553be5SDaniel Jasper : Style.IndentWidth); 17868b4bc76aSKrasimir Georgiev State.Stack.push_back(ParenState(State.NextToken, NewIndent, 17878b4bc76aSKrasimir Georgiev State.Stack.back().LastSpace, 17887d42f3f7SDaniel Jasper /*AvoidBinPacking=*/true, 17896501f7e8SDaniel Jasper /*NoLineBreak=*/false)); 179011a0ac66SDaniel Jasper State.Stack.back().NestedBlockIndent = NestedBlockIndent; 179160553be5SDaniel Jasper State.Stack.back().BreakBeforeParameter = true; 1792de0328aeSDaniel Jasper } 1793de0328aeSDaniel Jasper 17949ad83fe7SKrasimir Georgiev static unsigned getLastLineEndColumn(StringRef Text, unsigned StartColumn, 17959ad83fe7SKrasimir Georgiev unsigned TabWidth, 17969ad83fe7SKrasimir Georgiev encoding::Encoding Encoding) { 17979ad83fe7SKrasimir Georgiev size_t LastNewlinePos = Text.find_last_of("\n"); 17989ad83fe7SKrasimir Georgiev if (LastNewlinePos == StringRef::npos) { 17999ad83fe7SKrasimir Georgiev return StartColumn + 18009ad83fe7SKrasimir Georgiev encoding::columnWidthWithTabs(Text, StartColumn, TabWidth, Encoding); 18019ad83fe7SKrasimir Georgiev } else { 18029ad83fe7SKrasimir Georgiev return encoding::columnWidthWithTabs(Text.substr(LastNewlinePos), 18039ad83fe7SKrasimir Georgiev /*StartColumn=*/0, TabWidth, Encoding); 18049ad83fe7SKrasimir Georgiev } 18059ad83fe7SKrasimir Georgiev } 18069ad83fe7SKrasimir Georgiev 18079ad83fe7SKrasimir Georgiev unsigned ContinuationIndenter::reformatRawStringLiteral( 180845ab559eSManuel Klimek const FormatToken &Current, LineState &State, 1809bda8482aSKrasimir Georgiev const FormatStyle &RawStringStyle, bool DryRun, bool Newline) { 181045ab559eSManuel Klimek unsigned StartColumn = State.Column - Current.ColumnWidth; 1811412ed095SKrasimir Georgiev StringRef OldDelimiter = *getRawStringDelimiter(Current.TokenText); 1812412ed095SKrasimir Georgiev StringRef NewDelimiter = 1813412ed095SKrasimir Georgiev getCanonicalRawStringDelimiter(Style, RawStringStyle.Language); 1814587859d9SKrasimir Georgiev if (NewDelimiter.empty()) 1815412ed095SKrasimir Georgiev NewDelimiter = OldDelimiter; 18169ad83fe7SKrasimir Georgiev // The text of a raw string is between the leading 'R"delimiter(' and the 18179ad83fe7SKrasimir Georgiev // trailing 'delimiter)"'. 1818412ed095SKrasimir Georgiev unsigned OldPrefixSize = 3 + OldDelimiter.size(); 1819412ed095SKrasimir Georgiev unsigned OldSuffixSize = 2 + OldDelimiter.size(); 1820412ed095SKrasimir Georgiev // We create a virtual text environment which expects a null-terminated 1821412ed095SKrasimir Georgiev // string, so we cannot use StringRef. 1822adcd0268SBenjamin Kramer std::string RawText = std::string( 1823adcd0268SBenjamin Kramer Current.TokenText.substr(OldPrefixSize).drop_back(OldSuffixSize)); 1824412ed095SKrasimir Georgiev if (NewDelimiter != OldDelimiter) { 1825412ed095SKrasimir Georgiev // Don't update to the canonical delimiter 'deli' if ')deli"' occurs in the 1826412ed095SKrasimir Georgiev // raw string. 1827412ed095SKrasimir Georgiev std::string CanonicalDelimiterSuffix = (")" + NewDelimiter + "\"").str(); 1828412ed095SKrasimir Georgiev if (StringRef(RawText).contains(CanonicalDelimiterSuffix)) 1829412ed095SKrasimir Georgiev NewDelimiter = OldDelimiter; 1830412ed095SKrasimir Georgiev } 18319ad83fe7SKrasimir Georgiev 1832412ed095SKrasimir Georgiev unsigned NewPrefixSize = 3 + NewDelimiter.size(); 1833412ed095SKrasimir Georgiev unsigned NewSuffixSize = 2 + NewDelimiter.size(); 1834412ed095SKrasimir Georgiev 1835412ed095SKrasimir Georgiev // The first start column is the column the raw text starts after formatting. 1836412ed095SKrasimir Georgiev unsigned FirstStartColumn = StartColumn + NewPrefixSize; 18379ad83fe7SKrasimir Georgiev 18389ad83fe7SKrasimir Georgiev // The next start column is the intended indentation a line break inside 18399ad83fe7SKrasimir Georgiev // the raw string at level 0. It is determined by the following rules: 18409ad83fe7SKrasimir Georgiev // - if the content starts on newline, it is one level more than the current 18419ad83fe7SKrasimir Georgiev // indent, and 18429ad83fe7SKrasimir Georgiev // - if the content does not start on a newline, it is the first start 18439ad83fe7SKrasimir Georgiev // column. 18449ad83fe7SKrasimir Georgiev // These rules have the advantage that the formatted content both does not 18459ad83fe7SKrasimir Georgiev // violate the rectangle rule and visually flows within the surrounding 18469ad83fe7SKrasimir Georgiev // source. 1847412ed095SKrasimir Georgiev bool ContentStartsOnNewline = Current.TokenText[OldPrefixSize] == '\n'; 1848128fcffbSKrasimir Georgiev // If this token is the last parameter (checked by looking if it's followed by 1849bda8482aSKrasimir Georgiev // `)` and is not on a newline, the base the indent off the line's nested 1850bda8482aSKrasimir Georgiev // block indent. Otherwise, base the indent off the arguments indent, so we 1851bda8482aSKrasimir Georgiev // can achieve: 1852bda8482aSKrasimir Georgiev // 1853128fcffbSKrasimir Georgiev // fffffffffff(1, 2, 3, R"pb( 1854128fcffbSKrasimir Georgiev // key1: 1 # 1855128fcffbSKrasimir Georgiev // key2: 2)pb"); 1856128fcffbSKrasimir Georgiev // 1857128fcffbSKrasimir Georgiev // fffffffffff(1, 2, 3, 1858128fcffbSKrasimir Georgiev // R"pb( 1859128fcffbSKrasimir Georgiev // key1: 1 # 1860128fcffbSKrasimir Georgiev // key2: 2 1861bda8482aSKrasimir Georgiev // )pb"); 1862bda8482aSKrasimir Georgiev // 1863bda8482aSKrasimir Georgiev // fffffffffff(1, 2, 3, 1864bda8482aSKrasimir Georgiev // R"pb( 1865bda8482aSKrasimir Georgiev // key1: 1 # 1866bda8482aSKrasimir Georgiev // key2: 2 1867128fcffbSKrasimir Georgiev // )pb", 1868128fcffbSKrasimir Georgiev // 5); 1869bda8482aSKrasimir Georgiev unsigned CurrentIndent = 1870bda8482aSKrasimir Georgiev (!Newline && Current.Next && Current.Next->is(tok::r_paren)) 1871128fcffbSKrasimir Georgiev ? State.Stack.back().NestedBlockIndent 1872128fcffbSKrasimir Georgiev : State.Stack.back().Indent; 1873128fcffbSKrasimir Georgiev unsigned NextStartColumn = ContentStartsOnNewline 1874128fcffbSKrasimir Georgiev ? CurrentIndent + Style.IndentWidth 18759ad83fe7SKrasimir Georgiev : FirstStartColumn; 18769ad83fe7SKrasimir Georgiev 18779ad83fe7SKrasimir Georgiev // The last start column is the column the raw string suffix starts if it is 18789ad83fe7SKrasimir Georgiev // put on a newline. 18799ad83fe7SKrasimir Georgiev // The last start column is the intended indentation of the raw string postfix 18809ad83fe7SKrasimir Georgiev // if it is put on a newline. It is determined by the following rules: 18819ad83fe7SKrasimir Georgiev // - if the raw string prefix starts on a newline, it is the column where 18829ad83fe7SKrasimir Georgiev // that raw string prefix starts, and 18839ad83fe7SKrasimir Georgiev // - if the raw string prefix does not start on a newline, it is the current 18849ad83fe7SKrasimir Georgiev // indent. 18855bcf99b4SPaul Hoad unsigned LastStartColumn = 18865bcf99b4SPaul Hoad Current.NewlinesBefore ? FirstStartColumn - NewPrefixSize : CurrentIndent; 18879ad83fe7SKrasimir Georgiev 18889ad83fe7SKrasimir Georgiev std::pair<tooling::Replacements, unsigned> Fixes = internal::reformat( 18899ad83fe7SKrasimir Georgiev RawStringStyle, RawText, {tooling::Range(0, RawText.size())}, 18909ad83fe7SKrasimir Georgiev FirstStartColumn, NextStartColumn, LastStartColumn, "<stdin>", 18910fcb580bSKrasimir Georgiev /*Status=*/nullptr); 18929ad83fe7SKrasimir Georgiev 18939ad83fe7SKrasimir Georgiev auto NewCode = applyAllReplacements(RawText, Fixes.first); 18949ad83fe7SKrasimir Georgiev tooling::Replacements NoFixes; 1895d079995dSMarek Kurdej if (!NewCode) 1896128fcffbSKrasimir Georgiev return addMultilineToken(Current, State); 18979ad83fe7SKrasimir Georgiev if (!DryRun) { 1898412ed095SKrasimir Georgiev if (NewDelimiter != OldDelimiter) { 1899412ed095SKrasimir Georgiev // In 'R"delimiter(...', the delimiter starts 2 characters after the start 1900412ed095SKrasimir Georgiev // of the token. 1901412ed095SKrasimir Georgiev SourceLocation PrefixDelimiterStart = 1902412ed095SKrasimir Georgiev Current.Tok.getLocation().getLocWithOffset(2); 1903412ed095SKrasimir Georgiev auto PrefixErr = Whitespaces.addReplacement(tooling::Replacement( 1904412ed095SKrasimir Georgiev SourceMgr, PrefixDelimiterStart, OldDelimiter.size(), NewDelimiter)); 1905412ed095SKrasimir Georgiev if (PrefixErr) { 1906412ed095SKrasimir Georgiev llvm::errs() 1907412ed095SKrasimir Georgiev << "Failed to update the prefix delimiter of a raw string: " 1908412ed095SKrasimir Georgiev << llvm::toString(std::move(PrefixErr)) << "\n"; 1909412ed095SKrasimir Georgiev } 1910412ed095SKrasimir Georgiev // In 'R"delimiter(...)delimiter"', the suffix delimiter starts at 1911412ed095SKrasimir Georgiev // position length - 1 - |delimiter|. 1912412ed095SKrasimir Georgiev SourceLocation SuffixDelimiterStart = 1913412ed095SKrasimir Georgiev Current.Tok.getLocation().getLocWithOffset(Current.TokenText.size() - 1914412ed095SKrasimir Georgiev 1 - OldDelimiter.size()); 1915412ed095SKrasimir Georgiev auto SuffixErr = Whitespaces.addReplacement(tooling::Replacement( 1916412ed095SKrasimir Georgiev SourceMgr, SuffixDelimiterStart, OldDelimiter.size(), NewDelimiter)); 1917412ed095SKrasimir Georgiev if (SuffixErr) { 1918412ed095SKrasimir Georgiev llvm::errs() 1919412ed095SKrasimir Georgiev << "Failed to update the suffix delimiter of a raw string: " 1920412ed095SKrasimir Georgiev << llvm::toString(std::move(SuffixErr)) << "\n"; 1921412ed095SKrasimir Georgiev } 1922412ed095SKrasimir Georgiev } 19239ad83fe7SKrasimir Georgiev SourceLocation OriginLoc = 1924412ed095SKrasimir Georgiev Current.Tok.getLocation().getLocWithOffset(OldPrefixSize); 19259ad83fe7SKrasimir Georgiev for (const tooling::Replacement &Fix : Fixes.first) { 19269ad83fe7SKrasimir Georgiev auto Err = Whitespaces.addReplacement(tooling::Replacement( 19279ad83fe7SKrasimir Georgiev SourceMgr, OriginLoc.getLocWithOffset(Fix.getOffset()), 19289ad83fe7SKrasimir Georgiev Fix.getLength(), Fix.getReplacementText())); 19299ad83fe7SKrasimir Georgiev if (Err) { 19309ad83fe7SKrasimir Georgiev llvm::errs() << "Failed to reformat raw string: " 19319ad83fe7SKrasimir Georgiev << llvm::toString(std::move(Err)) << "\n"; 19329ad83fe7SKrasimir Georgiev } 19339ad83fe7SKrasimir Georgiev } 19349ad83fe7SKrasimir Georgiev } 19359ad83fe7SKrasimir Georgiev unsigned RawLastLineEndColumn = getLastLineEndColumn( 19369ad83fe7SKrasimir Georgiev *NewCode, FirstStartColumn, Style.TabWidth, Encoding); 1937412ed095SKrasimir Georgiev State.Column = RawLastLineEndColumn + NewSuffixSize; 1938ce00978bSKrasimir Georgiev // Since we're updating the column to after the raw string literal here, we 1939ce00978bSKrasimir Georgiev // have to manually add the penalty for the prefix R"delim( over the column 1940ce00978bSKrasimir Georgiev // limit. 1941ce00978bSKrasimir Georgiev unsigned PrefixExcessCharacters = 19425bcf99b4SPaul Hoad StartColumn + NewPrefixSize > Style.ColumnLimit 19435bcf99b4SPaul Hoad ? StartColumn + NewPrefixSize - Style.ColumnLimit 19445bcf99b4SPaul Hoad : 0; 1945128fcffbSKrasimir Georgiev bool IsMultiline = 1946128fcffbSKrasimir Georgiev ContentStartsOnNewline || (NewCode->find('\n') != std::string::npos); 1947128fcffbSKrasimir Georgiev if (IsMultiline) { 1948128fcffbSKrasimir Georgiev // Break before further function parameters on all levels. 1949545317cbSMarek Kurdej for (ParenState &Paren : State.Stack) 1950545317cbSMarek Kurdej Paren.BreakBeforeParameter = true; 1951128fcffbSKrasimir Georgiev } 1952ce00978bSKrasimir Georgiev return Fixes.second + PrefixExcessCharacters * Style.PenaltyExcessCharacter; 19539ad83fe7SKrasimir Georgiev } 19549ad83fe7SKrasimir Georgiev 1955917f9e04SAlexander Kornienko unsigned ContinuationIndenter::addMultilineToken(const FormatToken &Current, 1956d7b837e7SAlexander Kornienko LineState &State) { 1957d7b837e7SAlexander Kornienko // Break before further function parameters on all levels. 1958545317cbSMarek Kurdej for (ParenState &Paren : State.Stack) 1959545317cbSMarek Kurdej Paren.BreakBeforeParameter = true; 1960d7b837e7SAlexander Kornienko 196139856b71SAlexander Kornienko unsigned ColumnsUsed = State.Column; 1962632abb9bSAlexander Kornienko // We can only affect layout of the first and the last line, so the penalty 1963632abb9bSAlexander Kornienko // for all other lines is constant, and we ignore it. 1964ebb43caaSAlexander Kornienko State.Column = Current.LastLineColumnWidth; 1965632abb9bSAlexander Kornienko 19669fe0e8daSDaniel Jasper if (ColumnsUsed > getColumnLimit(State)) 19679fe0e8daSDaniel Jasper return Style.PenaltyExcessCharacter * (ColumnsUsed - getColumnLimit(State)); 1968d7b837e7SAlexander Kornienko return 0; 1969d7b837e7SAlexander Kornienko } 1970d7b837e7SAlexander Kornienko 197145ab559eSManuel Klimek unsigned ContinuationIndenter::handleEndOfLine(const FormatToken &Current, 197245ab559eSManuel Klimek LineState &State, bool DryRun, 1973bda8482aSKrasimir Georgiev bool AllowBreak, bool Newline) { 197445ab559eSManuel Klimek unsigned Penalty = 0; 19759ad83fe7SKrasimir Georgiev // Compute the raw string style to use in case this is a raw string literal 19769ad83fe7SKrasimir Georgiev // that can be reformatted. 197745ab559eSManuel Klimek auto RawStringStyle = getRawStringStyle(Current, State); 1978dd0c4e1dSDaniel Jasper if (RawStringStyle && !Current.Finalized) { 1979bda8482aSKrasimir Georgiev Penalty = reformatRawStringLiteral(Current, State, *RawStringStyle, DryRun, 1980bda8482aSKrasimir Georgiev Newline); 198145ab559eSManuel Klimek } else if (Current.IsMultiline && Current.isNot(TT_BlockComment)) { 19829ad83fe7SKrasimir Georgiev // Don't break multi-line tokens other than block comments and raw string 19839ad83fe7SKrasimir Georgiev // literals. Instead, just update the state. 198445ab559eSManuel Klimek Penalty = addMultilineToken(Current, State); 198545ab559eSManuel Klimek } else if (State.Line->Type != LT_ImportStatement) { 198645ab559eSManuel Klimek // We generally don't break import statements. 19870b58c328SManuel Klimek LineState OriginalState = State; 19880b58c328SManuel Klimek 19890b58c328SManuel Klimek // Whether we force the reflowing algorithm to stay strictly within the 19900b58c328SManuel Klimek // column limit. 19910b58c328SManuel Klimek bool Strict = false; 19920b58c328SManuel Klimek // Whether the first non-strict attempt at reflowing did intentionally 19930b58c328SManuel Klimek // exceed the column limit. 19940b58c328SManuel Klimek bool Exceeded = false; 19950b58c328SManuel Klimek std::tie(Penalty, Exceeded) = breakProtrudingToken( 19960b58c328SManuel Klimek Current, State, AllowBreak, /*DryRun=*/true, Strict); 19970b58c328SManuel Klimek if (Exceeded) { 19980b58c328SManuel Klimek // If non-strict reflowing exceeds the column limit, try whether strict 19990b58c328SManuel Klimek // reflowing leads to an overall lower penalty. 20000b58c328SManuel Klimek LineState StrictState = OriginalState; 20010b58c328SManuel Klimek unsigned StrictPenalty = 20020b58c328SManuel Klimek breakProtrudingToken(Current, StrictState, AllowBreak, 20030b58c328SManuel Klimek /*DryRun=*/true, /*Strict=*/true) 20040b58c328SManuel Klimek .first; 20050b58c328SManuel Klimek Strict = StrictPenalty <= Penalty; 20060b58c328SManuel Klimek if (Strict) { 20070b58c328SManuel Klimek Penalty = StrictPenalty; 20080b58c328SManuel Klimek State = StrictState; 20090b58c328SManuel Klimek } 20100b58c328SManuel Klimek } 20110b58c328SManuel Klimek if (!DryRun) { 20120b58c328SManuel Klimek // If we're not in dry-run mode, apply the changes with the decision on 20130b58c328SManuel Klimek // strictness made above. 20140b58c328SManuel Klimek breakProtrudingToken(Current, OriginalState, AllowBreak, /*DryRun=*/false, 20150b58c328SManuel Klimek Strict); 20160b58c328SManuel Klimek } 201745ab559eSManuel Klimek } 201845ab559eSManuel Klimek if (State.Column > getColumnLimit(State)) { 201945ab559eSManuel Klimek unsigned ExcessCharacters = State.Column - getColumnLimit(State); 202045ab559eSManuel Klimek Penalty += Style.PenaltyExcessCharacter * ExcessCharacters; 202145ab559eSManuel Klimek } 202245ab559eSManuel Klimek return Penalty; 202345ab559eSManuel Klimek } 2024917f9e04SAlexander Kornienko 20252537e220SKrasimir Georgiev // Returns the enclosing function name of a token, or the empty string if not 20262537e220SKrasimir Georgiev // found. 20272537e220SKrasimir Georgiev static StringRef getEnclosingFunctionName(const FormatToken &Current) { 20282537e220SKrasimir Georgiev // Look for: 'function(' or 'function<templates>(' before Current. 20292537e220SKrasimir Georgiev auto Tok = Current.getPreviousNonComment(); 20302537e220SKrasimir Georgiev if (!Tok || !Tok->is(tok::l_paren)) 20312537e220SKrasimir Georgiev return ""; 20322537e220SKrasimir Georgiev Tok = Tok->getPreviousNonComment(); 20332537e220SKrasimir Georgiev if (!Tok) 20342537e220SKrasimir Georgiev return ""; 20352537e220SKrasimir Georgiev if (Tok->is(TT_TemplateCloser)) { 20362537e220SKrasimir Georgiev Tok = Tok->MatchingParen; 20372537e220SKrasimir Georgiev if (Tok) 20382537e220SKrasimir Georgiev Tok = Tok->getPreviousNonComment(); 20392537e220SKrasimir Georgiev } 20402537e220SKrasimir Georgiev if (!Tok || !Tok->is(tok::identifier)) 20412537e220SKrasimir Georgiev return ""; 20422537e220SKrasimir Georgiev return Tok->TokenText; 20432537e220SKrasimir Georgiev } 20442537e220SKrasimir Georgiev 204545ab559eSManuel Klimek llvm::Optional<FormatStyle> 204645ab559eSManuel Klimek ContinuationIndenter::getRawStringStyle(const FormatToken &Current, 204745ab559eSManuel Klimek const LineState &State) { 204845ab559eSManuel Klimek if (!Current.isStringLiteral()) 204945ab559eSManuel Klimek return None; 205045ab559eSManuel Klimek auto Delimiter = getRawStringDelimiter(Current.TokenText); 205145ab559eSManuel Klimek if (!Delimiter) 205245ab559eSManuel Klimek return None; 20532537e220SKrasimir Georgiev auto RawStringStyle = RawStringFormats.getDelimiterStyle(*Delimiter); 2054*bebf7bdfSowenca if (!RawStringStyle && Delimiter->empty()) { 20552537e220SKrasimir Georgiev RawStringStyle = RawStringFormats.getEnclosingFunctionStyle( 20562537e220SKrasimir Georgiev getEnclosingFunctionName(Current)); 2057*bebf7bdfSowenca } 205845ab559eSManuel Klimek if (!RawStringStyle) 205945ab559eSManuel Klimek return None; 206045ab559eSManuel Klimek RawStringStyle->ColumnLimit = getColumnLimit(State); 206145ab559eSManuel Klimek return RawStringStyle; 206245ab559eSManuel Klimek } 20639885784dSDaniel Jasper 20645bcf99b4SPaul Hoad std::unique_ptr<BreakableToken> 20655bcf99b4SPaul Hoad ContinuationIndenter::createBreakableToken(const FormatToken &Current, 20665bcf99b4SPaul Hoad LineState &State, bool AllowBreak) { 206739856b71SAlexander Kornienko unsigned StartColumn = State.Column - Current.ColumnWidth; 206804b6a081SDaniel Jasper if (Current.isStringLiteral()) { 20698b7881a0Smydeveloperday // FIXME: String literal breaking is currently disabled for C#, Java, Json 20708b7881a0Smydeveloperday // and JavaScript, as it requires strings to be merged using "+" which we 2071cbb726d0SPaul Hoad // don't support. 2072142e79b8Smydeveloperday if (Style.Language == FormatStyle::LK_Java || Style.isJavaScript() || 2073142e79b8Smydeveloperday Style.isCSharp() || Style.isJson() || !Style.BreakStringLiterals || 2074*bebf7bdfSowenca !AllowBreak) { 207545ab559eSManuel Klimek return nullptr; 2076*bebf7bdfSowenca } 2077428f0b14SDaniel Jasper 2078384b40b9SAlexander Kornienko // Don't break string literals inside preprocessor directives (except for 2079384b40b9SAlexander Kornienko // #define directives, as their contents are stored in separate lines and 2080384b40b9SAlexander Kornienko // are not affected by this check). 2081384b40b9SAlexander Kornienko // This way we avoid breaking code with line directives and unknown 2082384b40b9SAlexander Kornienko // preprocessor directives that contain long string literals. 2083384b40b9SAlexander Kornienko if (State.Line->Type == LT_PreprocessorDirective) 208445ab559eSManuel Klimek return nullptr; 2085de0328aeSDaniel Jasper // Exempts unterminated string literals from line breaking. The user will 2086de0328aeSDaniel Jasper // likely want to terminate the string before any line breaking is done. 2087de0328aeSDaniel Jasper if (Current.IsUnterminatedLiteral) 208845ab559eSManuel Klimek return nullptr; 208909051f29SBen Hamilton // Don't break string literals inside Objective-C array literals (doing so 209009051f29SBen Hamilton // raises the warning -Wobjc-string-concatenation). 2091d079995dSMarek Kurdej if (State.Stack.back().IsInsideObjCArrayLiteral) 209209051f29SBen Hamilton return nullptr; 2093de0328aeSDaniel Jasper 209481e3294eSAlexander Kornienko StringRef Text = Current.TokenText; 209581e3294eSAlexander Kornienko StringRef Prefix; 209681e3294eSAlexander Kornienko StringRef Postfix; 209781e3294eSAlexander Kornienko // FIXME: Handle whitespace between '_T', '(', '"..."', and ')'. 209881e3294eSAlexander Kornienko // FIXME: Store Prefix and Suffix (or PrefixLength and SuffixLength to 209981e3294eSAlexander Kornienko // reduce the overhead) for each FormatToken, which is a string, so that we 210081e3294eSAlexander Kornienko // don't run multiple checks here on the hot path. 210181e3294eSAlexander Kornienko if ((Text.endswith(Postfix = "\"") && 2102d4fa2e63SAlexander Kornienko (Text.startswith(Prefix = "@\"") || Text.startswith(Prefix = "\"") || 2103174b0125SDaniel Jasper Text.startswith(Prefix = "u\"") || Text.startswith(Prefix = "U\"") || 2104174b0125SDaniel Jasper Text.startswith(Prefix = "u8\"") || 210581e3294eSAlexander Kornienko Text.startswith(Prefix = "L\""))) || 2106732b6bd4SAlexander Kornienko (Text.startswith(Prefix = "_T(\"") && Text.endswith(Postfix = "\")"))) { 210755c23a10SKrasimir Georgiev // We need this to address the case where there is an unbreakable tail 210855c23a10SKrasimir Georgiev // only if certain other formatting decisions have been taken. The 210955c23a10SKrasimir Georgiev // UnbreakableTailLength of Current is an overapproximation is that case 211055c23a10SKrasimir Georgiev // and we need to be correct here. 211155c23a10SKrasimir Georgiev unsigned UnbreakableTailLength = (State.NextToken && canBreak(State)) 211255c23a10SKrasimir Georgiev ? 0 211355c23a10SKrasimir Georgiev : Current.UnbreakableTailLength; 21142b3d49b6SJonas Devlieghere return std::make_unique<BreakableStringLiteral>( 211555c23a10SKrasimir Georgiev Current, StartColumn, Prefix, Postfix, UnbreakableTailLength, 211655c23a10SKrasimir Georgiev State.Line->InPPDirective, Encoding, Style); 211781e3294eSAlexander Kornienko } 2118acadc8e0SDaniel Jasper } else if (Current.is(TT_BlockComment)) { 211935599fdfSKrasimir Georgiev if (!Style.ReflowComments || 212091834227SKrasimir Georgiev // If a comment token switches formatting, like 212191834227SKrasimir Georgiev // /* clang-format on */, we don't want to break it further, 212291834227SKrasimir Georgiev // but we may still want to adjust its indentation. 2123*bebf7bdfSowenca switchesFormatting(Current)) { 212445ab559eSManuel Klimek return nullptr; 2125*bebf7bdfSowenca } 21262b3d49b6SJonas Devlieghere return std::make_unique<BreakableBlockComment>( 21277d42f3f7SDaniel Jasper Current, StartColumn, Current.OriginalColumn, !Current.Previous, 2128108cbbc2SOwen Pan State.Line->InPPDirective, Encoding, Style, Whitespaces.useCRLF()); 2129a98b7b01SDaniel Jasper } else if (Current.is(TT_LineComment) && 21302145bc02SCraig Topper (Current.Previous == nullptr || 2131a98b7b01SDaniel Jasper Current.Previous->isNot(TT_ImplicitStringLiteral))) { 21324dcfae6aSManuel Klimek bool RegularComments = [&]() { 21334dcfae6aSManuel Klimek for (const FormatToken *T = &Current; T && T->is(TT_LineComment); 2134*bebf7bdfSowenca T = T->Next) { 21354dcfae6aSManuel Klimek if (!(T->TokenText.startswith("//") || T->TokenText.startswith("#"))) 21364dcfae6aSManuel Klimek return false; 2137*bebf7bdfSowenca } 21384dcfae6aSManuel Klimek return true; 21394dcfae6aSManuel Klimek }(); 2140a0a5039dSDaniel Jasper if (!Style.ReflowComments || 214191834227SKrasimir Georgiev CommentPragmasRegex.match(Current.TokenText.substr(2)) || 2142*bebf7bdfSowenca switchesFormatting(Current) || !RegularComments) { 214345ab559eSManuel Klimek return nullptr; 2144*bebf7bdfSowenca } 21452b3d49b6SJonas Devlieghere return std::make_unique<BreakableLineCommentSection>( 21469aa38a06SBjörn Schäpers Current, StartColumn, /*InPPDirective=*/false, Encoding, Style); 214745ab559eSManuel Klimek } 214845ab559eSManuel Klimek return nullptr; 214945ab559eSManuel Klimek } 215045ab559eSManuel Klimek 21510b58c328SManuel Klimek std::pair<unsigned, bool> 21520b58c328SManuel Klimek ContinuationIndenter::breakProtrudingToken(const FormatToken &Current, 21530b58c328SManuel Klimek LineState &State, bool AllowBreak, 21540b58c328SManuel Klimek bool DryRun, bool Strict) { 215593699f4dSManuel Klimek std::unique_ptr<const BreakableToken> Token = 215645ab559eSManuel Klimek createBreakableToken(Current, State, AllowBreak); 215745ab559eSManuel Klimek if (!Token) 21580b58c328SManuel Klimek return {0, false}; 215993699f4dSManuel Klimek assert(Token->getLineCount() > 0); 216045ab559eSManuel Klimek unsigned ColumnLimit = getColumnLimit(State); 216145ab559eSManuel Klimek if (Current.is(TT_LineComment)) { 21623abbb8a4SAlexander Kornienko // We don't insert backslashes when breaking line comments. 21633abbb8a4SAlexander Kornienko ColumnLimit = Style.ColumnLimit; 2164de0328aeSDaniel Jasper } 2165418b4a7bSBjörn Schäpers if (ColumnLimit == 0) { 2166418b4a7bSBjörn Schäpers // To make the rest of the function easier set the column limit to the 2167418b4a7bSBjörn Schäpers // maximum, if there should be no limit. 2168418b4a7bSBjörn Schäpers ColumnLimit = std::numeric_limits<decltype(ColumnLimit)>::max(); 2169418b4a7bSBjörn Schäpers } 21703abbb8a4SAlexander Kornienko if (Current.UnbreakableTailLength >= ColumnLimit) 21710b58c328SManuel Klimek return {0, false}; 217293699f4dSManuel Klimek // ColumnWidth was already accounted into State.Column before calling 217393699f4dSManuel Klimek // breakProtrudingToken. 217493699f4dSManuel Klimek unsigned StartColumn = State.Column - Current.ColumnWidth; 217577866143SManuel Klimek unsigned NewBreakPenalty = Current.isStringLiteral() 217677866143SManuel Klimek ? Style.PenaltyBreakString 217777866143SManuel Klimek : Style.PenaltyBreakComment; 21780b58c328SManuel Klimek // Stores whether we intentionally decide to let a line exceed the column 21790b58c328SManuel Klimek // limit. 21800b58c328SManuel Klimek bool Exceeded = false; 218193699f4dSManuel Klimek // Stores whether we introduce a break anywhere in the token. 218277866143SManuel Klimek bool BreakInserted = Token->introducesBreakBeforeToken(); 218377866143SManuel Klimek // Store whether we inserted a new line break at the end of the previous 218477866143SManuel Klimek // logical line. 218577866143SManuel Klimek bool NewBreakBefore = false; 218691834227SKrasimir Georgiev // We use a conservative reflowing strategy. Reflow starts after a line is 218791834227SKrasimir Georgiev // broken or the corresponding whitespace compressed. Reflow ends as soon as a 218891834227SKrasimir Georgiev // line that doesn't get reflown with the previous line is reached. 218993699f4dSManuel Klimek bool Reflow = false; 219093699f4dSManuel Klimek // Keep track of where we are in the token: 219193699f4dSManuel Klimek // Where we are in the content of the current logical line. 219222d7e6b0SKrasimir Georgiev unsigned TailOffset = 0; 219393699f4dSManuel Klimek // The column number we're currently at. 219493699f4dSManuel Klimek unsigned ContentStartColumn = 219593699f4dSManuel Klimek Token->getContentStartColumn(0, /*Break=*/false); 219693699f4dSManuel Klimek // The number of columns left in the current logical line after TailOffset. 219793699f4dSManuel Klimek unsigned RemainingTokenColumns = 219893699f4dSManuel Klimek Token->getRemainingLength(0, TailOffset, ContentStartColumn); 219993699f4dSManuel Klimek // Adapt the start of the token, for example indent. 220093699f4dSManuel Klimek if (!DryRun) 220193699f4dSManuel Klimek Token->adaptStartOfLine(0, Whitespaces); 220293699f4dSManuel Klimek 22036a5c95bdSKrasimir Georgiev unsigned ContentIndent = 0; 220493699f4dSManuel Klimek unsigned Penalty = 0; 22053538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << "Breaking protruding token at column " 22063538b39eSNicola Zaghen << StartColumn << ".\n"); 2207de0328aeSDaniel Jasper for (unsigned LineIndex = 0, EndIndex = Token->getLineCount(); 2208de0328aeSDaniel Jasper LineIndex != EndIndex; ++LineIndex) { 22093538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 22103538b39eSNicola Zaghen << " Line: " << LineIndex << " (Reflow: " << Reflow << ")\n"); 221177866143SManuel Klimek NewBreakBefore = false; 221293699f4dSManuel Klimek // If we did reflow the previous line, we'll try reflowing again. Otherwise 221393699f4dSManuel Klimek // we'll start reflowing if the current line is broken or whitespace is 221493699f4dSManuel Klimek // compressed. 221593699f4dSManuel Klimek bool TryReflow = Reflow; 221693699f4dSManuel Klimek // Break the current token until we can fit the rest of the line. 221793699f4dSManuel Klimek while (ContentStartColumn + RemainingTokenColumns > ColumnLimit) { 22183538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Over limit, need: " 221993699f4dSManuel Klimek << (ContentStartColumn + RemainingTokenColumns) 222093699f4dSManuel Klimek << ", space: " << ColumnLimit 222193699f4dSManuel Klimek << ", reflown prefix: " << ContentStartColumn 222293699f4dSManuel Klimek << ", offset in line: " << TailOffset << "\n"); 222393699f4dSManuel Klimek // If the current token doesn't fit, find the latest possible split in the 222493699f4dSManuel Klimek // current line so that breaking at it will be under the column limit. 222593699f4dSManuel Klimek // FIXME: Use the earliest possible split while reflowing to correctly 222693699f4dSManuel Klimek // compress whitespace within a line. 222793699f4dSManuel Klimek BreakableToken::Split Split = 222893699f4dSManuel Klimek Token->getSplit(LineIndex, TailOffset, ColumnLimit, 222993699f4dSManuel Klimek ContentStartColumn, CommentPragmasRegex); 2230de0328aeSDaniel Jasper if (Split.first == StringRef::npos) { 223193699f4dSManuel Klimek // No break opportunity - update the penalty and continue with the next 223293699f4dSManuel Klimek // logical line. 2233*bebf7bdfSowenca if (LineIndex < EndIndex - 1) { 2234123ca80bSKrasimir Georgiev // The last line's penalty is handled in addNextStateToQueue() or when 2235123ca80bSKrasimir Georgiev // calling replaceWhitespaceAfterLastLine below. 2236de0328aeSDaniel Jasper Penalty += Style.PenaltyExcessCharacter * 223793699f4dSManuel Klimek (ContentStartColumn + RemainingTokenColumns - ColumnLimit); 2238*bebf7bdfSowenca } 22393538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " No break opportunity.\n"); 2240de0328aeSDaniel Jasper break; 2241de0328aeSDaniel Jasper } 2242de0328aeSDaniel Jasper assert(Split.first != 0); 2243875395f9SAlexander Kornienko 224493699f4dSManuel Klimek if (Token->supportsReflow()) { 224593699f4dSManuel Klimek // Check whether the next natural split point after the current one can 224693699f4dSManuel Klimek // still fit the line, either because we can compress away whitespace, 224793699f4dSManuel Klimek // or because the penalty the excess characters introduce is lower than 224893699f4dSManuel Klimek // the break penalty. 224993699f4dSManuel Klimek // We only do this for tokens that support reflowing, and thus allow us 225093699f4dSManuel Klimek // to change the whitespace arbitrarily (e.g. comments). 225193699f4dSManuel Klimek // Other tokens, like string literals, can be broken on arbitrary 225293699f4dSManuel Klimek // positions. 225377866143SManuel Klimek 225493699f4dSManuel Klimek // First, compute the columns from TailOffset to the next possible split 225593699f4dSManuel Klimek // position. 225693699f4dSManuel Klimek // For example: 225793699f4dSManuel Klimek // ColumnLimit: | 225893699f4dSManuel Klimek // // Some text that breaks 225993699f4dSManuel Klimek // ^ tail offset 226093699f4dSManuel Klimek // ^-- split 226193699f4dSManuel Klimek // ^-------- to split columns 226293699f4dSManuel Klimek // ^--- next split 226393699f4dSManuel Klimek // ^--------------- to next split columns 226493699f4dSManuel Klimek unsigned ToSplitColumns = Token->getRangeLength( 226593699f4dSManuel Klimek LineIndex, TailOffset, Split.first, ContentStartColumn); 22663538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " ToSplit: " << ToSplitColumns << "\n"); 226793699f4dSManuel Klimek 226893699f4dSManuel Klimek BreakableToken::Split NextSplit = Token->getSplit( 226993699f4dSManuel Klimek LineIndex, TailOffset + Split.first + Split.second, ColumnLimit, 227093699f4dSManuel Klimek ContentStartColumn + ToSplitColumns + 1, CommentPragmasRegex); 227193699f4dSManuel Klimek // Compute the columns necessary to fit the next non-breakable sequence 227293699f4dSManuel Klimek // into the current line. 227393699f4dSManuel Klimek unsigned ToNextSplitColumns = 0; 227493699f4dSManuel Klimek if (NextSplit.first == StringRef::npos) { 227593699f4dSManuel Klimek ToNextSplitColumns = Token->getRemainingLength(LineIndex, TailOffset, 227693699f4dSManuel Klimek ContentStartColumn); 227793699f4dSManuel Klimek } else { 227893699f4dSManuel Klimek ToNextSplitColumns = Token->getRangeLength( 227993699f4dSManuel Klimek LineIndex, TailOffset, 228093699f4dSManuel Klimek Split.first + Split.second + NextSplit.first, ContentStartColumn); 228193699f4dSManuel Klimek } 228293699f4dSManuel Klimek // Compress the whitespace between the break and the start of the next 228393699f4dSManuel Klimek // unbreakable sequence. 228493699f4dSManuel Klimek ToNextSplitColumns = 228593699f4dSManuel Klimek Token->getLengthAfterCompression(ToNextSplitColumns, Split); 22863538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 22873538b39eSNicola Zaghen << " ContentStartColumn: " << ContentStartColumn << "\n"); 22883538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 22893538b39eSNicola Zaghen << " ToNextSplit: " << ToNextSplitColumns << "\n"); 229093699f4dSManuel Klimek // If the whitespace compression makes us fit, continue on the current 229193699f4dSManuel Klimek // line. 229293699f4dSManuel Klimek bool ContinueOnLine = 229393699f4dSManuel Klimek ContentStartColumn + ToNextSplitColumns <= ColumnLimit; 229493699f4dSManuel Klimek unsigned ExcessCharactersPenalty = 0; 22950b58c328SManuel Klimek if (!ContinueOnLine && !Strict) { 229693699f4dSManuel Klimek // Similarly, if the excess characters' penalty is lower than the 229793699f4dSManuel Klimek // penalty of introducing a new break, continue on the current line. 229893699f4dSManuel Klimek ExcessCharactersPenalty = 229993699f4dSManuel Klimek (ContentStartColumn + ToNextSplitColumns - ColumnLimit) * 230077866143SManuel Klimek Style.PenaltyExcessCharacter; 23013538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 230293699f4dSManuel Klimek << " Penalty excess: " << ExcessCharactersPenalty 230393699f4dSManuel Klimek << "\n break : " << NewBreakPenalty << "\n"); 23040b58c328SManuel Klimek if (ExcessCharactersPenalty < NewBreakPenalty) { 23050b58c328SManuel Klimek Exceeded = true; 230693699f4dSManuel Klimek ContinueOnLine = true; 2307875395f9SAlexander Kornienko } 23080b58c328SManuel Klimek } 230993699f4dSManuel Klimek if (ContinueOnLine) { 23103538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Continuing on line...\n"); 231193699f4dSManuel Klimek // The current line fits after compressing the whitespace - reflow 231293699f4dSManuel Klimek // the next line into it if possible. 231393699f4dSManuel Klimek TryReflow = true; 2314*bebf7bdfSowenca if (!DryRun) { 231593699f4dSManuel Klimek Token->compressWhitespace(LineIndex, TailOffset, Split, 231693699f4dSManuel Klimek Whitespaces); 2317*bebf7bdfSowenca } 231893699f4dSManuel Klimek // When we continue on the same line, leave one space between content. 231993699f4dSManuel Klimek ContentStartColumn += ToSplitColumns + 1; 232093699f4dSManuel Klimek Penalty += ExcessCharactersPenalty; 232193699f4dSManuel Klimek TailOffset += Split.first + Split.second; 232293699f4dSManuel Klimek RemainingTokenColumns = Token->getRemainingLength( 232393699f4dSManuel Klimek LineIndex, TailOffset, ContentStartColumn); 232493699f4dSManuel Klimek continue; 232593699f4dSManuel Klimek } 232693699f4dSManuel Klimek } 23273538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Breaking...\n"); 23286a5c95bdSKrasimir Georgiev // Update the ContentIndent only if the current line was not reflown with 23296a5c95bdSKrasimir Georgiev // the previous line, since in that case the previous line should still 23306a5c95bdSKrasimir Georgiev // determine the ContentIndent. Also never intent the last line. 23316a5c95bdSKrasimir Georgiev if (!Reflow) 23326a5c95bdSKrasimir Georgiev ContentIndent = Token->getContentIndent(LineIndex); 23336a5c95bdSKrasimir Georgiev LLVM_DEBUG(llvm::dbgs() 23346a5c95bdSKrasimir Georgiev << " ContentIndent: " << ContentIndent << "\n"); 23356a5c95bdSKrasimir Georgiev ContentStartColumn = ContentIndent + Token->getContentStartColumn( 23366a5c95bdSKrasimir Georgiev LineIndex, /*Break=*/true); 23376a5c95bdSKrasimir Georgiev 233893699f4dSManuel Klimek unsigned NewRemainingTokenColumns = Token->getRemainingLength( 233993699f4dSManuel Klimek LineIndex, TailOffset + Split.first + Split.second, 234093699f4dSManuel Klimek ContentStartColumn); 23416a5c95bdSKrasimir Georgiev if (NewRemainingTokenColumns == 0) { 23426a5c95bdSKrasimir Georgiev // No content to indent. 23436a5c95bdSKrasimir Georgiev ContentIndent = 0; 23446a5c95bdSKrasimir Georgiev ContentStartColumn = 23456a5c95bdSKrasimir Georgiev Token->getContentStartColumn(LineIndex, /*Break=*/true); 23466a5c95bdSKrasimir Georgiev NewRemainingTokenColumns = Token->getRemainingLength( 23476a5c95bdSKrasimir Georgiev LineIndex, TailOffset + Split.first + Split.second, 23486a5c95bdSKrasimir Georgiev ContentStartColumn); 23496a5c95bdSKrasimir Georgiev } 235091834227SKrasimir Georgiev 235164a42b8dSAlexander Kornienko // When breaking before a tab character, it may be moved by a few columns, 235264a42b8dSAlexander Kornienko // but will still be expanded to the next tab stop, so we don't save any 235364a42b8dSAlexander Kornienko // columns. 235484bf5e32SManuel Klimek if (NewRemainingTokenColumns >= RemainingTokenColumns) { 235577866143SManuel Klimek // FIXME: Do we need to adjust the penalty? 235664a42b8dSAlexander Kornienko break; 235793699f4dSManuel Klimek } 235877866143SManuel Klimek 23593538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Breaking at: " << TailOffset + Split.first 236093699f4dSManuel Klimek << ", " << Split.second << "\n"); 2361*bebf7bdfSowenca if (!DryRun) { 23626a5c95bdSKrasimir Georgiev Token->insertBreak(LineIndex, TailOffset, Split, ContentIndent, 23636a5c95bdSKrasimir Georgiev Whitespaces); 2364*bebf7bdfSowenca } 236577866143SManuel Klimek 236693699f4dSManuel Klimek Penalty += NewBreakPenalty; 2367de0328aeSDaniel Jasper TailOffset += Split.first + Split.second; 2368de0328aeSDaniel Jasper RemainingTokenColumns = NewRemainingTokenColumns; 2369de0328aeSDaniel Jasper BreakInserted = true; 237077866143SManuel Klimek NewBreakBefore = true; 2371de0328aeSDaniel Jasper } 237293699f4dSManuel Klimek // In case there's another line, prepare the state for the start of the next 237393699f4dSManuel Klimek // line. 237493699f4dSManuel Klimek if (LineIndex + 1 != EndIndex) { 237593699f4dSManuel Klimek unsigned NextLineIndex = LineIndex + 1; 2376*bebf7bdfSowenca if (NewBreakBefore) { 237793699f4dSManuel Klimek // After breaking a line, try to reflow the next line into the current 237893699f4dSManuel Klimek // one once RemainingTokenColumns fits. 237993699f4dSManuel Klimek TryReflow = true; 2380*bebf7bdfSowenca } 238193699f4dSManuel Klimek if (TryReflow) { 238293699f4dSManuel Klimek // We decided that we want to try reflowing the next line into the 238393699f4dSManuel Klimek // current one. 238493699f4dSManuel Klimek // We will now adjust the state as if the reflow is successful (in 238593699f4dSManuel Klimek // preparation for the next line), and see whether that works. If we 238693699f4dSManuel Klimek // decide that we cannot reflow, we will later reset the state to the 238793699f4dSManuel Klimek // start of the next line. 238893699f4dSManuel Klimek Reflow = false; 238993699f4dSManuel Klimek // As we did not continue breaking the line, RemainingTokenColumns is 239093699f4dSManuel Klimek // known to fit after ContentStartColumn. Adapt ContentStartColumn to 239193699f4dSManuel Klimek // the position at which we want to format the next line if we do 239293699f4dSManuel Klimek // actually reflow. 239393699f4dSManuel Klimek // When we reflow, we need to add a space between the end of the current 239493699f4dSManuel Klimek // line and the next line's start column. 239593699f4dSManuel Klimek ContentStartColumn += RemainingTokenColumns + 1; 239693699f4dSManuel Klimek // Get the split that we need to reflow next logical line into the end 239793699f4dSManuel Klimek // of the current one; the split will include any leading whitespace of 239893699f4dSManuel Klimek // the next logical line. 239993699f4dSManuel Klimek BreakableToken::Split SplitBeforeNext = 240093699f4dSManuel Klimek Token->getReflowSplit(NextLineIndex, CommentPragmasRegex); 24013538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 24023538b39eSNicola Zaghen << " Size of reflown text: " << ContentStartColumn 240393699f4dSManuel Klimek << "\n Potential reflow split: "); 240493699f4dSManuel Klimek if (SplitBeforeNext.first != StringRef::npos) { 24053538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << SplitBeforeNext.first << ", " 240693699f4dSManuel Klimek << SplitBeforeNext.second << "\n"); 240793699f4dSManuel Klimek TailOffset = SplitBeforeNext.first + SplitBeforeNext.second; 240893699f4dSManuel Klimek // If the rest of the next line fits into the current line below the 240993699f4dSManuel Klimek // column limit, we can safely reflow. 241093699f4dSManuel Klimek RemainingTokenColumns = Token->getRemainingLength( 241193699f4dSManuel Klimek NextLineIndex, TailOffset, ContentStartColumn); 241293699f4dSManuel Klimek Reflow = true; 241393699f4dSManuel Klimek if (ContentStartColumn + RemainingTokenColumns > ColumnLimit) { 24143538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() 24153538b39eSNicola Zaghen << " Over limit after reflow, need: " 241693699f4dSManuel Klimek << (ContentStartColumn + RemainingTokenColumns) 241793699f4dSManuel Klimek << ", space: " << ColumnLimit 241893699f4dSManuel Klimek << ", reflown prefix: " << ContentStartColumn 241993699f4dSManuel Klimek << ", offset in line: " << TailOffset << "\n"); 242093699f4dSManuel Klimek // If the whole next line does not fit, try to find a point in 242193699f4dSManuel Klimek // the next line at which we can break so that attaching the part 242293699f4dSManuel Klimek // of the next line to that break point onto the current line is 242393699f4dSManuel Klimek // below the column limit. 242493699f4dSManuel Klimek BreakableToken::Split Split = 242593699f4dSManuel Klimek Token->getSplit(NextLineIndex, TailOffset, ColumnLimit, 242693699f4dSManuel Klimek ContentStartColumn, CommentPragmasRegex); 242793699f4dSManuel Klimek if (Split.first == StringRef::npos) { 24283538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Did not find later break\n"); 242993699f4dSManuel Klimek Reflow = false; 243093699f4dSManuel Klimek } else { 243193699f4dSManuel Klimek // Check whether the first split point gets us below the column 243293699f4dSManuel Klimek // limit. Note that we will execute this split below as part of 243393699f4dSManuel Klimek // the normal token breaking and reflow logic within the line. 243493699f4dSManuel Klimek unsigned ToSplitColumns = Token->getRangeLength( 243593699f4dSManuel Klimek NextLineIndex, TailOffset, Split.first, ContentStartColumn); 243693699f4dSManuel Klimek if (ContentStartColumn + ToSplitColumns > ColumnLimit) { 24373538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Next split protrudes, need: " 243893699f4dSManuel Klimek << (ContentStartColumn + ToSplitColumns) 243993699f4dSManuel Klimek << ", space: " << ColumnLimit); 244093699f4dSManuel Klimek unsigned ExcessCharactersPenalty = 244193699f4dSManuel Klimek (ContentStartColumn + ToSplitColumns - ColumnLimit) * 244293699f4dSManuel Klimek Style.PenaltyExcessCharacter; 2443d079995dSMarek Kurdej if (NewBreakPenalty < ExcessCharactersPenalty) 244493699f4dSManuel Klimek Reflow = false; 244593699f4dSManuel Klimek } 244693699f4dSManuel Klimek } 244793699f4dSManuel Klimek } 244893699f4dSManuel Klimek } else { 24493538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << "not found.\n"); 245093699f4dSManuel Klimek } 245193699f4dSManuel Klimek } 245293699f4dSManuel Klimek if (!Reflow) { 245393699f4dSManuel Klimek // If we didn't reflow into the next line, the only space to consider is 245493699f4dSManuel Klimek // the next logical line. Reset our state to match the start of the next 245593699f4dSManuel Klimek // line. 245693699f4dSManuel Klimek TailOffset = 0; 245793699f4dSManuel Klimek ContentStartColumn = 245893699f4dSManuel Klimek Token->getContentStartColumn(NextLineIndex, /*Break=*/false); 245993699f4dSManuel Klimek RemainingTokenColumns = Token->getRemainingLength( 246093699f4dSManuel Klimek NextLineIndex, TailOffset, ContentStartColumn); 246193699f4dSManuel Klimek // Adapt the start of the token, for example indent. 246293699f4dSManuel Klimek if (!DryRun) 246393699f4dSManuel Klimek Token->adaptStartOfLine(NextLineIndex, Whitespaces); 246493699f4dSManuel Klimek } else { 246593699f4dSManuel Klimek // If we found a reflow split and have added a new break before the next 246693699f4dSManuel Klimek // line, we are going to remove the line break at the start of the next 246793699f4dSManuel Klimek // logical line. For example, here we'll add a new line break after 246893699f4dSManuel Klimek // 'text', and subsequently delete the line break between 'that' and 246993699f4dSManuel Klimek // 'reflows'. 247093699f4dSManuel Klimek // // some text that 247193699f4dSManuel Klimek // // reflows 247293699f4dSManuel Klimek // -> 247393699f4dSManuel Klimek // // some text 247493699f4dSManuel Klimek // // that reflows 247593699f4dSManuel Klimek // When adding the line break, we also added the penalty for it, so we 247693699f4dSManuel Klimek // need to subtract that penalty again when we remove the line break due 247793699f4dSManuel Klimek // to reflowing. 247893699f4dSManuel Klimek if (NewBreakBefore) { 247993699f4dSManuel Klimek assert(Penalty >= NewBreakPenalty); 248093699f4dSManuel Klimek Penalty -= NewBreakPenalty; 248193699f4dSManuel Klimek } 248293699f4dSManuel Klimek if (!DryRun) 248393699f4dSManuel Klimek Token->reflow(NextLineIndex, Whitespaces); 248493699f4dSManuel Klimek } 248593699f4dSManuel Klimek } 2486de0328aeSDaniel Jasper } 2487de0328aeSDaniel Jasper 24883b86534aSKrasimir Georgiev BreakableToken::Split SplitAfterLastLine = 248993699f4dSManuel Klimek Token->getSplitAfterLastLine(TailOffset); 249022d7e6b0SKrasimir Georgiev if (SplitAfterLastLine.first != StringRef::npos) { 24913538b39eSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << "Replacing whitespace after last line.\n"); 2492123ca80bSKrasimir Georgiev 2493123ca80bSKrasimir Georgiev // We add the last line's penalty here, since that line is going to be split 2494123ca80bSKrasimir Georgiev // now. 2495123ca80bSKrasimir Georgiev Penalty += Style.PenaltyExcessCharacter * 2496123ca80bSKrasimir Georgiev (ContentStartColumn + RemainingTokenColumns - ColumnLimit); 2497123ca80bSKrasimir Georgiev 2498*bebf7bdfSowenca if (!DryRun) { 249922d7e6b0SKrasimir Georgiev Token->replaceWhitespaceAfterLastLine(TailOffset, SplitAfterLastLine, 250022d7e6b0SKrasimir Georgiev Whitespaces); 2501*bebf7bdfSowenca } 250293699f4dSManuel Klimek ContentStartColumn = 250393699f4dSManuel Klimek Token->getContentStartColumn(Token->getLineCount() - 1, /*Break=*/true); 250493699f4dSManuel Klimek RemainingTokenColumns = Token->getRemainingLength( 250593699f4dSManuel Klimek Token->getLineCount() - 1, 250693699f4dSManuel Klimek TailOffset + SplitAfterLastLine.first + SplitAfterLastLine.second, 250793699f4dSManuel Klimek ContentStartColumn); 250822d7e6b0SKrasimir Georgiev } 250922d7e6b0SKrasimir Georgiev 251093699f4dSManuel Klimek State.Column = ContentStartColumn + RemainingTokenColumns - 251193699f4dSManuel Klimek Current.UnbreakableTailLength; 2512de0328aeSDaniel Jasper 2513de0328aeSDaniel Jasper if (BreakInserted) { 2514de0328aeSDaniel Jasper // If we break the token inside a parameter list, we need to break before 2515de0328aeSDaniel Jasper // the next parameter on all levels, so that the next parameter is clearly 2516de0328aeSDaniel Jasper // visible. Line comments already introduce a break. 2517d079995dSMarek Kurdej if (Current.isNot(TT_LineComment)) 2518545317cbSMarek Kurdej for (ParenState &Paren : State.Stack) 2519545317cbSMarek Kurdej Paren.BreakBeforeParameter = true; 2520de0328aeSDaniel Jasper 252135599fdfSKrasimir Georgiev if (Current.is(TT_BlockComment)) 252235599fdfSKrasimir Georgiev State.NoContinuation = true; 252335599fdfSKrasimir Georgiev 2524de0328aeSDaniel Jasper State.Stack.back().LastSpace = StartColumn; 2525de0328aeSDaniel Jasper } 252691834227SKrasimir Georgiev 252791834227SKrasimir Georgiev Token->updateNextToken(State); 252891834227SKrasimir Georgiev 25290b58c328SManuel Klimek return {Penalty, Exceeded}; 2530de0328aeSDaniel Jasper } 2531de0328aeSDaniel Jasper 25329fe0e8daSDaniel Jasper unsigned ContinuationIndenter::getColumnLimit(const LineState &State) const { 2533de0328aeSDaniel Jasper // In preprocessor directives reserve two chars for trailing " \" 25349fe0e8daSDaniel Jasper return Style.ColumnLimit - (State.Line->InPPDirective ? 2 : 0); 2535de0328aeSDaniel Jasper } 2536de0328aeSDaniel Jasper 2537c39b56feSDaniel Jasper bool ContinuationIndenter::nextIsMultilineString(const LineState &State) { 2538f438cb76SDaniel Jasper const FormatToken &Current = *State.NextToken; 2539a98b7b01SDaniel Jasper if (!Current.isStringLiteral() || Current.is(TT_ImplicitStringLiteral)) 2540f438cb76SDaniel Jasper return false; 2541d7b837e7SAlexander Kornienko // We never consider raw string literals "multiline" for the purpose of 2542c39b56feSDaniel Jasper // AlwaysBreakBeforeMultilineStrings implementation as they are special-cased 2543c39b56feSDaniel Jasper // (see TokenAnnotator::mustBreakBefore(). 2544d7b837e7SAlexander Kornienko if (Current.TokenText.startswith("R\"")) 2545d7b837e7SAlexander Kornienko return false; 254639856b71SAlexander Kornienko if (Current.IsMultiline) 2547d7b837e7SAlexander Kornienko return true; 2548f438cb76SDaniel Jasper if (Current.getNextNonComment() && 2549*bebf7bdfSowenca Current.getNextNonComment()->isStringLiteral()) { 2550f438cb76SDaniel Jasper return true; // Implicit concatenation. 2551*bebf7bdfSowenca } 2552374e6de8SKrasimir Georgiev if (Style.ColumnLimit != 0 && Style.BreakStringLiterals && 25536fd5d646SDaniel Jasper State.Column + Current.ColumnWidth + Current.UnbreakableTailLength > 2554*bebf7bdfSowenca Style.ColumnLimit) { 2555f438cb76SDaniel Jasper return true; // String will be split. 2556*bebf7bdfSowenca } 2557d7b837e7SAlexander Kornienko return false; 2558f438cb76SDaniel Jasper } 2559f438cb76SDaniel Jasper 2560de0328aeSDaniel Jasper } // namespace format 2561de0328aeSDaniel Jasper } // namespace clang 2562