blob: a58a435b16a9030b01725ee242388c1ef4584a2b (
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
//===--- Diagnostics.h - Helper class for error diagnostics -----*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// Diagnostics class to manage error messages. Implementation shares similarity
// to clang-query Diagnostics.
//
//===----------------------------------------------------------------------===//
#ifndef MLIR_TOOLS_MLIRQUERY_MATCHER_DIAGNOSTICS_H
#define MLIR_TOOLS_MLIRQUERY_MATCHER_DIAGNOSTICS_H
#include "mlir/Query/Matcher/ErrorBuilder.h"
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/ADT/Twine.h"
#include "llvm/Support/raw_ostream.h"
#include <string>
#include <vector>
namespace mlir::query::matcher::internal {
// Diagnostics class to manage error messages.
class Diagnostics {
public:
// Helper stream class for constructing error messages.
class ArgStream {
public:
ArgStream(std::vector<std::string> *out) : out(out) {}
template <class T>
ArgStream &operator<<(const T &arg) {
return operator<<(llvm::Twine(arg));
}
ArgStream &operator<<(const llvm::Twine &arg);
private:
std::vector<std::string> *out;
};
// Add an error message with the specified range and error type.
// Returns an ArgStream object to allow constructing the error message using
// the << operator.
ArgStream addError(SourceRange range, ErrorType error);
// Print all error messages to the specified output stream.
void print(llvm::raw_ostream &os) const;
private:
// Information stored for one frame of the context.
struct ContextFrame {
SourceRange range;
std::vector<std::string> args;
};
// Information stored for each error found.
struct ErrorContent {
std::vector<ContextFrame> contextStack;
struct Message {
SourceRange range;
ErrorType type;
std::vector<std::string> args;
};
std::vector<Message> messages;
};
void printMessage(const ErrorContent::Message &message,
const llvm::Twine Prefix, llvm::raw_ostream &os) const;
void printErrorContent(const ErrorContent &content,
llvm::raw_ostream &os) const;
std::vector<ContextFrame> contextStack;
std::vector<ErrorContent> errorValues;
};
} // namespace mlir::query::matcher::internal
#endif // MLIR_TOOLS_MLIRQUERY_MATCHER_DIAGNOSTICS_H
|
