Use PreprocessorLexer::getFileID() instead of Lexer::getFileLoc().  This is an intermediate step to having getCurrentLexer() return a PreprocessorLexer* instead of a Lexer*.


git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@59672 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Lex/Pragma.cpp b/lib/Lex/Pragma.cpp
index 0e0841d..5e6de0f 100644
--- a/lib/Lex/Pragma.cpp
+++ b/lib/Lex/Pragma.cpp
@@ -186,10 +186,10 @@
   }
   
   // Get the current file lexer we're looking at.  Ignore _Pragma 'files' etc.
-  SourceLocation FileLoc = getCurrentFileLexer()->getFileLoc();
+  unsigned FileID = getCurrentFileLexer()->getFileID();
   
   // Mark the file as a once-only file now.
-  HeaderInfo.MarkFileIncludeOnce(SourceMgr.getFileEntryForLoc(FileLoc));
+  HeaderInfo.MarkFileIncludeOnce(SourceMgr.getFileEntryForID(FileID));
 }
 
 void Preprocessor::HandlePragmaMark() {
@@ -251,7 +251,7 @@
   Lexer *TheLexer = getCurrentFileLexer();
   
   // Mark the file as a system header.
-  const FileEntry *File = SourceMgr.getFileEntryForLoc(TheLexer->getFileLoc());
+  const FileEntry *File = SourceMgr.getFileEntryForID(TheLexer->getFileID());
   HeaderInfo.MarkFileSystemHeader(File);
   
   // Notify the client, if desired, that we are in a new source file.
@@ -294,8 +294,8 @@
     return;
   }
   
-  SourceLocation FileLoc = getCurrentFileLexer()->getFileLoc();
-  const FileEntry *CurFile = SourceMgr.getFileEntryForLoc(FileLoc);
+  unsigned FileID = getCurrentFileLexer()->getFileID();
+  const FileEntry *CurFile = SourceMgr.getFileEntryForID(FileID);
 
   // If this file is older than the file it depends on, emit a diagnostic.
   if (CurFile && CurFile->getModificationTime() < File->getModificationTime()) {