STL之自定义缓冲区

简介

流缓冲区是一种I/O缓冲区，接口由basic_streambuf定义。针对字符型别char和wchar,标准程序库分别提供预先定义好的流缓冲区（streambuf）和宽字符流缓冲区（wstreambuf）。尤其是在特殊通道上，各类可以当做基类。

• 函数eback(),gptr()和egptr()构成了read(input)缓冲区的界面。
• 函数pbase(),pptr(),epptr()构成了write(output)缓冲区的界面。

pbase()是指output stream缓冲区的起始位置。pptr()是当前写入位置。epptr()是output 缓冲区的结尾。pbase()至pptr()之间的序列字符已被写至相应的输出通道，但未被清空。

输出缓冲区

对于程序员和程序开发者来说，类basic_streambuf仅仅是发送（sent）和提取（extracted）字符的地方。通常有两个公共函数，用于写入字符。

int streambuf::sputc( int nCh );

int streambuf::sputn( const char* pch, int nCount );

函数sputc发生错误时，会返回traits_type::eof(）。

成员函数sputc()可用来向缓冲区中写入一个字符，如果当时有一个空的改写位置，字符就会被复制到该位置上。之后，指向当前改写位置的那个指针就会加1。如果缓冲区是空的，就调用虚函数overflow（）将output缓冲区的内容发送至对应的输出管道中。

所以，通过重载overflow()函数可以自定义streambuffer。

#include <IOSTREAM>#include <streambuf>#include <LOCALE>#include <CSTDIO>using namespace std;class outbuf : public std::streambuf{protected:virtual int_type overflow(int_type c){if(c != EOF){c = std::toupper(c ,getloc());if(putchar(c) == EOF){return EOF;}}return c;}};void main(){outbuf ob;std::ostream out(&ob);int num = 56;out << "56 十六进制数值: " << std::hex << std::showbase << num << endl;}

上面是一个没有缓冲区的demo。因为缓冲区为空，每输出一个字符都会调用一次overflow()函数，在overflow中直接将字符输出。

#include <iostream>#include <io.h>#include <streambuf>#include <cstdio>using namespace std;static const int bufferSize = 10;class outbuf : public std::streambuf{protected:char buffer[bufferSize];//定义一个数组表示缓冲区。public:outbuf(){setp(buffer ,buffer + bufferSize - 1);//设置streambuf对象的缓冲区；}virtual ~outbuf(){sync();//程序退出时，将数组中的残余数据输出至外部设备。}protected:int flushBuffer(){int num = pptr() - pbase();if(write(1 ,buffer , num) != num)//输出。{return EOF;}pbump(-num);//将当前写指针清空，指向缓冲区的开始位置。return num;}virtual int_type overflow(int_type c){/*注意setp(buffer ,buffer + bufferSize - 1)，所以pptr()==epptr()的时候,（也就是调用underflow的时候）,pptr()指向缓冲区的最后一个位置（数组的最后一个位置）。所以pptr()指向有效内存。*/if(c != EOF){*pptr() = c;pbump(1);//使当前位置指向数组最后一个元素之后。}if(flushBuffer() == EOF){return EOF;}return c;}virtual int sync(){if(flushBuffer() == EOF){return -1;}return 0;}};class fdostream : public std::ostream{protected:outbuf buf;public:fdostream(int fd):ostream(0){rdbuf(&buf);//设置streambuf对象。}};void main(){fdostream out(1);out << "51 hexadecimal: " << std::hex << std::showbase << 51 << endl;}

输入缓冲区

函数sgetc()和sbumpc()可以从streambuf中读取一个字符，不同之处是sbumpc()会使当前指针后移，而sgetc()仅返回当前字符。

如果缓冲区为空，就没有可用字符了。缓冲区必须重新补给。如果没有可用字符，函数sbumpc()会调用虚函数uflow()，而uflow()的默认行为是调用underflow(),移动“读取指针”。

需要说明一下的时,eback()返回缓冲区中有效数据的起始位置。epptr()返回缓冲区有效数据末端位置。pptr()返回当前读取位置。这里需要注意的是，eback()并不一定等于pptr()，因为为了支持回退，eback()和pptr()之间存储已经读取过的字符。而epptr()并不一定是缓冲区数组的最后一个元素，因为可能没有从输入设备读取这么多的数据。

在给出demo之前，由于输入缓冲区支持回退，也就是读取指针可以左移，读取之前读取过的字符。所以，在重写函数underflow自定义自己的streambuf时，也需要支持这种特性。

#include <iostream>#include <streambuf>#include <cstring>#include <io.h>using namespace std;static const int bufferSize = 10;static const int maxBackNum = 4;class inbuf : public std::streambuf{protected:char buffer[bufferSize];public:inbuf(){setg(buffer + maxBackNum ,buffer + maxBackNum ,buffer + maxBackNum);}protected:virtual int_type underflow(){if(gptr() < egptr()){return *gptr();}int numputback;numputback = gptr() - eback();if(numputback > maxBackNum)numputback = maxBackNum;memcpy(buffer + maxBackNum - numputback ,gptr() - numputback ,numputback);int num ;num = read(0  ,buffer + maxBackNum ,bufferSize - maxBackNum);if(num < 0)return EOF;setg(buffer + (maxBackNum - numputback) ,buffer + maxBackNum ,buffer + maxBackNum + num);return *gptr();}};void main(){inbuf ib;std::istream in(&ib);char c;for(int i = 1 ;i <= 20 ;i++){in.get(c);cout << c << flush;if(i == 8){in.unget();in.unget();}}cout << endl;}