Index: clang/lib/Format/ContinuationIndenter.cpp =================================================================== --- clang/lib/Format/ContinuationIndenter.cpp +++ clang/lib/Format/ContinuationIndenter.cpp @@ -677,7 +677,8 @@ if (Current.isNot(tok::comment) && Previous.is(tok::l_paren) && Previous.Previous && (Previous.Previous->isOneOf(tok::kw_if, tok::kw_for) || - Previous.Previous->endsSequence(tok::kw_constexpr, tok::kw_if))) { + Previous.Previous->endsSequence(tok::kw_constexpr, tok::kw_if) || + Previous.Previous->endsSequence(tok::identifier, tok::kw_if))) { // Treat the condition inside an if as if it was a second function // parameter, i.e. let nested calls have a continuation indent. State.Stack.back().LastSpace = State.Column; Index: clang/lib/Format/FormatToken.h =================================================================== --- clang/lib/Format/FormatToken.h +++ clang/lib/Format/FormatToken.h @@ -344,6 +344,10 @@ /// \c true if this token ends a sequence with the given tokens in order, /// following the ``Previous`` pointers, ignoring comments. + /// For example, given tokens [T1, T2, T3], the function returns true if + /// 3 tokens ending at this (ignoring comments) are [T3, T2, T1]. In other + /// words, the tokens passed to this function need to the reverse of the + /// order the tokens appear in code. template bool endsSequence(A K1, Ts... Tokens) const { return endsSequenceInternal(K1, Tokens...); Index: clang/lib/Format/TokenAnnotator.cpp =================================================================== --- clang/lib/Format/TokenAnnotator.cpp +++ clang/lib/Format/TokenAnnotator.cpp @@ -178,6 +178,7 @@ tok::kw_if, tok::kw_while, tok::l_paren, tok::comma) || Left->Previous->endsSequence(tok::kw_constexpr, tok::kw_if) || + Left->Previous->endsSequence(tok::identifier, tok::kw_if) || Left->Previous->is(TT_BinaryOperator))) { // static_assert, if and while usually contain expressions. Contexts.back().IsExpression = true; @@ -825,8 +826,9 @@ break; case tok::kw_if: case tok::kw_while: + assert(!Line.startsWith(tok::hash)); if (Tok->is(tok::kw_if) && CurrentToken && - CurrentToken->is(tok::kw_constexpr)) + CurrentToken->isOneOf(tok::kw_constexpr, tok::identifier)) next(); if (CurrentToken && CurrentToken->is(tok::l_paren)) { next(); @@ -1078,6 +1080,7 @@ case tok::pp_if: case tok::pp_elif: Contexts.back().IsExpression = true; + next(); parseLine(); break; default: @@ -2410,7 +2413,8 @@ return 100; if (Left.is(tok::l_paren) && Left.Previous && (Left.Previous->isOneOf(tok::kw_if, tok::kw_for) || - Left.Previous->endsSequence(tok::kw_constexpr, tok::kw_if))) + Left.Previous->endsSequence(tok::kw_constexpr, tok::kw_if) || + Left.Previous->endsSequence(tok::identifier, tok::kw_if))) return 1000; if (Left.is(tok::equal) && InFunctionDecl) return 110; @@ -2615,6 +2619,8 @@ tok::kw_switch, tok::kw_case, TT_ForEachMacro, TT_ObjCForIn) || Left.endsSequence(tok::kw_constexpr, tok::kw_if) || + (Left.endsSequence(tok::identifier, tok::kw_if) && + Line.Type != LT_PreprocessorDirective) || (Left.isOneOf(tok::kw_try, Keywords.kw___except, tok::kw_catch, tok::kw_new, tok::kw_delete) && (!Left.Previous || Left.Previous->isNot(tok::period))))) || Index: clang/lib/Format/UnwrappedLineParser.cpp =================================================================== --- clang/lib/Format/UnwrappedLineParser.cpp +++ clang/lib/Format/UnwrappedLineParser.cpp @@ -1756,7 +1756,7 @@ void UnwrappedLineParser::parseIfThenElse() { assert(FormatTok->Tok.is(tok::kw_if) && "'if' expected"); nextToken(); - if (FormatTok->Tok.is(tok::kw_constexpr)) + if (FormatTok->Tok.isOneOf(tok::kw_constexpr, tok::identifier)) nextToken(); if (FormatTok->Tok.is(tok::l_paren)) parseParens(); Index: clang/unittests/Format/FormatTest.cpp =================================================================== --- clang/unittests/Format/FormatTest.cpp +++ clang/unittests/Format/FormatTest.cpp @@ -426,16 +426,28 @@ verifyFormat("if (a)\n if (b) {\n f();\n }\ng();"); verifyFormat("if constexpr (true)\n" " f();\ng();"); + verifyFormat("if CONSTEXPR (true)\n" + " f();\ng();"); verifyFormat("if constexpr (a)\n" " if constexpr (b)\n" " if constexpr (c)\n" " g();\n" "h();"); + verifyFormat("if CONSTEXPR (a)\n" + " if CONSTEXPR (b)\n" + " if CONSTEXPR (c)\n" + " g();\n" + "h();"); verifyFormat("if constexpr (a)\n" " if constexpr (b) {\n" " f();\n" " }\n" "g();"); + verifyFormat("if CONSTEXPR (a)\n" + " if CONSTEXPR (b) {\n" + " f();\n" + " }\n" + "g();"); FormatStyle AllowsMergedIf = getLLVMStyle(); AllowsMergedIf.AlignEscapedNewlines = FormatStyle::ENAS_Left; @@ -561,10 +573,12 @@ verifyFormat("if (true) {}", AllowSimpleBracedStatements); verifyFormat("if constexpr (true) {}", AllowSimpleBracedStatements); + verifyFormat("if CONSTEXPR (true) {}", AllowSimpleBracedStatements); verifyFormat("while (true) {}", AllowSimpleBracedStatements); verifyFormat("for (;;) {}", AllowSimpleBracedStatements); verifyFormat("if (true) { f(); }", AllowSimpleBracedStatements); verifyFormat("if constexpr (true) { f(); }", AllowSimpleBracedStatements); + verifyFormat("if CONSTEXPR (true) { f(); }", AllowSimpleBracedStatements); verifyFormat("while (true) { f(); }", AllowSimpleBracedStatements); verifyFormat("for (;;) { f(); }", AllowSimpleBracedStatements); verifyFormat("if (true) {\n" @@ -633,10 +647,12 @@ verifyFormat("if (true) {}", AllowSimpleBracedStatements); verifyFormat("if constexpr (true) {}", AllowSimpleBracedStatements); + verifyFormat("if CONSTEXPR (true) {}", AllowSimpleBracedStatements); verifyFormat("while (true) {}", AllowSimpleBracedStatements); verifyFormat("for (;;) {}", AllowSimpleBracedStatements); verifyFormat("if (true) { f(); }", AllowSimpleBracedStatements); verifyFormat("if constexpr (true) { f(); }", AllowSimpleBracedStatements); + verifyFormat("if CONSTEXPR (true) { f(); }", AllowSimpleBracedStatements); verifyFormat("while (true) { f(); }", AllowSimpleBracedStatements); verifyFormat("for (;;) { f(); }", AllowSimpleBracedStatements); verifyFormat("if (true)\n" @@ -750,6 +766,19 @@ "else {\n" " i();\n" "}"); + verifyFormat("if (true)\n" + " if CONSTEXPR (true)\n" + " if (true) {\n" + " if CONSTEXPR (true)\n" + " f();\n" + " } else {\n" + " g();\n" + " }\n" + " else\n" + " h();\n" + "else {\n" + " i();\n" + "}"); verifyFormat("void f() {\n" " if (a) {\n" " } else {\n" @@ -771,6 +800,12 @@ " g();\n" "else\n" " h();"); + verifyFormat("if CONSTEXPR (a)\n" + " f();\n" + "else if CONSTEXPR (b)\n" + " g();\n" + "else\n" + " h();"); verifyFormat("if (a) {\n" " f();\n" "}\n" @@ -783,6 +818,14 @@ "} else if (aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa(\n" " aaaaaaaaaaaaaaaaaaaaaaaaaaaa)) {\n" "}"); + verifyFormat("if (a) {\n" + "} else if constexpr (aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa(\n" + " aaaaaaaaaaaaaaaaaaaaaaaaaaaa)) {\n" + "}"); + verifyFormat("if (a) {\n" + "} else if CONSTEXPR (aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa(\n" + " aaaaaaaaaaaaaaaaaaaaaaaaaaaa)) {\n" + "}"); verifyFormat("if (a) {\n" "} else if (\n" " aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa) {\n" @@ -793,6 +836,11 @@ " aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa) {\n" "}", getLLVMStyleWithColumns(62)); + verifyFormat("if (a) {\n" + "} else if CONSTEXPR (\n" + " aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa) {\n" + "}", + getLLVMStyleWithColumns(62)); } TEST_F(FormatTest, FormatsForLoop) { @@ -3725,6 +3773,9 @@ verifyFormat("if constexpr ((aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa ||\n" " bbbbbbbbbbbbbbbbbb) && // aaaaaaaaaaa\n" " cccccc) {\n}"); + verifyFormat("if CONSTEXPR ((aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa ||\n" + " bbbbbbbbbbbbbbbbbb) && // aaaaaaaaaaa\n" + " cccccc) {\n}"); verifyFormat("b = a &&\n" " // Comment\n" " b.c && d;"); @@ -3812,6 +3863,14 @@ "} else if (aaaaa && bbbbb > // break\n" " ccccc) {\n" "}"); + verifyFormat("if () {\n" + "} else if constexpr (aaaaa && bbbbb > // break\n" + " ccccc) {\n" + "}"); + verifyFormat("if () {\n" + "} else if CONSTEXPR (aaaaa && bbbbb > // break\n" + " ccccc) {\n" + "}"); verifyFormat("if () {\n" "} else if (aaaaa &&\n" " bbbbb > // break\n" @@ -6913,7 +6972,11 @@ verifyIndependentOfContext("if (int *a = &b)"); verifyIndependentOfContext("if (int &a = *b)"); verifyIndependentOfContext("if (a & b[i])"); + verifyIndependentOfContext("if constexpr (a & b[i])"); + verifyIndependentOfContext("if CONSTEXPR (a & b[i])"); verifyIndependentOfContext("if (a * (b * c))"); + verifyIndependentOfContext("if constexpr (a * (b * c))"); + verifyIndependentOfContext("if CONSTEXPR (a * (b * c))"); verifyIndependentOfContext("if (a::b::c::d & b[i])"); verifyIndependentOfContext("if (*b[i])"); verifyIndependentOfContext("if (int *a = (&b))"); @@ -8602,6 +8665,9 @@ verifyFormat("#define A \\\n" " if constexpr (true) return 42;", ShortMergedIf); + verifyFormat("#define A \\\n" + " if CONSTEXPR (true) return 42;", + ShortMergedIf); ShortMergedIf.ColumnLimit = 29; verifyFormat("#define A \\\n" " if (aaaaaaaaaa) return 1; \\\n" @@ -8618,6 +8684,11 @@ " return 1; \\\n" " return 2;", ShortMergedIf); + verifyFormat("#define A \\\n" + " if CONSTEXPR (aaaaaaa) \\\n" + " return 1; \\\n" + " return 2;", + ShortMergedIf); } TEST_F(FormatTest, FormatStarDependingOnContext) { @@ -11181,6 +11252,14 @@ " }\n" "}\n", BreakBeforeBraceShortIfs); + verifyFormat("void f(bool b)\n" + "{\n" + " if CONSTEXPR (b)\n" + " {\n" + " return;\n" + " }\n" + "}\n", + BreakBeforeBraceShortIfs); verifyFormat("void f(bool b)\n" "{\n" " if (b) return;\n" @@ -11191,6 +11270,11 @@ " if constexpr (b) return;\n" "}\n", BreakBeforeBraceShortIfs); + verifyFormat("void f(bool b)\n" + "{\n" + " if CONSTEXPR (b) return;\n" + "}\n", + BreakBeforeBraceShortIfs); verifyFormat("void f(bool b)\n" "{\n" " while (b)\n" @@ -12858,6 +12942,11 @@ " doo_dah();\n" " })) {\n" "}"); + verifyFormat("if CONSTEXPR (blah_blah(whatever, whatever, [] {\n" + " doo_dah();\n" + " doo_dah();\n" + " })) {\n" + "}"); verifyFormat("auto lambda = []() {\n" " int a = 2\n" "#if A\n"