1 //===----- EditedSource.h - Collection of source edits ----------*- 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_EDIT_EDITEDSOURCE_H 11 #define LLVM_CLANG_EDIT_EDITEDSOURCE_H 12 13 #include "clang/Edit/FileOffset.h" 14 #include "llvm/Support/Allocator.h" 15 #include "llvm/ADT/DenseMap.h" 16 #include "llvm/ADT/StringRef.h" 17 #include <map> 18 19 namespace clang { 20 class LangOptions; 21 class PreprocessingRecord; 22 23 namespace edit { 24 class Commit; 25 class EditsReceiver; 26 27 class EditedSource { 28 const SourceManager &SourceMgr; 29 const LangOptions &LangOpts; 30 const PreprocessingRecord *PPRec; 31 32 struct FileEdit { 33 StringRef Text; 34 unsigned RemoveLen; 35 FileEditFileEdit36 FileEdit() : RemoveLen(0) {} 37 }; 38 39 typedef std::map<FileOffset, FileEdit> FileEditsTy; 40 FileEditsTy FileEdits; 41 42 llvm::DenseMap<unsigned, SourceLocation> ExpansionToArgMap; 43 44 llvm::BumpPtrAllocator StrAlloc; 45 46 public: 47 EditedSource(const SourceManager &SM, const LangOptions &LangOpts, 48 const PreprocessingRecord *PPRec = 0) SourceMgr(SM)49 : SourceMgr(SM), LangOpts(LangOpts), PPRec(PPRec), 50 StrAlloc(/*size=*/512) { } 51 getSourceManager()52 const SourceManager &getSourceManager() const { return SourceMgr; } getLangOpts()53 const LangOptions &getLangOpts() const { return LangOpts; } getPreprocessingRecord()54 const PreprocessingRecord *getPreprocessingRecord() const { return PPRec; } 55 56 bool canInsertInOffset(SourceLocation OrigLoc, FileOffset Offs); 57 58 bool commit(const Commit &commit); 59 60 void applyRewrites(EditsReceiver &receiver); 61 void clearRewrites(); 62 copyString(StringRef str)63 StringRef copyString(StringRef str) { 64 char *buf = StrAlloc.Allocate<char>(str.size()); 65 std::memcpy(buf, str.data(), str.size()); 66 return StringRef(buf, str.size()); 67 } 68 StringRef copyString(const Twine &twine); 69 70 private: 71 bool commitInsert(SourceLocation OrigLoc, FileOffset Offs, StringRef text, 72 bool beforePreviousInsertions); 73 bool commitInsertFromRange(SourceLocation OrigLoc, FileOffset Offs, 74 FileOffset InsertFromRangeOffs, unsigned Len, 75 bool beforePreviousInsertions); 76 void commitRemove(SourceLocation OrigLoc, FileOffset BeginOffs, unsigned Len); 77 78 StringRef getSourceText(FileOffset BeginOffs, FileOffset EndOffs, 79 bool &Invalid); 80 FileEditsTy::iterator getActionForOffset(FileOffset Offs); 81 }; 82 83 } 84 85 } // end namespace clang 86 87 #endif 88