iostream 等效于 snprintf(NULL, 0, format_string, args...)

Question

我想找出流格式化操作在不从堆中分配内存的情况下会产生的字符数。在 C 中，可以使用

int nchars = snprintf(NULL, 0, format_string, args...);

如何在C++的ostream框架内完成？

std::ostringstream 的实现可以从堆中分配内存：

template <class T>
int find_nchar(const T& value) {
  std::ostringstream os; // may allocate memory from the heap
  os << value;
  return os.str().size(); // may allocate memory from the heap
}

我想我需要制作一个自定义的 ostream 类来实现这一点。自定义 ostream 应该尊重可以为普通 std::ostream 设置的所有格式标志。

我正在寻找只使用 C++ 标准库的解决方案，例如，不使用 boost::iostreams。

Answer 1

与自定义 std::ostream 相比，实现自定义 std::streambuf 可能更容易（也许更灵活），然后可以与任何 std::ostream 一起使用。

#include <streambuf>

template <class CharT, class Traits = std::char_traits<CharT>>
struct counting_streambuf: std::basic_streambuf<CharT, Traits> {
  using base_t = std::basic_streambuf<CharT, Traits>;
  using typename base_t::char_type;
  using typename base_t::int_type;

  std::streamsize count = 0;

  std::streamsize xsputn(const char_type* /* unused */, std::streamsize n)
    override
    {
      count += n;
      return n;
    }

  int_type overflow(int_type ch)
    override
    {
      ++count;
      return ch;
    }

};

然后用作...

#include <iostream>

int
main (int argc, char **argv)
{
  using char_type = decltype(std::cout)::char_type;

  counting_streambuf<char_type> csb;

  /*
   * Associate the counting_streambuf with std::cout whilst
   * retaining a pointer to the original std::streambuf.
   */
  auto *oldbuf = std::cout.rdbuf(&csb);
  std::cout << "Some text goes here...\n";

  /*
   * Restore the original std::streambuf.
   */
  std::cout.rdbuf(oldbuf);
  std::cout << "output length is " << csb.count << " characters\n";
}

运行上述结果...

output length is 23 characters

编辑：原始解决方案没有超载overflow。这适用于 Linux，但不适用于 Windows。感谢 Boost 的 Peter Dimov，他找到了解决方案。