1 //===-- Internals.h - Implementation Details---------------------*- 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_LIB_ARCMIGRATE_INTERNALS_H
11 #define LLVM_CLANG_LIB_ARCMIGRATE_INTERNALS_H
13 #include "clang/ARCMigrate/ARCMT.h"
14 #include "clang/Basic/Diagnostic.h"
15 #include "llvm/ADT/ArrayRef.h"
16 #include "llvm/ADT/Optional.h"
25 class CapturedDiagList {
26 typedef std::list<StoredDiagnostic> ListTy;
30 void push_back(const StoredDiagnostic &diag) { List.push_back(diag); }
32 bool clearDiagnostic(ArrayRef<unsigned> IDs, SourceRange range);
33 bool hasDiagnostic(ArrayRef<unsigned> IDs, SourceRange range) const;
35 void reportDiagnostics(DiagnosticsEngine &diags) const;
37 bool hasErrors() const;
39 typedef ListTy::const_iterator iterator;
40 iterator begin() const { return List.begin(); }
41 iterator end() const { return List.end(); }
44 void writeARCDiagsToPlist(const std::string &outPath,
45 ArrayRef<StoredDiagnostic> diags,
46 SourceManager &SM, const LangOptions &LangOpts);
48 class TransformActions {
49 DiagnosticsEngine &Diags;
50 CapturedDiagList &CapturedDiags;
52 void *Impl; // TransformActionsImpl.
55 TransformActions(DiagnosticsEngine &diag, CapturedDiagList &capturedDiags,
56 ASTContext &ctx, Preprocessor &PP);
59 void startTransaction();
60 bool commitTransaction();
61 void abortTransaction();
63 void insert(SourceLocation loc, StringRef text);
64 void insertAfterToken(SourceLocation loc, StringRef text);
65 void remove(SourceRange range);
66 void removeStmt(Stmt *S);
67 void replace(SourceRange range, StringRef text);
68 void replace(SourceRange range, SourceRange replacementRange);
69 void replaceStmt(Stmt *S, StringRef text);
70 void replaceText(SourceLocation loc, StringRef text,
71 StringRef replacementText);
72 void increaseIndentation(SourceRange range,
73 SourceLocation parentIndent);
75 bool clearDiagnostic(ArrayRef<unsigned> IDs, SourceRange range);
76 bool clearAllDiagnostics(SourceRange range) {
77 return clearDiagnostic(ArrayRef<unsigned>(), range);
79 bool clearDiagnostic(unsigned ID1, unsigned ID2, SourceRange range) {
80 unsigned IDs[] = { ID1, ID2 };
81 return clearDiagnostic(IDs, range);
83 bool clearDiagnostic(unsigned ID1, unsigned ID2, unsigned ID3,
85 unsigned IDs[] = { ID1, ID2, ID3 };
86 return clearDiagnostic(IDs, range);
89 bool hasDiagnostic(unsigned ID, SourceRange range) {
90 return CapturedDiags.hasDiagnostic(ID, range);
93 bool hasDiagnostic(unsigned ID1, unsigned ID2, SourceRange range) {
94 unsigned IDs[] = { ID1, ID2 };
95 return CapturedDiags.hasDiagnostic(IDs, range);
98 void reportError(StringRef error, SourceLocation loc,
99 SourceRange range = SourceRange());
100 void reportWarning(StringRef warning, SourceLocation loc,
101 SourceRange range = SourceRange());
102 void reportNote(StringRef note, SourceLocation loc,
103 SourceRange range = SourceRange());
105 bool hasReportedErrors() const { return ReportedErrors; }
107 class RewriteReceiver {
109 virtual ~RewriteReceiver();
111 virtual void insert(SourceLocation loc, StringRef text) = 0;
112 virtual void remove(CharSourceRange range) = 0;
113 virtual void increaseIndentation(CharSourceRange range,
114 SourceLocation parentIndent) = 0;
117 void applyRewrites(RewriteReceiver &receiver);
121 TransformActions &TA;
125 Transaction(TransformActions &TA) : TA(TA), Aborted(false) {
126 TA.startTransaction();
131 TA.commitTransaction();
135 TA.abortTransaction();
139 bool isAborted() const { return Aborted; }
142 class MigrationPass {
145 LangOptions::GCMode OrigGCMode;
146 MigratorOptions MigOptions;
148 TransformActions &TA;
149 const CapturedDiagList &CapturedDiags;
150 std::vector<SourceLocation> &ARCMTMacroLocs;
151 Optional<bool> EnableCFBridgeFns;
153 MigrationPass(ASTContext &Ctx, LangOptions::GCMode OrigGCMode,
154 Sema &sema, TransformActions &TA,
155 const CapturedDiagList &capturedDiags,
156 std::vector<SourceLocation> &ARCMTMacroLocs)
157 : Ctx(Ctx), OrigGCMode(OrigGCMode), MigOptions(),
158 SemaRef(sema), TA(TA), CapturedDiags(capturedDiags),
159 ARCMTMacroLocs(ARCMTMacroLocs) { }
161 const CapturedDiagList &getDiags() const { return CapturedDiags; }
163 bool isGCMigration() const { return OrigGCMode != LangOptions::NonGC; }
164 bool noNSAllocReallocError() const { return MigOptions.NoNSAllocReallocError; }
165 void setNSAllocReallocError(bool val) { MigOptions.NoNSAllocReallocError = val; }
166 bool noFinalizeRemoval() const { return MigOptions.NoFinalizeRemoval; }
167 void setNoFinalizeRemoval(bool val) {MigOptions.NoFinalizeRemoval = val; }
169 bool CFBridgingFunctionsDefined();
172 static inline StringRef getARCMTMacroName() {
173 return "__IMPL_ARCMT_REMOVED_EXPR__";
176 } // end namespace arcmt
178 } // end namespace clang