Extending fmt::join to support C++20-only ranges.

fmtlib · Oct 15, 2021 · df003a7 · df003a7
1 parent f5371a7
commit df003a7
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 3 deletions.
diff --git a/include/fmt/format.h b/include/fmt/format.h
@@ -2748,7 +2748,12 @@ using arg_join FMT_DEPRECATED_ALIAS = join_view<It, Sentinel, Char>;
 template <typename It, typename Sentinel, typename Char>
 struct formatter<join_view<It, Sentinel, Char>, Char> {
  private:
-  using value_type = typename std::iterator_traits<It>::value_type;
+  using value_type =
+    #ifdef __cpp_lib_ranges
+    std::iter_value_t<It>;
+    #else
+    typename std::iterator_traits<It>::value_type;
+    #endif
   using context = buffer_context<Char>;
   using mapper = detail::arg_mapper<context>;
 
@@ -2782,11 +2787,13 @@ struct formatter<join_view<It, Sentinel, Char>, Char> {
     auto it = value.begin;
     auto out = ctx.out();
     if (it != value.end) {
-      out = value_formatter_.format(map(*it++), ctx);
+      out = value_formatter_.format(map(*it), ctx);
+      ++it;
       while (it != value.end) {
         out = detail::copy_str<Char>(value.sep.begin(), value.sep.end(), out);
         ctx.advance_to(out);
-        out = value_formatter_.format(map(*it++), ctx);
+        out = value_formatter_.format(map(*it), ctx);
+        ++it;
       }
     }
     return out;

diff --git a/test/ranges-test.cc b/test/ranges-test.cc
@@ -258,6 +258,33 @@ struct zstring {
   zstring_sentinel end() const { return {}; }
 };
 
+#ifdef __cpp_lib_ranges
+struct cpp20_only_range {
+  struct iterator {
+    int i = 0;
+
+    using value_type = int;
+    using difference_type = std::ptrdiff_t;
+    using iterator_concept = std::input_iterator_tag;
+
+    iterator() = default;
+    iterator(int i) : i(i) { }
+    int operator*() const { return i; }
+    iterator& operator++() { ++i; return *this; }
+    void operator++(int) { ++*this; }
+    bool operator==(const iterator& rhs) const { return i == rhs.i; }
+  };
+
+  int lo;
+  int hi;
+
+  iterator begin() const { return iterator(lo); }
+  iterator end() const { return iterator(hi); }
+};
+
+static_assert(std::input_iterator<cpp20_only_range::iterator>);
+#endif
+
 TEST(ranges_test, join_sentinel) {
   auto hello = zstring{"hello"};
   EXPECT_EQ(fmt::format("{}", hello), "['h', 'e', 'l', 'l', 'o']");
@@ -282,6 +309,10 @@ TEST(ranges_test, join_range) {
 
   const auto z = std::vector<int>(3u, 0);
   EXPECT_EQ(fmt::format("{}", fmt::join(z, ",")), "0,0,0");
+
+  #ifdef __cpp_lib_ranges
+  EXPECT_EQ(fmt::format("{}", fmt::join(cpp20_only_range{.lo=0, .hi=5}, ",")), "0,1,2,3,4");
+  #endif
 }
 #endif  // FMT_RANGES_TEST_ENABLE_JOIN