1 files changed, 76 insertions, 0 deletions
diff --git a/include/clang/ARCMigrate/FileRemapper.h b/include/clang/ARCMigrate/FileRemapper.h
new file mode 100644
index 000000000000..809f6a5f71be
--- /dev/null
+++ b/include/clang/ARCMigrate/FileRemapper.h
@@ -0,0 +1,76 @@
+//===-- FileRemapper.h - File Remapping Helper ------------------*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_CLANG_ARCMIGRATE_FILEREMAPPER_H
+#define LLVM_CLANG_ARCMIGRATE_FILEREMAPPER_H
+
+#include "llvm/ADT/OwningPtr.h"
+#include "llvm/ADT/PointerUnion.h"
+#include "llvm/ADT/DenseMap.h"
+#include "llvm/ADT/StringRef.h"
+
+namespace llvm {
+  class MemoryBuffer;
+}
+
+namespace clang {
+  class FileManager;
+  class FileEntry;
+  class Diagnostic;
+  class CompilerInvocation;
+
+namespace arcmt {
+
+class FileRemapper {
+  // FIXME: Reuse the same FileManager for multiple ASTContexts.
+  llvm::OwningPtr<FileManager> FileMgr;
+
+  typedef llvm::PointerUnion<const FileEntry *, llvm::MemoryBuffer *> Target;
+  typedef llvm::DenseMap<const FileEntry *, Target> MappingsTy;
+  MappingsTy FromToMappings;
+
+  llvm::DenseMap<const FileEntry *, const FileEntry *> ToFromMappings;
+
+public:
+  FileRemapper();
+  ~FileRemapper();
+  
+  bool initFromDisk(llvm::StringRef outputDir, Diagnostic &Diag,
+                    bool ignoreIfFilesChanged);
+  bool flushToDisk(llvm::StringRef outputDir, Diagnostic &Diag);
+
+  bool overwriteOriginal(Diagnostic &Diag,
+                         llvm::StringRef outputDir = llvm::StringRef());
+
+  void remap(llvm::StringRef filePath, llvm::MemoryBuffer *memBuf);
+  void remap(llvm::StringRef filePath, llvm::StringRef newPath);
+
+  void applyMappings(CompilerInvocation &CI) const;
+
+  void transferMappingsAndClear(CompilerInvocation &CI);
+
+  void clear(llvm::StringRef outputDir = llvm::StringRef());
+
+private:
+  void remap(const FileEntry *file, llvm::MemoryBuffer *memBuf);
+  void remap(const FileEntry *file, const FileEntry *newfile);
+
+  const FileEntry *getOriginalFile(llvm::StringRef filePath);
+  void resetTarget(Target &targ);
+
+  bool report(const std::string &err, Diagnostic &Diag);
+
+  std::string getRemapInfoFile(llvm::StringRef outputDir);
+};
+
+} // end namespace arcmt
+
+}  // end namespace clang
+
+#endif