17 #include "llvm/ADT/STLExtras.h"
18 #include "llvm/Support/Debug.h"
19 #include "llvm/Support/raw_ostream.h"
21 #define DEBUG_TYPE "format-parser"
37 class ScopedDeclarationState {
39 ScopedDeclarationState(UnwrappedLine &
Line, std::vector<bool> &
Stack,
40 bool MustBeDeclaration)
41 : Line(Line), Stack(Stack) {
42 Line.MustBeDeclaration = MustBeDeclaration;
43 Stack.push_back(MustBeDeclaration);
45 ~ScopedDeclarationState() {
48 Line.MustBeDeclaration = Stack.back();
50 Line.MustBeDeclaration =
true;
58 class ScopedMacroState :
public FormatTokenSource {
60 ScopedMacroState(UnwrappedLine &Line, FormatTokenSource *&
TokenSource,
67 Line.InPPDirective =
true;
70 ~ScopedMacroState()
override {
73 Line.InPPDirective =
false;
77 FormatToken *getNextToken()
override {
89 FormatToken *setPosition(
unsigned Position)
override {
97 FormatToken *getFakeEOF() {
98 static bool EOFInitialized =
false;
100 if (!EOFInitialized) {
101 FormatTok.Tok.startToken();
103 EOFInitialized =
true;
122 bool SwitchToPreprocessorLines =
false)
123 : Parser(Parser), OriginalLines(Parser.CurrentLines) {
124 if (SwitchToPreprocessorLines)
125 Parser.CurrentLines = &Parser.PreprocessorDirectives;
126 else if (!Parser.Line->Tokens.empty())
127 Parser.CurrentLines = &Parser.Line->Tokens.back().Children;
128 PreBlockLine = std::move(Parser.Line);
129 Parser.Line = llvm::make_unique<UnwrappedLine>();
130 Parser.Line->Level = PreBlockLine->Level;
131 Parser.Line->InPPDirective = PreBlockLine->InPPDirective;
135 if (!
Parser.Line->Tokens.empty()) {
136 Parser.addUnwrappedLine();
138 assert(
Parser.Line->Tokens.empty());
139 Parser.Line = std::move(PreBlockLine);
140 if (
Parser.CurrentLines == &
Parser.PreprocessorDirectives)
141 Parser.MustBreakBeforeNextToken =
true;
142 Parser.CurrentLines = OriginalLines;
148 std::unique_ptr<UnwrappedLine> PreBlockLine;
156 : LineLevel(LineLevel), OldLineLevel(LineLevel) {
158 Parser->addUnwrappedLine();
160 Parser->addUnwrappedLine();
168 unsigned OldLineLevel;
173 class IndexedTokenSource :
public FormatTokenSource {
176 : Tokens(Tokens), Position(-1) {}
178 FormatToken *getNextToken()
override {
183 unsigned getPosition()
override {
184 assert(Position >= 0);
188 FormatToken *setPosition(
unsigned P)
override {
193 void reset() { Position = -1; }
207 CurrentLines(&Lines), Style(Style), Keywords(Keywords), Tokens(nullptr),
208 Callback(Callback), AllTokens(Tokens), PPBranchLevel(-1) {}
210 void UnwrappedLineParser::reset() {
213 CommentsBeforeNextToken.clear();
215 MustBreakBeforeNextToken =
false;
216 PreprocessorDirectives.clear();
217 CurrentLines = &Lines;
218 DeclarationScopeStack.clear();
225 DEBUG(llvm::dbgs() <<
"----\n");
233 pushToken(FormatTok);
243 while (!PPLevelBranchIndex.empty() &&
244 PPLevelBranchIndex.back() + 1 >= PPLevelBranchCount.back()) {
245 PPLevelBranchIndex.resize(PPLevelBranchIndex.size() - 1);
246 PPLevelBranchCount.resize(PPLevelBranchCount.size() - 1);
248 if (!PPLevelBranchIndex.empty()) {
249 ++PPLevelBranchIndex.back();
250 assert(PPLevelBranchIndex.size() == PPLevelBranchCount.size());
251 assert(PPLevelBranchIndex.back() <= PPLevelBranchCount.back());
253 }
while (!PPLevelBranchIndex.empty());
256 void UnwrappedLineParser::parseFile() {
259 bool MustBeDeclaration =
261 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack,
269 void UnwrappedLineParser::parseLevel(
bool HasOpeningBrace) {
270 bool SwitchLabelEncountered =
false;
273 if (FormatTok->Type == TT_MacroBlockBegin) {
275 }
else if (FormatTok->Type == TT_MacroBlockEnd) {
296 case tok::kw_default:
298 if (!SwitchLabelEncountered &&
301 SwitchLabelEncountered =
true;
302 parseStructuralElement();
305 parseStructuralElement();
311 void UnwrappedLineParser::calculateBraceTypes(
bool ExpectClassBody) {
317 FormatToken *Tok = FormatTok;
321 SmallVector<FormatToken *, 8> LBraceStack;
322 assert(Tok->Tok.is(tok::l_brace));
325 FormatToken *NextTok;
326 unsigned ReadTokens = 0;
330 }
while (NextTok->is(tok::comment));
332 switch (Tok->Tok.getKind()) {
335 LBraceStack.push_back(Tok);
338 if (!LBraceStack.empty()) {
339 if (LBraceStack.back()->BlockKind ==
BK_Unknown) {
340 bool ProbablyBracedList =
false;
342 ProbablyBracedList = NextTok->isOneOf(tok::comma, tok::r_square);
346 bool NextIsObjCMethod = NextTok->isOneOf(tok::plus, tok::minus) &&
347 NextTok->OriginalColumn == 0;
358 NextTok->isOneOf(tok::comma, tok::period, tok::colon,
359 tok::r_paren, tok::r_square, tok::l_brace,
360 tok::l_paren, tok::ellipsis) ||
361 (NextTok->is(tok::semi) &&
362 (!ExpectClassBody || LBraceStack.size() != 1)) ||
363 (NextTok->isBinaryOperator() && !NextIsObjCMethod);
365 if (ProbablyBracedList) {
370 LBraceStack.back()->BlockKind =
BK_Block;
373 LBraceStack.pop_back();
384 if (!LBraceStack.empty())
385 LBraceStack.back()->BlockKind =
BK_Block;
391 }
while (Tok->Tok.isNot(
tok::eof) && !LBraceStack.empty());
393 for (
unsigned i = 0, e = LBraceStack.size(); i != e; ++i) {
395 LBraceStack[i]->BlockKind =
BK_Block;
401 void UnwrappedLineParser::parseBlock(
bool MustBeDeclaration,
bool AddLevel,
403 assert(FormatTok->isOneOf(tok::l_brace, TT_MacroBlockBegin) &&
404 "'{' or macro block token expected");
405 const bool MacroBlock = FormatTok->
is(TT_MacroBlockBegin);
407 unsigned InitialLevel = Line->Level;
410 if (MacroBlock && FormatTok->is(tok::l_paren))
415 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack,
421 if (MacroBlock ? !FormatTok->is(TT_MacroBlockEnd)
422 : !FormatTok->is(tok::r_brace)) {
423 Line->Level = InitialLevel;
429 if (MacroBlock && FormatTok->is(tok::l_paren))
432 if (MunchSemi && FormatTok->Tok.is(tok::semi))
434 Line->Level = InitialLevel;
440 if (Line.
Tokens.size() < 4)
442 auto I = Line.
Tokens.begin();
443 if (I->Tok->TokenText !=
"goog")
446 if (I->Tok->isNot(tok::period))
449 if (I->Tok->TokenText !=
"scope")
452 return I->Tok->is(tok::l_paren);
459 return InitialToken.
isOneOf(tok::kw_namespace, tok::kw_class);
461 return InitialToken.
isOneOf(tok::kw_class, tok::kw_struct, tok::kw_union);
470 void UnwrappedLineParser::parseChildBlock() {
477 ScopedDeclarationState DeclarationState(*Line, DeclarationScopeStack,
479 Line->Level += GoogScope ? 0 : 1;
481 flushComments(isOnNewLine(*FormatTok));
482 Line->Level -= GoogScope ? 0 : 1;
487 void UnwrappedLineParser::parsePPDirective() {
488 assert(FormatTok->Tok.is(tok::hash) &&
"'#' expected");
489 ScopedMacroState MacroState(*Line, Tokens, FormatTok);
492 if (!FormatTok->Tok.getIdentifierInfo()) {
497 switch (FormatTok->Tok.getIdentifierInfo()->getPPKeywordID()) {
523 void UnwrappedLineParser::conditionalCompilationCondition(
bool Unreachable) {
524 if (Unreachable || (!PPStack.empty() && PPStack.back() == PP_Unreachable))
525 PPStack.push_back(PP_Unreachable);
527 PPStack.push_back(PP_Conditional);
530 void UnwrappedLineParser::conditionalCompilationStart(
bool Unreachable) {
532 assert(PPBranchLevel >= 0 && PPBranchLevel <= (
int)PPLevelBranchIndex.size());
533 if (PPBranchLevel == (
int)PPLevelBranchIndex.size()) {
534 PPLevelBranchIndex.push_back(0);
535 PPLevelBranchCount.push_back(0);
537 PPChainBranchIndex.push(0);
538 bool Skip = PPLevelBranchIndex[PPBranchLevel] > 0;
539 conditionalCompilationCondition(Unreachable || Skip);
542 void UnwrappedLineParser::conditionalCompilationAlternative() {
543 if (!PPStack.empty())
545 assert(PPBranchLevel < (
int)PPLevelBranchIndex.size());
546 if (!PPChainBranchIndex.empty())
547 ++PPChainBranchIndex.top();
548 conditionalCompilationCondition(
549 PPBranchLevel >= 0 && !PPChainBranchIndex.empty() &&
550 PPLevelBranchIndex[PPBranchLevel] != PPChainBranchIndex.top());
553 void UnwrappedLineParser::conditionalCompilationEnd() {
554 assert(PPBranchLevel < (
int)PPLevelBranchIndex.size());
555 if (PPBranchLevel >= 0 && !PPChainBranchIndex.empty()) {
556 if (PPChainBranchIndex.top() + 1 > PPLevelBranchCount[PPBranchLevel]) {
557 PPLevelBranchCount[PPBranchLevel] = PPChainBranchIndex.top() + 1;
561 if (PPBranchLevel > 0)
563 if (!PPChainBranchIndex.empty())
564 PPChainBranchIndex.pop();
565 if (!PPStack.empty())
569 void UnwrappedLineParser::parsePPIf(
bool IfDef) {
571 bool IsLiteralFalse = (FormatTok->Tok.isLiteral() &&
572 FormatTok->Tok.getLiteralData() !=
nullptr &&
573 StringRef(FormatTok->Tok.getLiteralData(),
574 FormatTok->Tok.getLength()) ==
"0") ||
575 FormatTok->Tok.is(tok::kw_false);
576 conditionalCompilationStart(!IfDef && IsLiteralFalse);
580 void UnwrappedLineParser::parsePPElse() {
581 conditionalCompilationAlternative();
585 void UnwrappedLineParser::parsePPElIf() { parsePPElse(); }
587 void UnwrappedLineParser::parsePPEndIf() {
588 conditionalCompilationEnd();
592 void UnwrappedLineParser::parsePPDefine() {
595 if (FormatTok->Tok.getKind() != tok::identifier) {
600 if (FormatTok->Tok.getKind() == tok::l_paren &&
601 FormatTok->WhitespaceRange.getBegin() ==
602 FormatTok->WhitespaceRange.getEnd()) {
616 void UnwrappedLineParser::parsePPUnknown() {
629 return Tok.
isNot(tok::semi) && Tok.
isNot(tok::l_brace) &&
630 Tok.
isNot(tok::l_square) &&
633 Tok.
isNot(tok::period) && Tok.
isNot(tok::periodstar) &&
634 Tok.
isNot(tok::arrow) && Tok.
isNot(tok::arrowstar) &&
635 Tok.
isNot(tok::less) && Tok.
isNot(tok::greater) &&
636 Tok.
isNot(tok::slash) && Tok.
isNot(tok::percent) &&
637 Tok.
isNot(tok::lessless) && Tok.
isNot(tok::greatergreater) &&
638 Tok.
isNot(tok::equal) && Tok.
isNot(tok::plusequal) &&
639 Tok.
isNot(tok::minusequal) && Tok.
isNot(tok::starequal) &&
640 Tok.
isNot(tok::slashequal) && Tok.
isNot(tok::percentequal) &&
641 Tok.
isNot(tok::ampequal) && Tok.
isNot(tok::pipeequal) &&
642 Tok.
isNot(tok::caretequal) && Tok.
isNot(tok::greatergreaterequal) &&
643 Tok.
isNot(tok::lesslessequal) &&
647 Tok.
isNot(tok::colon) &&
649 Tok.
isNot(tok::kw_noexcept);
652 void UnwrappedLineParser::parseStructuralElement() {
653 assert(!FormatTok->Tok.is(tok::l_brace));
654 switch (FormatTok->Tok.getKind()) {
657 if (FormatTok->Tok.is(tok::l_brace)) {
661 switch (FormatTok->Tok.getObjCKeywordID()) {
662 case tok::objc_public:
663 case tok::objc_protected:
664 case tok::objc_package:
665 case tok::objc_private:
666 return parseAccessSpecifier();
667 case tok::objc_interface:
668 case tok::objc_implementation:
669 return parseObjCInterfaceOrImplementation();
670 case tok::objc_protocol:
671 return parseObjCProtocol();
674 case tok::objc_optional:
675 case tok::objc_required:
679 case tok::objc_autoreleasepool:
681 if (FormatTok->Tok.is(tok::l_brace)) {
700 if (FormatTok->is(tok::l_brace)) {
701 FormatTok->Type = TT_InlineASMBrace;
703 while (FormatTok && FormatTok->isNot(
tok::eof)) {
704 if (FormatTok->is(tok::r_brace)) {
705 FormatTok->Type = TT_InlineASMBrace;
710 FormatTok->Finalized =
true;
715 case tok::kw_namespace:
720 if (FormatTok->Tok.is(tok::kw_namespace)) {
726 case tok::kw_protected:
727 case tok::kw_private:
732 parseAccessSpecifier();
739 parseForOrWhileLoop();
747 case tok::kw_default:
760 if (FormatTok->Tok.is(tok::string_literal)) {
762 if (FormatTok->Tok.is(tok::l_brace)) {
763 parseBlock(
true,
false);
771 parseJavaScriptEs6ImportExport();
775 case tok::identifier:
776 if (FormatTok->is(TT_ForEachMacro)) {
777 parseForOrWhileLoop();
780 if (FormatTok->is(TT_MacroBlockBegin)) {
781 parseBlock(
false,
true,
787 parseJavaScriptEs6ImportExport();
792 if (FormatTok->is(tok::colon)) {
804 switch (FormatTok->Tok.getKind()) {
807 if (FormatTok->Tok.is(tok::l_brace))
821 case tok::kw_typedef:
844 FormatTok->is(tok::kw_class))
859 if (FormatTok->Tok.isAnyIdentifier() ||
860 FormatTok->isSimpleTypeSpecifier())
862 if (FormatTok->is(tok::l_paren))
864 if (FormatTok->is(tok::l_brace))
868 if (!tryToParseBracedList()) {
875 FormatTok->Type = TT_FunctionLBrace;
887 case tok::identifier: {
888 if (FormatTok->is(TT_MacroBlockEnd)) {
896 FormatTok->is(Keywords.
kw_function) && Line->Tokens.size() > 0) {
897 tryToParseJSFunction();
908 StringRef Text = FormatTok->TokenText;
910 if (Line->Tokens.size() == 1 &&
914 if (FormatTok->Tok.is(tok::colon) && !Line->MustBeDeclaration) {
920 bool FunctionLike = FormatTok->is(tok::l_paren);
924 bool FollowedByNewline =
925 CommentsBeforeNextToken.empty()
926 ? FormatTok->NewlinesBefore > 0
927 : CommentsBeforeNextToken.front()->NewlinesBefore > 0;
929 if (FollowedByNewline && (Text.size() >= 5 || FunctionLike) &&
941 if (FormatTok->is(TT_JsFatArrow)) {
943 if (FormatTok->is(tok::l_brace))
949 if (FormatTok->Tok.is(tok::l_brace)) {
966 bool UnwrappedLineParser::tryToParseLambda() {
973 if (!Line->Tokens.empty() &&
974 (Line->Tokens.back().Tok->isOneOf(tok::identifier, tok::kw_operator,
975 tok::kw_new, tok::kw_delete) ||
976 Line->Tokens.back().Tok->closesScope() ||
977 Line->Tokens.back().Tok->isSimpleTypeSpecifier())) {
981 assert(FormatTok->is(tok::l_square));
982 FormatToken &LSquare = *FormatTok;
983 if (!tryToParseLambdaIntroducer())
986 while (FormatTok->isNot(tok::l_brace)) {
987 if (FormatTok->isSimpleTypeSpecifier()) {
991 switch (FormatTok->Tok.getKind()) {
1003 case tok::identifier:
1004 case tok::coloncolon:
1005 case tok::kw_mutable:
1009 FormatTok->Type = TT_LambdaArrow;
1016 LSquare.Type = TT_LambdaLSquare;
1021 bool UnwrappedLineParser::tryToParseLambdaIntroducer() {
1023 if (FormatTok->is(tok::equal)) {
1025 if (FormatTok->is(tok::r_square)) {
1029 if (FormatTok->isNot(tok::comma))
1032 }
else if (FormatTok->is(tok::amp)) {
1034 if (FormatTok->is(tok::r_square)) {
1038 if (!FormatTok->isOneOf(tok::comma, tok::identifier)) {
1041 if (FormatTok->is(tok::comma))
1043 }
else if (FormatTok->is(tok::r_square)) {
1048 if (FormatTok->is(tok::amp))
1050 if (!FormatTok->isOneOf(tok::identifier, tok::kw_this))
1053 if (FormatTok->is(tok::ellipsis))
1055 if (FormatTok->is(tok::comma)) {
1057 }
else if (FormatTok->is(tok::r_square)) {
1067 void UnwrappedLineParser::tryToParseJSFunction() {
1071 if (FormatTok->is(tok::identifier))
1074 if (FormatTok->isNot(tok::l_paren))
1080 if (FormatTok->is(tok::colon)) {
1086 if (FormatTok->is(tok::l_brace))
1087 tryToParseBracedList();
1089 while (FormatTok->isNot(tok::l_brace) && !eof())
1096 bool UnwrappedLineParser::tryToParseBracedList() {
1098 calculateBraceTypes();
1100 if (FormatTok->BlockKind ==
BK_Block)
1106 bool UnwrappedLineParser::parseBracedList(
bool ContinueOnSemicolons) {
1107 bool HasError =
false;
1115 tryToParseJSFunction();
1118 if (FormatTok->is(TT_JsFatArrow)) {
1122 if (FormatTok->is(tok::l_brace)) {
1128 switch (FormatTok->Tok.getKind()) {
1131 if (FormatTok->is(tok::l_brace)) {
1149 if (FormatTok->is(tok::l_brace))
1159 if (!ContinueOnSemicolons)
1174 void UnwrappedLineParser::parseParens() {
1175 assert(FormatTok->Tok.is(tok::l_paren) &&
"'(' expected.");
1178 switch (FormatTok->Tok.getKind()) {
1194 if (!tryToParseBracedList())
1199 if (FormatTok->Tok.is(tok::l_brace))
1202 case tok::identifier:
1205 tryToParseJSFunction();
1216 void UnwrappedLineParser::parseSquare() {
1217 assert(FormatTok->Tok.is(tok::l_square) &&
"'[' expected.");
1218 if (tryToParseLambda())
1221 switch (FormatTok->Tok.getKind()) {
1234 case tok::l_brace: {
1235 if (!tryToParseBracedList())
1241 if (FormatTok->Tok.is(tok::l_brace))
1251 void UnwrappedLineParser::parseIfThenElse() {
1252 assert(FormatTok->Tok.is(tok::kw_if) &&
"'if' expected");
1254 if (FormatTok->Tok.is(tok::l_paren))
1256 bool NeedsUnwrappedLine =
false;
1257 if (FormatTok->Tok.is(tok::l_brace)) {
1264 NeedsUnwrappedLine =
true;
1269 parseStructuralElement();
1272 if (FormatTok->Tok.is(tok::kw_else)) {
1276 if (FormatTok->Tok.is(tok::l_brace)) {
1280 }
else if (FormatTok->Tok.is(tok::kw_if)) {
1285 parseStructuralElement();
1288 }
else if (NeedsUnwrappedLine) {
1293 void UnwrappedLineParser::parseTryCatch() {
1294 assert(FormatTok->isOneOf(tok::kw_try, tok::kw___try) &&
"'try' expected");
1296 bool NeedsUnwrappedLine =
false;
1297 if (FormatTok->is(tok::colon)) {
1300 while (FormatTok->is(tok::identifier)) {
1302 if (FormatTok->is(tok::l_paren))
1304 if (FormatTok->is(tok::comma))
1312 if (FormatTok->is(tok::l_brace)) {
1320 NeedsUnwrappedLine =
true;
1322 }
else if (!FormatTok->is(tok::kw_catch)) {
1328 parseStructuralElement();
1332 if (FormatTok->is(tok::at))
1334 if (!(FormatTok->isOneOf(tok::kw_catch, Keywords.
kw___except,
1335 tok::kw___finally) ||
1339 (FormatTok->Tok.isObjCAtKeyword(tok::objc_catch) ||
1340 FormatTok->Tok.isObjCAtKeyword(tok::objc_finally))))
1343 while (FormatTok->isNot(tok::l_brace)) {
1344 if (FormatTok->is(tok::l_paren)) {
1348 if (FormatTok->isOneOf(tok::semi, tok::r_brace,
tok::eof))
1352 NeedsUnwrappedLine =
false;
1360 NeedsUnwrappedLine =
true;
1363 if (NeedsUnwrappedLine) {
1368 void UnwrappedLineParser::parseNamespace() {
1369 assert(FormatTok->Tok.is(tok::kw_namespace) &&
"'namespace' expected");
1371 const FormatToken &InitialToken = *FormatTok;
1373 if (FormatTok->Tok.is(tok::identifier))
1375 if (FormatTok->Tok.is(tok::l_brace)) {
1381 DeclarationScopeStack.size() > 1);
1382 parseBlock(
true, AddLevel);
1385 if (FormatTok->Tok.is(tok::semi))
1392 void UnwrappedLineParser::parseNew() {
1393 assert(FormatTok->is(tok::kw_new) &&
"'new' expected");
1401 if (FormatTok->isOneOf(tok::semi, tok::l_brace, tok::r_brace))
1405 if (FormatTok->is(tok::l_paren)) {
1409 if (FormatTok->is(tok::l_brace))
1417 void UnwrappedLineParser::parseForOrWhileLoop() {
1418 assert(FormatTok->isOneOf(tok::kw_for, tok::kw_while, TT_ForEachMacro) &&
1419 "'for', 'while' or foreach macro expected");
1421 if (FormatTok->Tok.is(tok::l_paren))
1423 if (FormatTok->Tok.is(tok::l_brace)) {
1430 parseStructuralElement();
1435 void UnwrappedLineParser::parseDoWhile() {
1436 assert(FormatTok->Tok.is(tok::kw_do) &&
"'do' expected");
1438 if (FormatTok->Tok.is(tok::l_brace)) {
1446 parseStructuralElement();
1451 if (!FormatTok->Tok.is(tok::kw_while)) {
1457 parseStructuralElement();
1460 void UnwrappedLineParser::parseLabel() {
1462 unsigned OldLineLevel = Line->Level;
1463 if (Line->Level > 1 || (!Line->InPPDirective && Line->Level > 0))
1465 if (CommentsBeforeNextToken.empty() && FormatTok->Tok.is(tok::l_brace)) {
1468 if (FormatTok->Tok.is(tok::kw_break)) {
1474 parseStructuralElement();
1478 if (FormatTok->is(tok::semi))
1482 Line->Level = OldLineLevel;
1485 void UnwrappedLineParser::parseCaseLabel() {
1486 assert(FormatTok->Tok.is(tok::kw_case) &&
"'case' expected");
1490 }
while (!eof() && !FormatTok->Tok.is(tok::colon));
1494 void UnwrappedLineParser::parseSwitch() {
1495 assert(FormatTok->Tok.is(tok::kw_switch) &&
"'switch' expected");
1497 if (FormatTok->Tok.is(tok::l_paren))
1499 if (FormatTok->Tok.is(tok::l_brace)) {
1506 parseStructuralElement();
1511 void UnwrappedLineParser::parseAccessSpecifier() {
1517 if (FormatTok->Tok.is(tok::colon))
1522 void UnwrappedLineParser::parseEnum() {
1524 if (FormatTok->Tok.is(tok::kw_enum))
1528 if (FormatTok->Tok.is(tok::kw_class) || FormatTok->Tok.is(tok::kw_struct))
1531 while (FormatTok->Tok.getIdentifierInfo() ||
1532 FormatTok->isOneOf(tok::colon, tok::coloncolon, tok::less,
1533 tok::greater, tok::comma, tok::question)) {
1536 if (FormatTok->is(tok::l_paren))
1538 if (FormatTok->is(tok::identifier)) {
1543 FormatTok->is(tok::identifier))
1549 if (FormatTok->isNot(tok::l_brace))
1555 parseJavaEnumBody();
1560 bool HasError = !parseBracedList(
true);
1562 if (FormatTok->is(tok::semi))
1572 void UnwrappedLineParser::parseJavaEnumBody() {
1577 bool IsSimple =
true;
1580 if (Tok->is(tok::r_brace))
1582 if (Tok->isOneOf(tok::l_brace, tok::semi)) {
1606 if (FormatTok->is(tok::l_brace)) {
1608 parseBlock(
true,
true,
1610 }
else if (FormatTok->is(tok::l_paren)) {
1612 }
else if (FormatTok->is(tok::comma)) {
1615 }
else if (FormatTok->is(tok::semi)) {
1619 }
else if (FormatTok->is(tok::r_brace)) {
1634 void UnwrappedLineParser::parseRecord() {
1635 const FormatToken &InitialToken = *FormatTok;
1640 while (FormatTok->isOneOf(tok::identifier, tok::coloncolon, tok::hashhash,
1641 tok::kw___attribute, tok::kw___declspec,
1645 FormatTok->isOneOf(tok::period, tok::comma))) {
1646 bool IsNonMacroIdentifier =
1647 FormatTok->is(tok::identifier) &&
1648 FormatTok->TokenText != FormatTok->TokenText.upper();
1651 if (!IsNonMacroIdentifier && FormatTok->Tok.is(tok::l_paren))
1665 if (FormatTok->isOneOf(tok::colon, tok::less)) {
1667 if (FormatTok->is(tok::l_brace)) {
1668 calculateBraceTypes(
true);
1669 if (!tryToParseBracedList())
1672 if (FormatTok->Tok.is(tok::semi))
1677 if (FormatTok->Tok.is(tok::l_brace)) {
1681 parseBlock(
true,
true,
1689 void UnwrappedLineParser::parseObjCProtocolList() {
1690 assert(FormatTok->Tok.is(tok::less) &&
"'<' expected.");
1693 while (!eof() && FormatTok->Tok.isNot(tok::greater));
1697 void UnwrappedLineParser::parseObjCUntilAtEnd() {
1699 if (FormatTok->Tok.isObjCAtKeyword(tok::objc_end)) {
1704 if (FormatTok->is(tok::l_brace)) {
1708 }
else if (FormatTok->is(tok::r_brace)) {
1713 parseStructuralElement();
1718 void UnwrappedLineParser::parseObjCInterfaceOrImplementation() {
1723 if (FormatTok->Tok.is(tok::colon)) {
1726 }
else if (FormatTok->Tok.is(tok::l_paren))
1730 if (FormatTok->Tok.is(tok::less))
1731 parseObjCProtocolList();
1733 if (FormatTok->Tok.is(tok::l_brace)) {
1744 parseObjCUntilAtEnd();
1747 void UnwrappedLineParser::parseObjCProtocol() {
1751 if (FormatTok->Tok.is(tok::less))
1752 parseObjCProtocolList();
1755 if (FormatTok->Tok.is(tok::semi)) {
1757 return addUnwrappedLine();
1761 parseObjCUntilAtEnd();
1764 void UnwrappedLineParser::parseJavaScriptEs6ImportExport() {
1765 assert(FormatTok->isOneOf(Keywords.
kw_import, tok::kw_export));
1769 if (FormatTok->is(tok::kw_default))
1779 if (FormatTok->isOneOf(tok::kw_const, tok::kw_class, tok::kw_enum,
1783 if (FormatTok->is(tok::l_brace)) {
1788 while (!eof() && FormatTok->isNot(tok::semi) &&
1789 FormatTok->isNot(tok::l_brace)) {
1795 StringRef Prefix =
"") {
1796 llvm::dbgs() << Prefix <<
"Line(" << Line.
Level <<
")"
1798 for (std::list<UnwrappedLineNode>::const_iterator I = Line.
Tokens.begin(),
1801 llvm::dbgs() << I->Tok->Tok.getName() <<
"[" << I->Tok->Type <<
"] ";
1803 for (std::list<UnwrappedLineNode>::const_iterator I = Line.
Tokens.begin(),
1814 llvm::dbgs() <<
"\n";
1817 void UnwrappedLineParser::addUnwrappedLine() {
1818 if (Line->Tokens.empty())
1821 if (CurrentLines == &Lines)
1824 CurrentLines->push_back(std::move(*Line));
1825 Line->Tokens.clear();
1826 if (CurrentLines == &Lines && !PreprocessorDirectives.empty()) {
1827 CurrentLines->append(
1828 std::make_move_iterator(PreprocessorDirectives.begin()),
1829 std::make_move_iterator(PreprocessorDirectives.end()));
1830 PreprocessorDirectives.clear();
1834 bool UnwrappedLineParser::eof()
const {
return FormatTok->Tok.is(
tok::eof); }
1836 bool UnwrappedLineParser::isOnNewLine(
const FormatToken &FormatTok) {
1837 return (Line->InPPDirective || FormatTok.HasUnescapedNewline) &&
1838 FormatTok.NewlinesBefore > 0;
1841 void UnwrappedLineParser::flushComments(
bool NewlineBeforeNext) {
1842 bool JustComments = Line->Tokens.empty();
1843 for (SmallVectorImpl<FormatToken *>::const_iterator
1844 I = CommentsBeforeNextToken.begin(),
1845 E = CommentsBeforeNextToken.end();
1847 if (isOnNewLine(**I) && JustComments)
1851 if (NewlineBeforeNext && JustComments)
1853 CommentsBeforeNextToken.clear();
1856 void UnwrappedLineParser::nextToken() {
1859 flushComments(isOnNewLine(*FormatTok));
1860 pushToken(FormatTok);
1864 void UnwrappedLineParser::readToken() {
1865 bool CommentsInCurrentLine =
true;
1869 while (!Line->InPPDirective && FormatTok->Tok.is(tok::hash) &&
1870 (FormatTok->HasUnescapedNewline || FormatTok->IsFirst)) {
1873 bool SwitchToPreprocessorLines = !Line->Tokens.empty();
1878 flushComments(isOnNewLine(*FormatTok));
1881 while (FormatTok->Type == TT_ConflictStart ||
1882 FormatTok->Type == TT_ConflictEnd ||
1883 FormatTok->Type == TT_ConflictAlternative) {
1884 if (FormatTok->Type == TT_ConflictStart) {
1885 conditionalCompilationStart(
false);
1886 }
else if (FormatTok->Type == TT_ConflictAlternative) {
1887 conditionalCompilationAlternative();
1888 }
else if (FormatTok->Type == TT_ConflictEnd) {
1889 conditionalCompilationEnd();
1895 if (!PPStack.empty() && (PPStack.back() == PP_Unreachable) &&
1896 !Line->InPPDirective) {
1900 if (!FormatTok->Tok.is(tok::comment))
1902 if (isOnNewLine(*FormatTok) || FormatTok->IsFirst) {
1903 CommentsInCurrentLine =
false;
1905 if (CommentsInCurrentLine) {
1906 pushToken(FormatTok);
1908 CommentsBeforeNextToken.push_back(FormatTok);
1913 void UnwrappedLineParser::pushToken(FormatToken *Tok) {
1914 Line->Tokens.push_back(UnwrappedLineNode(Tok));
1915 if (MustBreakBeforeNextToken) {
1916 Line->Tokens.back().Tok->MustBreakBefore =
true;
1917 MustBreakBeforeNextToken =
false;
FormatToken *& ResetToken
std::vector< bool > & Stack
FormatTokenSource *& TokenSource
MatchFinder::MatchCallback * Callback
bool isNot(tok::TokenKind K) const
ArrayRef< FormatToken * > Tokens
This file contains the declaration of the UnwrappedLineParser, which turns a stream of tokens into Un...
TokenKind
Provides a simple uniform namespace for tokens from all C languages.
ast_type_traits::DynTypedNode Node
unsigned kind
All of the diagnostics that can be emitted by the frontend.
unsigned PreviousLineLevel
FormatTokenSource * PreviousTokenSource