blob: 64a994d88b9d96bdc81eb8acee25361f4546a27c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
//===--- ToolRefactoringResultConsumer.h - ----------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H
#define LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H
#include "clang/AST/ASTContext.h"
#include "clang/Tooling/Refactoring/RefactoringResultConsumer.h"
namespace clang {
namespace refactor {
/// An interface that subclasses the \c RefactoringResultConsumer interface
/// that stores the reference to the TU-specific diagnostics engine.
class ClangRefactorToolConsumerInterface
: public tooling::RefactoringResultConsumer {
public:
/// Called when a TU is entered.
void beginTU(ASTContext &Context) {
assert(!Diags && "Diags has been set");
Diags = &Context.getDiagnostics();
}
/// Called when the tool is done with a TU.
void endTU() {
assert(Diags && "Diags unset");
Diags = nullptr;
}
DiagnosticsEngine &getDiags() const {
assert(Diags && "no diags");
return *Diags;
}
private:
DiagnosticsEngine *Diags = nullptr;
};
} // end namespace refactor
} // end namespace clang
#endif // LLVM_CLANG_TOOLS_CLANG_REFACTOR_TOOL_REFACTORING_RESULT_CONSUMER_H
|
