cpp-common/bt2c: add format_as() template functions for wise enums

[babeltrace.git] / src / cpp-common / bt2c / fmt.hpp

diff --git a/src/cpp-common/bt2c/fmt.hpp b/src/cpp-common/bt2c/fmt.hpp
index 0121806da584e17138e4c40a730056a43f0f5489..4633ebca73b03233e1cc2c9ec7e9df120c66d463 100644 (file)
--- a/src/cpp-common/bt2c/fmt.hpp
+++ b/src/cpp-common/bt2c/fmt.hpp
@@ -7,11 +7,26 @@
 #include "common/common.h"
 #include "cpp-common/bt2/message.hpp"
 #include "cpp-common/vendor/fmt/format.h" /* IWYU pragma: keep */
+#include "cpp-common/vendor/wise-enum/wise_enum.h"
 
 #include "uuid.hpp"
 
+namespace internal {
+
+template <typename T>
+using EnableIfIsWiseEnum =
+    typename std::enable_if<wise_enum::is_wise_enum<T>::value, const char *>::type;
+
+} /* namespace internal */
+
 namespace bt2 {
 
+template <typename T>
+::internal::EnableIfIsWiseEnum<T> format_as(const T val) noexcept
+{
+    return wise_enum::to_string<T>(val);
+}
+
 inline const char *format_as(const MessageType type)
 {
     return bt_common_message_type_string(static_cast<bt_message_type>(type));
@@ -21,6 +36,12 @@ inline const char *format_as(const MessageType type)
 
 namespace bt2c {
 
+template <typename T>
+::internal::EnableIfIsWiseEnum<T> format_as(const T val) noexcept
+{
+    return wise_enum::to_string<T>(val);
+}
+
 inline std::string format_as(const UuidView uuid)
 {
     return uuid.str();