Added "getCanonicalID()", "isFromSameFile", and "isFromMainFile" to compare
the files of different SourceLocations. These methods correctly handle the
case where a file may have multiple FileIDs due to it being large enough
to be spread across several chunks.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@49682 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/include/clang/Basic/SourceManager.h b/include/clang/Basic/SourceManager.h
index 59f1690..700dcd3 100644
--- a/include/clang/Basic/SourceManager.h
+++ b/include/clang/Basic/SourceManager.h
@@ -371,7 +371,15 @@
const FileEntry* getFileEntryForID(unsigned id) const {
return getContentCache(id)->Entry;
}
-
+
+ /// getCanonicalFileID - Return the canonical FileID for a SourceLocation.
+ /// A file can have multiple FileIDs if it is large enough to be broken
+ /// into multiple chunks. This method returns the unique FileID without
+ /// chunk information for a given SourceLocation. Use this method when
+ /// you want to compare FileIDs across SourceLocations.
+ unsigned getCanonicalFileID(SourceLocation PhysLoc) const {
+ return getDecomposedFileLoc(PhysLoc).first;
+ }
/// getDecomposedFileLoc - Decompose the specified file location into a raw
/// FileID + Offset pair. The first element is the FileID, the second is the
@@ -399,6 +407,18 @@
return getDecomposedFileLoc(PhysLoc).second;
}
+ /// isFromSameFile - Returns true if both SourceLocations correspond to
+ /// the same file.
+ bool isFromSameFile(SourceLocation Loc1, SourceLocation Loc2) const {
+ return getCanonicalFileID(Loc1) == getCanonicalFileID(Loc2);
+ }
+
+ /// isFromMainFile - Returns true if the file of provided SourceLocation is
+ /// the main file.
+ bool isFromMainFile(SourceLocation Loc) const {
+ return getCanonicalFileID(Loc) == getMainFileID();
+ }
+
/// PrintStats - Print statistics to stderr.
///
void PrintStats() const;