averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

libSyntax: add getAbsoluteEndPosition() method to syntax nodes.

Browse Source 
This implementation uses sibling's absolute start position to help
populate caches while getting the end position.
This commit is contained in:
Xi Ge
2018-05-01 12:04:26 -07:00
parent 574e5564f5
commit 5a8053e7ef
5 changed files with 73 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
lib/Syntax/SyntaxData.cpp
View File

@@ -66,6 +66,18 @@ RC<SyntaxData> SyntaxData::getPreviousNode() const {
return hasParent() ? Parent->getPreviousNode() : nullptr;
}
RC<SyntaxData> SyntaxData::getNextNode() const {
if (hasParent()) {
for (size_t I = getIndexInParent() + 1, N = Parent->getNumChildren();
I != N; I++) {
if (auto C = getParent()->getChild(I))
return C;
}
return Parent->getNextNode();
}
return nullptr;
}
AbsolutePosition SyntaxData::getAbsolutePositionWithLeadingTrivia() const {
if (PositionCache.hasValue())
return *PositionCache;
@@ -85,3 +97,13 @@ AbsolutePosition SyntaxData::getAbsolutePosition() const {
getRaw()->accumulateLeadingTrivia(Result);
return Result;
}
AbsolutePosition SyntaxData::getAbsoluteEndPosition() const {
if (auto N = getNextNode()) {
return N->getAbsolutePositionWithLeadingTrivia();
} else {
auto Result = getAbsolutePositionWithLeadingTrivia();
getRaw()->accumulateAbsolutePosition(Result);
return Result;
}
}