1 //===-- FileRemapper.h - File Remapping Helper ------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #ifndef LLVM_CLANG_ARCMIGRATE_FILEREMAPPER_H
11 #define LLVM_CLANG_ARCMIGRATE_FILEREMAPPER_H
13 #include "clang/Basic/LLVM.h"
14 #include "llvm/ADT/DenseMap.h"
15 #include "llvm/ADT/PointerUnion.h"
16 #include "llvm/ADT/StringRef.h"
26 class DiagnosticsEngine;
27 class PreprocessorOptions;
32 // FIXME: Reuse the same FileManager for multiple ASTContexts.
33 std::unique_ptr<FileManager> FileMgr;
35 typedef llvm::PointerUnion<const FileEntry *, llvm::MemoryBuffer *> Target;
36 typedef llvm::DenseMap<const FileEntry *, Target> MappingsTy;
37 MappingsTy FromToMappings;
39 llvm::DenseMap<const FileEntry *, const FileEntry *> ToFromMappings;
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);
52 bool overwriteOriginal(DiagnosticsEngine &Diag,
53 StringRef outputDir = StringRef());
55 void remap(StringRef filePath, std::unique_ptr<llvm::MemoryBuffer> memBuf);
57 void applyMappings(PreprocessorOptions &PPOpts) const;
59 void clear(StringRef outputDir = StringRef());
62 void remap(const FileEntry *file, std::unique_ptr<llvm::MemoryBuffer> memBuf);
63 void remap(const FileEntry *file, const FileEntry *newfile);
65 const FileEntry *getOriginalFile(StringRef filePath);
66 void resetTarget(Target &targ);
68 bool report(const Twine &err, DiagnosticsEngine &Diag);
70 std::string getRemapInfoFile(StringRef outputDir);
73 } // end namespace arcmt
75 } // end namespace clang