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