1 | //===-- ErrorHandling.h - Error handler -------------------------*- 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_TOOLS_LLVM_PROFGEN_ERRORHANDLING_H |
10 | #define LLVM_TOOLS_LLVM_PROFGEN_ERRORHANDLING_H |
11 | |
12 | #include "llvm/ADT/Twine.h" |
13 | #include "llvm/Support/Errc.h" |
14 | #include "llvm/Support/Error.h" |
15 | #include "llvm/Support/ErrorOr.h" |
16 | #include "llvm/Support/WithColor.h" |
17 | #include <system_error> |
18 | |
19 | using namespace llvm; |
20 | |
21 | [[noreturn]] inline void exitWithError(const Twine &Message, |
22 | StringRef Whence = StringRef(), |
23 | StringRef Hint = StringRef()) { |
24 | WithColor::error(OS&: errs(), Prefix: "llvm-profgen" ); |
25 | if (!Whence.empty()) |
26 | errs() << Whence.str() << ": " ; |
27 | errs() << Message << "\n" ; |
28 | if (!Hint.empty()) |
29 | WithColor::note() << Hint.str() << "\n" ; |
30 | ::exit(EXIT_FAILURE); |
31 | } |
32 | |
33 | [[noreturn]] inline void exitWithError(std::error_code EC, |
34 | StringRef Whence = StringRef()) { |
35 | exitWithError(Message: EC.message(), Whence); |
36 | } |
37 | |
38 | [[noreturn]] inline void exitWithError(Error E, StringRef Whence) { |
39 | exitWithError(EC: errorToErrorCode(Err: std::move(E)), Whence); |
40 | } |
41 | |
42 | template <typename T, typename... Ts> |
43 | T unwrapOrError(Expected<T> EO, Ts &&... Args) { |
44 | if (EO) |
45 | return std::move(*EO); |
46 | exitWithError(EO.takeError(), std::forward<Ts>(Args)...); |
47 | } |
48 | |
49 | inline void emitWarningSummary(uint64_t Num, uint64_t Total, StringRef Msg) { |
50 | if (!Total || !Num) |
51 | return; |
52 | WithColor::warning() << format(Fmt: "%.2f" , Vals: static_cast<double>(Num) * 100 / Total) |
53 | << "%(" << Num << "/" << Total << ") " << Msg << "\n" ; |
54 | } |
55 | |
56 | #endif |
57 | |