1 files changed, 50 insertions, 0 deletions
diff --git a/clang/lib/CodeGen/TrapReasonBuilder.cpp b/clang/lib/CodeGen/TrapReasonBuilder.cpp
new file mode 100644
index 000000000000..5881229bf747
--- /dev/null
+++ b/clang/lib/CodeGen/TrapReasonBuilder.cpp
@@ -0,0 +1,50 @@
+//===----------------------------------------------------------------------===//
+//
+// 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
+//
+//===----------------------------------------------------------------------===//
+///
+/// \file
+/// This file implements TrapReasonBuilder and related classes.
+///
+//===----------------------------------------------------------------------===//
+#include "TrapReasonBuilder.h"
+
+namespace clang {
+namespace CodeGen {
+
+TrapReasonBuilder::TrapReasonBuilder(DiagnosticsEngine *DiagObj,
+                                     unsigned DiagID, TrapReason &TR)
+    : DiagnosticBuilder(DiagObj, SourceLocation(), DiagID), TR(TR) {
+  assert(DiagObj->getDiagnosticIDs()->isTrapDiag(DiagID));
+}
+
+TrapReasonBuilder::~TrapReasonBuilder() {
+  // Store the trap message and category into the TrapReason object.
+  getMessage(TR.Message);
+  TR.Category = getCategory();
+
+  // Make sure that when `DiagnosticBuilder::~DiagnosticBuilder()`
+  // calls `Emit()` that it does nothing.
+  Clear();
+}
+
+void TrapReasonBuilder::getMessage(SmallVectorImpl<char> &Storage) {
+  // Render the Diagnostic
+  Diagnostic Info(getDiagnosticsEngine(), *this);
+  Info.FormatDiagnostic(Storage);
+}
+
+StringRef TrapReasonBuilder::getCategory() {
+  auto CategoryID =
+      getDiagnosticsEngine()->getDiagnosticIDs()->getCategoryNumberForDiag(
+          getDiagID());
+  if (CategoryID == 0)
+    return "";
+  return getDiagnosticsEngine()->getDiagnosticIDs()->getCategoryNameFromID(
+      CategoryID);
+}
+} // namespace CodeGen
+} // namespace clang