Add a helper that either opens a file or stdin.

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

diff --git a/include/llvm/Support/MemoryBuffer.h b/include/llvm/Support/MemoryBuffer.h
index e4635d0e7ab28e2b8d973be7d13ac989f0f835d5..80672165f7f9149f42093728a21827d386cc708c 100644 (file)
--- a/include/llvm/Support/MemoryBuffer.h
+++ b/include/llvm/Support/MemoryBuffer.h
@@ -76,6 +76,17 @@ public:
   /// getSTDIN - Read all of stdin into a file buffer, and return it.  This
   /// fails if stdin is empty.
   static MemoryBuffer *getSTDIN();
+  
+  
+  /// getFileOrSTDIN - Open the specified file as a MemoryBuffer, or open stdin
+  /// if the Filename is "-".
+  static MemoryBuffer *getFileOrSTDIN(const char *FilenameStart,unsigned FnSize,
+                                      int64_t FileSize = -1) {
+    if (FnSize == 1 && FilenameStart[0] == '-')
+      return getSTDIN();
+    return getFile(FilenameStart, FnSize, FileSize);
+  }
+  
 };
 
 } // end namespace llvm