Home | History | Annotate | Download | only in ARCMigrate 
      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
     81