use simpler method

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@62974 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/Driver/RewriteBlocks.cpp b/Driver/RewriteBlocks.cpp
index 706eef8918200242dd51ad7124db147965bedd91..75aa3c863a1990a8eb694c3346a8847255edcf04 100644 (file)
--- a/Driver/RewriteBlocks.cpp
+++ b/Driver/RewriteBlocks.cpp
@@ -354,7 +354,7 @@ void RewriteBlocks::HandleTopLevelDecl(Decl *D) {
     RewriteProtocolDecl(PD);
 
   // If we have a decl in the main file, see if we should rewrite it.
-  if (SM->getDecomposedFileLoc(Loc).first == MainFileID)
+  if (SM->isFromMainFile(Loc))
     HandleDeclInMainFile(D);
   return;
 }