扩展C ++ ostream

时间:2012-06-06 20:30:00

标签: c++ iostream

我正在尝试通过扩展std::streambuf来了解有关C ++ I / O流库的工作原理的更多信息。作为学习实验,我的目标是简单地创建一个自定义流,将所有输出定向到std::cerr。看起来很简单:

#include <iostream>
using namespace std;

class my_ostreambuf : public std::streambuf
{
    public:

    protected:

    std::streamsize xsputn(const char * s, std::streamsize n)
    {
        std::cerr << "Redirecting to cerr: " << s << std::endl;
        return n;
    }

};

int main()
{
    my_ostreambuf buf;
    std::ostream os(&buf);
    os << "TEST";
}

这似乎有效,因为它会打印Redirecting to cerr: TEST。当单个字符(而不是字符串)通过std::ostream::sputc插入到流中时,问题是无效。例如:

int main()
{
    my_ostreambuf buf;
    std::ostream os(&buf);
    os << "ABC"; // works
    std::string s("TEST");
    std::copy(s.begin(), s.end(), std::ostreambuf_iterator<char>(os)); // DOESN'T WORK
}

我猜的问题是xsputn无法处理单个字符的插入。 (我猜sputc内部没有调用xsputn?)但是,查看std::streambuf中的list of virtual protected functions,我看不到任何我应该覆盖的函数处理单个字符插入。

那么,我该怎么做呢?

1 个答案:

答案 0 :(得分:5)

单字符输出由overflow处理。如果overflow执行实际输出,您可以按照xsputn的方式实施xsputn

int_type overflow(int_type c = traits_type::eof())
{
    if (c == traits_type::eof())
        return traits_type::eof();
    else
    {
        char_type ch = traits_type::to_char_type(c);
        return xsputn(&ch, 1) == 1 ? c : traits_type::eof();
    }
}