Add support for _BitInt on clang

Issue #4007
fmtlib · Jul 13, 2024 · 5a7b886 · 5a7b886
1 parent 3fe4641
commit 5a7b886
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 1 deletion.
diff --git a/include/fmt/base.h b/include/fmt/base.h
@@ -427,6 +427,14 @@ enum class uint128_opt {};
 template <typename T> auto convert_for_visit(T) -> monostate { return {}; }
 #endif
 
+#ifndef FMT_USE_BITINT
+#  if FMT_CLANG_VERSION && FMT_CLANG_VERSION >= 1400
+#    define FMT_USE_BITINT 1
+#  else
+#    define FMT_USE_BITINT 0
+#  endif
+#endif
+
 // Casts a nonnegative integer to unsigned.
 template <typename Int>
 FMT_CONSTEXPR auto to_unsigned(Int value) -> make_unsigned_t<Int> {
@@ -1188,7 +1196,7 @@ namespace detail {
 namespace detect {
 template <typename T>
 struct is_back_insert_iterator<basic_appender<T>> : std::true_type {};
-}
+}  // namespace detect
 
 template <typename T, typename Enable = void>
 struct locking : std::true_type {};
@@ -1487,6 +1495,19 @@ template <typename Context> struct arg_mapper {
   FMT_MAP_API auto map(double val) -> double { return val; }
   FMT_MAP_API auto map(long double val) -> long double { return val; }
 
+#if FMT_USE_BITINT
+  template <class T, int N = 0> struct is_bitint : std::false_type {};
+
+  template <int N> struct is_bitint<_BitInt(N)> : std::true_type {};
+
+  template <int N> struct is_bitint<unsigned _BitInt(N)> : std::true_type {};
+
+  template <class T, FMT_ENABLE_IF(is_bitint<remove_cvref_t<T>>::value)>
+  FMT_MAP_API auto map(T&& val) -> decltype(val) {
+    return val;
+  }
+#endif
+
   FMT_MAP_API auto map(char_type* val) -> const char_type* { return val; }
   FMT_MAP_API auto map(const char_type* val) -> const char_type* { return val; }
   template <typename T, typename Char = char_t<T>,

diff --git a/test/format-test.cc b/test/format-test.cc
@@ -2442,3 +2442,27 @@ FMT_END_NAMESPACE
 TEST(format_test, ustring) {
   EXPECT_EQ(fmt::format("{}", ustring()), "ustring");
 }
+
+#if FMT_USE_BITINT
+using BitIntS7 = _BitInt(7);
+using BitIntU32 = unsigned _BitInt(32);
+FMT_BEGIN_NAMESPACE
+template <> struct formatter<BitIntS7> : formatter<int8_t> {};
+
+template <> struct formatter<BitIntU32> : formatter<uint32_t> {};
+FMT_END_NAMESPACE
+
+TEST(format_test, bitint) {
+  EXPECT_EQ(fmt::format("{}", BitIntS7{0}), "0");
+  EXPECT_EQ(fmt::format("{}", BitIntU32{4294967295}), "4294967295");
+
+  auto a = BitIntS7{0};
+  auto b = BitIntU32{4294967295};
+  const auto c = BitIntS7{0};
+  const auto d = BitIntU32{4294967295};
+  EXPECT_EQ(fmt::format("{}", a), "0");
+  EXPECT_EQ(fmt::format("{}", b), "4294967295");
+  EXPECT_EQ(fmt::format("{}", c), "0");
+  EXPECT_EQ(fmt::format("{}", d), "4294967295");
+}
+#endif