实现std :: basic_streambuf子类来操作输入

时间:2014-01-29 01:44:37

标签: c++ streambuf

我有一个std::basic_streambuf子类,导致所有输出都以大写形式写成,如下所示:

class upper_streambuf : public std::streambuf
{
    public:
        upper_streambuf(std::streambuf &real)
          : m_realBuffer(real)
        {
        }

    protected:
        virtual int overflow(int c)
        {
            int uc = std::toupper(c);
            m_realBuffer.sputc(uc);
            return uc;
        }

    private:
        std::streambuf &m_realBuffer;
};

我像这样使用它(似乎工作正常):

upper_streambuf buf(*std::cout.rdbuf());
std::ostream ucout(&buf);

ucout << "Hello, world!" << std::endl; // prints "HELLO, WORLD!"

我想要实现的是或多或少的逆,我想从流中读取并将所有输入转换为小写。我有以下内容:

class lower_streambuf : public std::streambuf
{
    public:
        lower_streambuf(std::streambuf &real)
          : m_realBuffer(real)
        {
        }

    protected:
        virtual int underflow()
        {
            return std::tolower(m_realBuffer.sbumpc());
        }

    private:
        std::streambuf &m_realBuffer;
};

然而,当我尝试使用这样的时候:

lower_streambuf buf(*std::cin.rdbuf());
std::istream lcin(&buf);
std::string line;

std::getline(lcin, line);

结果是分段错误。我是否压倒或调用了错误的功能?请注意我在C ++中有点新手。还请注意,我理解我可以完全阅读输入,并在读完后将其简单地转换为小写,但这更多用于学习/学术目的而不是任何实际用途。

1 个答案:

答案 0 :(得分:4)

underflow()有你未能确定的后置条件:当你从它返回时,你需要在你的streambuf中有一个获取区域来保存你正在返回的角色。

单字符缓冲区就足够了:

    protected:
        virtual int underflow()
        {
            ch = std::tolower(m_realBuffer.sbumpc());
  +         setg(&ch, &ch, &ch+1); 
            return ch;
        }
    private:
  +     char ch; // input buffer!
        std::streambuf &m_realBuffer;
};