commit 9a13bde64336bb46f01916753d9811d157048528
author Chris Lattner <sabre@nondot.org> Wed Jun 21 04:57:09 2006 +0000
committer Chris Lattner <sabre@nondot.org> Wed Jun 21 04:57:09 2006 +0000
tree a37ce66a3c7ae9cc4f6b16b9cd01b492d69f5c4e
parent d12ad77e40134bd763f882b022c9203ceec4d76a

Implement a new SourceManager::getSourceName method

llvm-svn: 38560

clang/Basic/SourceManager.cpp

diff --git a/clang/Basic/SourceManager.cpp b/clang/Basic/SourceManager.cpp
index 07565a3..7a6542e 100644
--- a/clang/Basic/SourceManager.cpp
+++ b/clang/Basic/SourceManager.cpp
@@ -133,18 +133,18 @@
 /// getColumnNumber - Return the column # for the specified include position.
 /// this is significantly cheaper to compute than the line number.  This returns
 /// zero if the column number isn't known.
-unsigned SourceManager::getColumnNumber(SourceLocation IncludePos) const {
-  unsigned FileID = IncludePos.getFileID();
+unsigned SourceManager::getColumnNumber(SourceLocation Loc) const {
+  unsigned FileID = Loc.getFileID();
   if (FileID == 0) return 0;
   
   // If this is a macro, we need to get the instantiation location.
   const SrcMgr::FileIDInfo *FIDInfo = getFIDInfo(FileID);
   if (FIDInfo->IDType == SrcMgr::FileIDInfo::MacroExpansion) {
-    IncludePos = FIDInfo->IncludeLoc;
-    FileID = IncludePos.getFileID();
+    Loc = FIDInfo->IncludeLoc;
+    FileID = Loc.getFileID();
   }
   
-  unsigned FilePos = getFilePos(IncludePos);
+  unsigned FilePos = getFilePos(Loc);
   const SourceBuffer *Buffer = getBuffer(FileID);
   const char *Buf = Buffer->getBufferStart();
 
@@ -154,17 +154,35 @@
   return FilePos-LineStart+1;
 }
 
+/// getSourceName - This method returns the name of the file or buffer that
+/// the SourceLocation specifies.  This can be modified with #line directives,
+/// etc.
+std::string SourceManager::getSourceName(SourceLocation Loc) {
+  unsigned FileID = Loc.getFileID();
+  if (FileID == 0) return "";
+  
+  // If this is a macro, we need to get the instantiation location.
+  const SrcMgr::FileIDInfo *FIDInfo = getFIDInfo(FileID);
+  if (FIDInfo->IDType == SrcMgr::FileIDInfo::MacroExpansion) {
+    Loc = FIDInfo->IncludeLoc;
+    FIDInfo = getFIDInfo(Loc.getFileID());
+  }
+  
+  return getFileInfo(FIDInfo)->Buffer->getBufferIdentifier();
+}
+
+
 /// getLineNumber - Given a SourceLocation, return the physical line number
 /// for the position indicated.  This requires building and caching a table of
 /// line offsets for the SourceBuffer, so this is not cheap: use only when
 /// about to emit a diagnostic.
-unsigned SourceManager::getLineNumber(SourceLocation IncludePos) {
-  unsigned FileID = IncludePos.getFileID();
+unsigned SourceManager::getLineNumber(SourceLocation Loc) {
+  unsigned FileID = Loc.getFileID();
   // If this is a macro, we need to get the instantiation location.
   const SrcMgr::FileIDInfo *FIDInfo = getFIDInfo(FileID);
   if (FIDInfo->IDType == SrcMgr::FileIDInfo::MacroExpansion) {
-    IncludePos = FIDInfo->IncludeLoc;
-    FileID = IncludePos.getFileID();
+    Loc = FIDInfo->IncludeLoc;
+    FileID = Loc.getFileID();
     FIDInfo = getFIDInfo(FileID);
   }
 
@@ -226,7 +244,7 @@
   // type approaches to make good (tight?) initial guesses based on the
   // assumption that all lines are the same average size.
   unsigned *Pos = std::lower_bound(SourceLineCache, SourceLineCache+NumLines,
-                                   getFilePos(IncludePos)+1);
+                                   getFilePos(Loc)+1);
   return Pos-SourceLineCache;
 }