diff --git a/clang/include/clang/Tooling/Syntax/Tree.h b/clang/include/clang/Tooling/Syntax/Tree.h --- a/clang/include/clang/Tooling/Syntax/Tree.h +++ b/clang/include/clang/Tooling/Syntax/Tree.h @@ -214,6 +214,16 @@ using Tree::Tree; static bool classof(const Node *N); + + ElementAndDelimiter getWithDelimiter(Node *Element); + llvm::Optional> + getElementAndDelimiterAfterDelimiter(Leaf *Delimiter); + llvm::Optional> + getElementAndDelimiterAfterElement(Node *Element); + llvm::Optional> + getNext(ElementAndDelimiter &B); + llvm::Optional> getFirst(); + /// Returns the elements and corresponding delimiters. Missing elements /// and delimiters are represented as null pointers. /// diff --git a/clang/lib/Tooling/Syntax/Nodes.cpp b/clang/lib/Tooling/Syntax/Nodes.cpp --- a/clang/lib/Tooling/Syntax/Nodes.cpp +++ b/clang/lib/Tooling/Syntax/Nodes.cpp @@ -226,68 +226,58 @@ // vector std::vector syntax::NestedNameSpecifier::getSpecifiers() { - auto SpecifiersAsNodes = getElementsAsNodes(); - std::vector Children; - for (const auto &Element : SpecifiersAsNodes) { - Children.push_back(llvm::cast(Element)); - } + auto Children = std::vector(); + for (auto C = getFirst(); C; C = getNext(*C)) + Children.push_back(cast(C->element)); + return Children; } std::vector> syntax::NestedNameSpecifier::getSpecifiersAndDoubleColons() { - auto SpecifiersAsNodesAndDoubleColons = getElementsAsNodesAndDelimiters(); - std::vector> - Children; - for (const auto &SpecifierAndDoubleColon : SpecifiersAsNodesAndDoubleColons) { - Children.push_back( - {llvm::cast(SpecifierAndDoubleColon.element), - SpecifierAndDoubleColon.delimiter}); - } + auto Children = + std::vector>(); + for (auto C = getFirst(); C; C = getNext(*C)) + Children.push_back({cast(C->element), C->delimiter}); + return Children; } std::vector syntax::CallArguments::getArguments() { - auto ArgumentsAsNodes = getElementsAsNodes(); - std::vector Children; - for (const auto &ArgumentAsNode : ArgumentsAsNodes) { - Children.push_back(llvm::cast(ArgumentAsNode)); - } + auto Children = std::vector(); + for (auto C = getFirst(); C; C = getNext(*C)) + Children.push_back(cast(C->element)); + return Children; } std::vector> syntax::CallArguments::getArgumentsAndCommas() { - auto ArgumentsAsNodesAndCommas = getElementsAsNodesAndDelimiters(); - std::vector> Children; - for (const auto &ArgumentAsNodeAndComma : ArgumentsAsNodesAndCommas) { - Children.push_back( - {llvm::cast(ArgumentAsNodeAndComma.element), - ArgumentAsNodeAndComma.delimiter}); - } + auto Children = + std::vector>(); + for (auto C = getFirst(); C; C = getNext(*C)) + Children.push_back({cast(C->element), C->delimiter}); + return Children; } std::vector syntax::ParameterDeclarationList::getParameterDeclarations() { - auto ParametersAsNodes = getElementsAsNodes(); - std::vector Children; - for (const auto &ParameterAsNode : ParametersAsNodes) { - Children.push_back(llvm::cast(ParameterAsNode)); - } + auto Children = std::vector(); + for (auto C = getFirst(); C; C = getNext(*C)) + Children.push_back(cast(C->element)); + return Children; } std::vector> syntax::ParameterDeclarationList::getParametersAndCommas() { - auto ParametersAsNodesAndCommas = getElementsAsNodesAndDelimiters(); - std::vector> - Children; - for (const auto &ParameterAsNodeAndComma : ParametersAsNodesAndCommas) { + auto Children = std::vector< + syntax::List::ElementAndDelimiter>(); + for (auto C = getFirst(); C; C = getNext(*C)) Children.push_back( - {llvm::cast(ParameterAsNodeAndComma.element), - ParameterAsNodeAndComma.delimiter}); - } + {cast(C->element), C->delimiter}); + return Children; } diff --git a/clang/lib/Tooling/Syntax/Tree.cpp b/clang/lib/Tooling/Syntax/Tree.cpp --- a/clang/lib/Tooling/Syntax/Tree.cpp +++ b/clang/lib/Tooling/Syntax/Tree.cpp @@ -9,6 +9,7 @@ #include "clang/Basic/TokenKinds.h" #include "clang/Tooling/Syntax/Nodes.h" #include "llvm/ADT/ArrayRef.h" +#include "llvm/ADT/Optional.h" #include "llvm/ADT/STLExtras.h" #include "llvm/Support/Casting.h" #include @@ -294,46 +295,107 @@ } } -std::vector> -syntax::List::getElementsAsNodesAndDelimiters() { - if (!getFirstChild()) - return {}; +syntax::List::ElementAndDelimiter +syntax::List::getWithDelimiter(syntax::Node *Element) { + assert(Element && Element->getRole() == syntax::NodeRole::ListElement); + auto *Next = Element->getNextSibling(); + if (!Next) + return {Element, nullptr}; + switch (Next->getRole()) { + case syntax::NodeRole::ListElement: + return {Element, nullptr}; + case syntax::NodeRole::ListDelimiter: + return {Element, cast(Next)}; + default: + llvm_unreachable( + "A list can have only elements and delimiters as children."); + } +} - std::vector> Children; - syntax::Node *ElementWithoutDelimiter = nullptr; - for (auto *C = getFirstChild(); C; C = C->getNextSibling()) { - switch (C->getRole()) { - case syntax::NodeRole::ListElement: { - if (ElementWithoutDelimiter) { - Children.push_back({ElementWithoutDelimiter, nullptr}); - } - ElementWithoutDelimiter = C; - break; - } - case syntax::NodeRole::ListDelimiter: { - Children.push_back({ElementWithoutDelimiter, cast(C)}); - ElementWithoutDelimiter = nullptr; - break; - } - default: - llvm_unreachable( - "A list can have only elements and delimiters as children."); +llvm::Optional> +syntax::List::getElementAndDelimiterAfterDelimiter(syntax::Leaf *Delimiter) { + assert(Delimiter && Delimiter->getRole() == syntax::NodeRole::ListDelimiter); + auto *Next = Delimiter->getNextSibling(); + if (!Next) { + switch (getTerminationKind()) { + // List is separated and ends with delimiter + // => missing last ElementAndDelimiter. + case syntax::List::TerminationKind::Separated: + return llvm::Optional>( + {nullptr, nullptr}); + case syntax::List::TerminationKind::Terminated: + case syntax::List::TerminationKind::MaybeTerminated: + return None; } } - - switch (getTerminationKind()) { - case syntax::List::TerminationKind::Separated: { - Children.push_back({ElementWithoutDelimiter, nullptr}); - break; + switch (Next->getRole()) { + case syntax::NodeRole::ListElement: + return getWithDelimiter(Next); + + // Consecutive Delimiters => missing Element + case syntax::NodeRole::ListDelimiter: + return llvm::Optional>( + {nullptr, cast(Next)}); + default: + llvm_unreachable( + "A list can have only elements and delimiters as children."); } - case syntax::List::TerminationKind::Terminated: - case syntax::List::TerminationKind::MaybeTerminated: { - if (ElementWithoutDelimiter) { - Children.push_back({ElementWithoutDelimiter, nullptr}); - } - break; +} + +llvm::Optional> +syntax::List::getElementAndDelimiterAfterElement(syntax::Node *Element) { + assert(Element && Element->getRole() == syntax::NodeRole::ListElement); + auto *Next = Element->getNextSibling(); + // a b, x => End of list, this was the last ElementAndDelimiter. + if (!Next) + return None; + + switch (Next->getRole()) { + // x b, c => next ElementAndDelimiter starts with 'b'. + case syntax::NodeRole::ListElement: + return getWithDelimiter(Next); + + // a x, c => next ElementAndDelimiter starts after ','. + case syntax::NodeRole::ListDelimiter: + return getElementAndDelimiterAfterDelimiter(cast(Next)); + + default: + llvm_unreachable( + "A list can have only elements and delimiters as children."); } +} + +llvm::Optional> +syntax::List::getNext(syntax::List::ElementAndDelimiter &ED) { + if (auto *Delimiter = ED.delimiter) + return getElementAndDelimiterAfterDelimiter(Delimiter); + if (auto *Element = ED.element) + return getElementAndDelimiterAfterElement(Element); + return None; +} + +llvm::Optional> +syntax::List::getFirst() { + auto *First = this->getFirstChild(); + if (!First) + return None; + switch (First->getRole()) { + case syntax::NodeRole::ListElement: + return getWithDelimiter(First); + case syntax::NodeRole::ListDelimiter: + return llvm::Optional>( + {nullptr, cast(First)}); + default: + llvm_unreachable( + "A list can have only elements and delimiters as children."); } +} + +std::vector> +syntax::List::getElementsAsNodesAndDelimiters() { + auto Children = std::vector>(); + for (auto C = getFirst(); C; C = getNext(*C)) + Children.push_back(*C); return Children; } @@ -341,43 +403,9 @@ // Almost the same implementation of `getElementsAsNodesAndDelimiters` but // ignoring delimiters std::vector syntax::List::getElementsAsNodes() { - if (!getFirstChild()) - return {}; - - std::vector Children; - syntax::Node *ElementWithoutDelimiter = nullptr; - for (auto *C = getFirstChild(); C; C = C->getNextSibling()) { - switch (C->getRole()) { - case syntax::NodeRole::ListElement: { - if (ElementWithoutDelimiter) { - Children.push_back(ElementWithoutDelimiter); - } - ElementWithoutDelimiter = C; - break; - } - case syntax::NodeRole::ListDelimiter: { - Children.push_back(ElementWithoutDelimiter); - ElementWithoutDelimiter = nullptr; - break; - } - default: - llvm_unreachable("A list has only elements or delimiters."); - } - } - - switch (getTerminationKind()) { - case syntax::List::TerminationKind::Separated: { - Children.push_back(ElementWithoutDelimiter); - break; - } - case syntax::List::TerminationKind::Terminated: - case syntax::List::TerminationKind::MaybeTerminated: { - if (ElementWithoutDelimiter) { - Children.push_back(ElementWithoutDelimiter); - } - break; - } - } + auto Children = std::vector(); + for (auto C = getFirst(); C; C = getNext(*C)) + Children.push_back(C->element); return Children; }