contrib/llvm/tools/clang/include/clang/ARCMigrate/FileRemapper.h

   1 //===-- FileRemapper.h - File Remapping Helper ------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #ifndef LLVM_CLANG_ARCMIGRATE_FILEREMAPPER_H
  11 #define LLVM_CLANG_ARCMIGRATE_FILEREMAPPER_H
  12
  13 #include "clang/Basic/LLVM.h"
  14 #include "llvm/ADT/DenseMap.h"
  15 #include "llvm/ADT/OwningPtr.h"
  16 #include "llvm/ADT/PointerUnion.h"
  17 #include "llvm/ADT/StringRef.h"
  18
  19 namespace llvm {
  20   class MemoryBuffer;
  21 }
  22
  23 namespace clang {
  24   class FileManager;
  25   class FileEntry;
  26   class DiagnosticsEngine;
  27   class PreprocessorOptions;
  28
  29 namespace arcmt {
  30
  31 class FileRemapper {
  32   // FIXME: Reuse the same FileManager for multiple ASTContexts.
  33   OwningPtr<FileManager> FileMgr;
  34
  35   typedef llvm::PointerUnion<const FileEntry *, llvm::MemoryBuffer *> Target;
  36   typedef llvm::DenseMap<const FileEntry *, Target> MappingsTy;
  37   MappingsTy FromToMappings;
  38
  39   llvm::DenseMap<const FileEntry *, const FileEntry *> ToFromMappings;
  40
  41 public:
  42   FileRemapper();
  43   ~FileRemapper();
  44
  45   bool initFromDisk(StringRef outputDir, DiagnosticsEngine &Diag,
  46                     bool ignoreIfFilesChanged);
  47   bool initFromFile(StringRef filePath, DiagnosticsEngine &Diag,
  48                     bool ignoreIfFilesChanged);
  49   bool flushToDisk(StringRef outputDir, DiagnosticsEngine &Diag);
  50   bool flushToFile(StringRef outputPath, DiagnosticsEngine &Diag);
  51
  52   bool overwriteOriginal(DiagnosticsEngine &Diag,
  53                          StringRef outputDir = StringRef());
  54
  55   void remap(StringRef filePath, llvm::MemoryBuffer *memBuf);
  56   void remap(StringRef filePath, StringRef newPath);
  57
  58   void applyMappings(PreprocessorOptions &PPOpts) const;
  59
  60   void transferMappingsAndClear(PreprocessorOptions &PPOpts);
  61
  62   void clear(StringRef outputDir = StringRef());
  63
  64 private:
  65   void remap(const FileEntry *file, llvm::MemoryBuffer *memBuf);
  66   void remap(const FileEntry *file, const FileEntry *newfile);
  67
  68   const FileEntry *getOriginalFile(StringRef filePath);
  69   void resetTarget(Target &targ);
  70
  71   bool report(const Twine &err, DiagnosticsEngine &Diag);
  72
  73   std::string getRemapInfoFile(StringRef outputDir);
  74 };
  75
  76 } // end namespace arcmt
  77
  78 }  // end namespace clang
  79
  80 #endif