/Users/buildslave/jenkins/workspace/coverage/llvm-project/clang/include/clang/Edit/EditedSource.h
Line | Count | Source |
1 | | //===- EditedSource.h - Collection of source edits --------------*- C++ -*-===// |
2 | | // |
3 | | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | | // See https://llvm.org/LICENSE.txt for license information. |
5 | | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | | // |
7 | | //===----------------------------------------------------------------------===// |
8 | | |
9 | | #ifndef LLVM_CLANG_EDIT_EDITEDSOURCE_H |
10 | | #define LLVM_CLANG_EDIT_EDITEDSOURCE_H |
11 | | |
12 | | #include "clang/Basic/IdentifierTable.h" |
13 | | #include "clang/Basic/LLVM.h" |
14 | | #include "clang/Basic/SourceLocation.h" |
15 | | #include "clang/Edit/FileOffset.h" |
16 | | #include "llvm/ADT/DenseMap.h" |
17 | | #include "llvm/ADT/SmallVector.h" |
18 | | #include "llvm/ADT/StringRef.h" |
19 | | #include "llvm/Support/Allocator.h" |
20 | | #include <map> |
21 | | #include <tuple> |
22 | | #include <utility> |
23 | | |
24 | | namespace clang { |
25 | | |
26 | | class LangOptions; |
27 | | class PPConditionalDirectiveRecord; |
28 | | class SourceManager; |
29 | | |
30 | | namespace edit { |
31 | | |
32 | | class Commit; |
33 | | class EditsReceiver; |
34 | | |
35 | | class EditedSource { |
36 | | const SourceManager &SourceMgr; |
37 | | const LangOptions &LangOpts; |
38 | | const PPConditionalDirectiveRecord *PPRec; |
39 | | |
40 | | struct FileEdit { |
41 | | StringRef Text; |
42 | | unsigned RemoveLen = 0; |
43 | | |
44 | 7.27k | FileEdit() = default; |
45 | | }; |
46 | | |
47 | | using FileEditsTy = std::map<FileOffset, FileEdit>; |
48 | | |
49 | | FileEditsTy FileEdits; |
50 | | |
51 | | struct MacroArgUse { |
52 | | IdentifierInfo *Identifier; |
53 | | SourceLocation ImmediateExpansionLoc; |
54 | | |
55 | | // Location of argument use inside the top-level macro |
56 | | SourceLocation UseLoc; |
57 | | |
58 | 38 | bool operator==(const MacroArgUse &Other) const { |
59 | 38 | return std::tie(Identifier, ImmediateExpansionLoc, UseLoc) == |
60 | 38 | std::tie(Other.Identifier, Other.ImmediateExpansionLoc, |
61 | 38 | Other.UseLoc); |
62 | 38 | } |
63 | | }; |
64 | | |
65 | | llvm::DenseMap<SourceLocation, SmallVector<MacroArgUse, 2>> ExpansionToArgMap; |
66 | | SmallVector<std::pair<SourceLocation, MacroArgUse>, 2> |
67 | | CurrCommitMacroArgExps; |
68 | | |
69 | | IdentifierTable IdentTable; |
70 | | llvm::BumpPtrAllocator StrAlloc; |
71 | | |
72 | | public: |
73 | | EditedSource(const SourceManager &SM, const LangOptions &LangOpts, |
74 | | const PPConditionalDirectiveRecord *PPRec = nullptr) |
75 | 3.59k | : SourceMgr(SM), LangOpts(LangOpts), PPRec(PPRec), IdentTable(LangOpts) {} |
76 | | |
77 | 2.41k | const SourceManager &getSourceManager() const { return SourceMgr; } |
78 | 2.41k | const LangOptions &getLangOpts() const { return LangOpts; } |
79 | | |
80 | 2.41k | const PPConditionalDirectiveRecord *getPPCondDirectiveRecord() const { |
81 | 2.41k | return PPRec; |
82 | 2.41k | } |
83 | | |
84 | | bool canInsertInOffset(SourceLocation OrigLoc, FileOffset Offs); |
85 | | |
86 | | bool commit(const Commit &commit); |
87 | | |
88 | | void applyRewrites(EditsReceiver &receiver, bool adjustRemovals = true); |
89 | | void clearRewrites(); |
90 | | |
91 | 5.99k | StringRef copyString(StringRef str) { return str.copy(StrAlloc); } |
92 | | StringRef copyString(const Twine &twine); |
93 | | |
94 | | private: |
95 | | bool commitInsert(SourceLocation OrigLoc, FileOffset Offs, StringRef text, |
96 | | bool beforePreviousInsertions); |
97 | | bool commitInsertFromRange(SourceLocation OrigLoc, FileOffset Offs, |
98 | | FileOffset InsertFromRangeOffs, unsigned Len, |
99 | | bool beforePreviousInsertions); |
100 | | void commitRemove(SourceLocation OrigLoc, FileOffset BeginOffs, unsigned Len); |
101 | | |
102 | | StringRef getSourceText(FileOffset BeginOffs, FileOffset EndOffs, |
103 | | bool &Invalid); |
104 | | FileEditsTy::iterator getActionForOffset(FileOffset Offs); |
105 | | void deconstructMacroArgLoc(SourceLocation Loc, |
106 | | SourceLocation &ExpansionLoc, |
107 | | MacroArgUse &ArgUse); |
108 | | |
109 | | void startingCommit(); |
110 | | void finishedCommit(); |
111 | | }; |
112 | | |
113 | | } // namespace edit |
114 | | |
115 | | } // namespace clang |
116 | | |
117 | | #endif // LLVM_CLANG_EDIT_EDITEDSOURCE_H |