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

ModulePrinter: Print decls from the same source file near each other.

Browse Source
This commit is contained in:
Xi Ge
2016-03-07 18:31:36 -08:00
parent 739c409140
commit a5c9072344
9 changed files with 79 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
lib/Serialization/ModuleFile.cpp
View File

@@ -1539,13 +1539,9 @@ Optional<CommentInfo> ModuleFile::getCommentForDecl(const Decl *D) const {
return getCommentForDeclByUSR(USRBuffer.str());
}
Optional<StringRef> ModuleFile::getGroupNameById(unsigned Id) const {
if(!GroupNamesMap || GroupNamesMap->count(Id) == 0)
return None;
return (*GroupNamesMap)[Id];
}
const static std::string Separator = "/";
Optional<StringRef> ModuleFile::getGroupNameForDecl(const Decl *D) const {
const Decl* getGroupDecl(const Decl *D) {
auto GroupD = D;
// Extensions always exist in the same group with the nominal.
@@ -1553,6 +1549,28 @@ Optional<StringRef> ModuleFile::getGroupNameForDecl(const Decl *D) const {
getInnermostTypeContext())) {
GroupD = ED->getExtendedType()->getAnyNominal();
}
return GroupD;
}
Optional<StringRef> ModuleFile::getGroupNameById(unsigned Id) const {
if(!GroupNamesMap || GroupNamesMap->count(Id) == 0)
return None;
auto Original = (*GroupNamesMap)[Id];
return StringRef(Original.data(), Original.find_last_of(Separator));
}
Optional<StringRef> ModuleFile::getSourceFileNameById(unsigned Id) const {
if(!GroupNamesMap || GroupNamesMap->count(Id) == 0)
return None;
auto Original = (*GroupNamesMap)[Id];
auto SepPos = Original.find_last_of(Separator);
auto Start = Original.data() + SepPos + 1;
auto Len = Original.size() - SepPos - 1;
return StringRef(Start, Len);
}
Optional<StringRef> ModuleFile::getGroupNameForDecl(const Decl *D) const {
auto GroupD = getGroupDecl(D);
auto Triple = getCommentForDecl(GroupD);
if (!Triple.hasValue()) {
return None;
@@ -1560,6 +1578,16 @@ Optional<StringRef> ModuleFile::getGroupNameForDecl(const Decl *D) const {
return getGroupNameById(Triple.getValue().Group);
}
Optional<StringRef>
ModuleFile::getSourceFileNameForDecl(const Decl *D) const {
auto GroupD = getGroupDecl(D);
auto Triple = getCommentForDecl(GroupD);
if (!Triple.hasValue()) {
return None;
}
return getSourceFileNameById(Triple.getValue().Group);
}
void ModuleFile::collectAllGroups(std::vector<StringRef> &Names) const {
if (!GroupNamesMap)
return;