[Openmp-commits] [openmp] [OpenMP] Add ompTest library to OpenMP (PR #147381)

[Jan Patrick Lehr via Openmp-commits]

================
@@ -0,0 +1,177 @@
+//===- Logging.cpp - General logging class implementation -------*- 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
+//
+//===----------------------------------------------------------------------===//
+///
+/// \file
+/// Implements ompTest-tailored logging.
+///
+//===----------------------------------------------------------------------===//
+
+#include "Logging.h"
+
+using namespace omptest;
+using namespace logging;
+
+Logger::Logger(Level LogLevel, std::ostream &OutStream, bool FormatOutput)
+    : LoggingLevel(LogLevel), OutStream(OutStream), FormatOutput(FormatOutput) {
+  // Flush any buffered output
+  OutStream << std::flush;
+}
+
+Logger::~Logger() {
+  // Flush any buffered output
+  OutStream << std::flush;
+}
+
+std::map<Level, std::set<FormatOption>> AggregatedFormatOptions{
+    {Level::DIAGNOSTIC, {FormatOption::COLOR_LightBlue}},
+    {Level::INFO, {FormatOption::COLOR_LightGray}},
+    {Level::WARNING, {FormatOption::COLOR_LightYellow}},
+    {Level::ERROR, {FormatOption::COLOR_Red}},
+    {Level::CRITICAL, {FormatOption::COLOR_LightRed}},
+    {Level::Default, {FormatOption::NONE}},
+    {Level::ExpectedEvent, {FormatOption::BOLD, FormatOption::COLOR_Cyan}},
+    {Level::ObservedEvent, {FormatOption::COLOR_Cyan}},
+    {Level::OffendingEvent, {FormatOption::COLOR_Yellow}}};
+
+const char *logging::to_string(Level LogLevel) {
+  switch (LogLevel) {
+  case Level::DIAGNOSTIC:
+    return "DIAGNOSTIC";
+  case Level::INFO:
+    return "INFO";
+  case Level::WARNING:
+    return "WARNING";
+  case Level::ERROR:
+    return "ERROR";
+  case Level::CRITICAL:
+    return "CRITICAL";
+  default:
+    assert(false && "Requested string representation for unknown LogLevel");
+    return "UNKNOWN";
+  }
+}
+
+std::string logging::getFormatSequence(Level LogLevel) {
+  auto Options = AggregatedFormatOptions[LogLevel];
+  std::stringstream SS{"\033["};
+  SS << "\033[";
+  if (!Options.empty()) {
+    for (auto &Option : AggregatedFormatOptions[LogLevel])
+      SS << int(Option) << ';';
+    SS.seekp(-1, SS.cur);
+    SS << 'm';
+  } else {
+    // Fallback to NONE / reset formatting
+    SS << "0m";
+  }
+  return SS.str();
+}
+
+std::string logging::format(const std::string &Message, FormatOption Option) {
+  std::stringstream SS{"\033["};
+  SS << "\033[";
+  SS << int(Option) << 'm' << Message << "\033[0m";
+  return SS.str();
+}
+
+std::string logging::format(const std::string &Message,
+                            std::set<FormatOption> Options) {
+  std::stringstream SS{"\033["};
+  SS << "\033[";
+  for (auto &Option : Options)
+    SS << int(Option) << ';';
+  SS.seekp(-1, SS.cur);
+  SS << 'm' << Message << "\033[0m";
+  return SS.str();
+}
+
+void Logger::log(Level LogLevel, const std::string &Message) const {
+  // Serialize logging
+  std::lock_guard<std::mutex> Lock(LogMutex);
+
+  if (LoggingLevel > LogLevel)
+    return;
+
+  if (FormatOutput) {
+    OutStream << getFormatSequence(LogLevel) << '[' << to_string(LogLevel)
+              << "] " << Message << getFormatSequence() << std::endl;
+  } else {
+    OutStream << '[' << to_string(LogLevel) << "] " << Message << std::endl;
+  }
+}
+
+void Logger::eventMismatch(const omptest::OmptAssertEvent &OffendingEvent,
----------------
jplehr wrote:

Should this contain a verb?
```suggestion
void Logger::logEventMismatch(const omptest::OmptAssertEvent &OffendingEvent,
```

https://github.com/llvm/llvm-project/pull/147381