1//===- Diagnostics.cpp - C Interface for MLIR Diagnostics -----------------===//
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#include "mlir-c/Diagnostics.h"
10#include "mlir/CAPI/Diagnostics.h"
11#include "mlir/CAPI/IR.h"
12#include "mlir/CAPI/Support.h"
13#include "mlir/CAPI/Utils.h"
14#include "mlir/IR/Diagnostics.h"
15
16using namespace mlir;
17
18void mlirDiagnosticPrint(MlirDiagnostic diagnostic, MlirStringCallback callback,
19 void *userData) {
20 detail::CallbackOstream stream(callback, userData);
21 unwrap(diagnostic).print(os&: stream);
22}
23
24MlirLocation mlirDiagnosticGetLocation(MlirDiagnostic diagnostic) {
25 return wrap(cpp: unwrap(diagnostic).getLocation());
26}
27
28MlirDiagnosticSeverity mlirDiagnosticGetSeverity(MlirDiagnostic diagnostic) {
29 switch (unwrap(diagnostic).getSeverity()) {
30 case mlir::DiagnosticSeverity::Error:
31 return MlirDiagnosticError;
32 case mlir::DiagnosticSeverity::Warning:
33 return MlirDiagnosticWarning;
34 case mlir::DiagnosticSeverity::Note:
35 return MlirDiagnosticNote;
36 case mlir::DiagnosticSeverity::Remark:
37 return MlirDiagnosticRemark;
38 }
39 llvm_unreachable("unhandled diagnostic severity");
40}
41
42// Notes are stored in a vector, so note iterator range is a pair of
43// random access iterators, for which it is cheap to compute the size.
44intptr_t mlirDiagnosticGetNumNotes(MlirDiagnostic diagnostic) {
45 return static_cast<intptr_t>(llvm::size(Range: unwrap(diagnostic).getNotes()));
46}
47
48// Notes are stored in a vector, so the iterator is a random access iterator,
49// cheap to advance multiple steps at a time.
50MlirDiagnostic mlirDiagnosticGetNote(MlirDiagnostic diagnostic, intptr_t pos) {
51 return wrap(diagnostic&: *std::next(x: unwrap(diagnostic).getNotes().begin(), n: pos));
52}
53
54static void deleteUserDataNoop(void *userData) {}
55
56MlirDiagnosticHandlerID mlirContextAttachDiagnosticHandler(
57 MlirContext context, MlirDiagnosticHandler handler, void *userData,
58 void (*deleteUserData)(void *)) {
59 assert(handler && "unexpected null diagnostic handler");
60 if (deleteUserData == nullptr)
61 deleteUserData = deleteUserDataNoop;
62 DiagnosticEngine::HandlerID id =
63 unwrap(c: context)->getDiagEngine().registerHandler(
64 handler: [handler,
65 ownedUserData = std::unique_ptr<void, decltype(deleteUserData)>(
66 userData, deleteUserData)](Diagnostic &diagnostic) {
67 return unwrap(res: handler(wrap(diagnostic), ownedUserData.get()));
68 });
69 return static_cast<MlirDiagnosticHandlerID>(id);
70}
71
72void mlirContextDetachDiagnosticHandler(MlirContext context,
73 MlirDiagnosticHandlerID id) {
74 unwrap(c: context)->getDiagEngine().eraseHandler(
75 id: static_cast<DiagnosticEngine::HandlerID>(id));
76}
77
78void mlirEmitError(MlirLocation location, const char *message) {
79 emitError(loc: unwrap(c: location)) << message;
80}
81

source code of mlir/lib/CAPI/IR/Diagnostics.cpp