stream.h

来自「这是一个完整的图书馆供销存管理系统」· C头文件 代码 · 共 52 行

/***************************************************************************
                          stream.h  -  description
                             -------------------
    begin                : Fri Jul 20 2001
    copyright            : (C) 2001 by Mark
    email                : [e-mail]
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/


#ifndef STREAM_H
#define STREAM_H

#include <iostream.h>


class CStream : public streambuf, public iostream
{
protected:
	int m_iBufSize;
	char* m_pchGetBuf;
	char* m_pchPutBuf;
	
protected:
	CStream(int iBufSize = 512)
			: streambuf(), iostream((streambuf*)this),
			  m_iBufSize(iBufSize), m_pchGetBuf(0), m_pchPutBuf(0)
			{
				AllocateBuffer();
			}
	virtual ~CStream() { FreeBuffer(); }
	
public:
	inline char* GetPtr() { return gptr(); }
	int BufSize() { return m_iBufSize; }
	virtual int sync();	
	
private:
	void AllocateBuffer();
	void FreeBuffer();
};

#endif