thirdparty/scummvm
2
0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

COMMON: Turn BufferedWriteStream into a template class, with disposeParentStream as param

Browse source 
svn-id: r54336
This commit is contained in:
Max Horn 2010-11-18 17:56:21 +00:00
parent c02a233084
commit a9dcb11c54
1 changed files with 57 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

48
common/stream.cpp
View file

@ -421,10 +421,10 @@ namespace {
/**
* Wrapper class which adds buffering to any WriteStream.
*/
template <DisposeAfterUse::Flag _disposeParentStream>
class BufferedWriteStream : public WriteStream {
protected:
WriteStream *_parentStream;
DisposeAfterUse::Flag _disposeParentStream;
byte *_buf;
uint32 _pos;
const uint32 _bufSize;
@ -436,19 +436,20 @@ protected:
* implemented by calling this method), except that it is not
* virtual, hence there is less overhead calling it.
*/
bool flushBuffer();
bool flushBuffer() {
const uint32 bytesToWrite = _pos;
if (bytesToWrite) {
_pos = 0;
if (_parentStream->write(_buf, bytesToWrite) != bytesToWrite)
return false;
}
return true;
}
public:
BufferedWriteStream(WriteStream *parentStream, uint32 bufSize, DisposeAfterUse::Flag disposeParentStream = DisposeAfterUse::NO);
virtual ~BufferedWriteStream();
virtual uint32 write(const void *dataPtr, uint32 dataSize);
virtual bool flush() { return flushBuffer(); }
};
BufferedWriteStream::BufferedWriteStream(WriteStream *parentStream, uint32 bufSize, DisposeAfterUse::Flag disposeParentStream)
BufferedWriteStream(WriteStream *parentStream, uint32 bufSize)
: _parentStream(parentStream),
_disposeParentStream(disposeParentStream),
_pos(0),
_bufSize(bufSize) {
@ -457,7 +458,7 @@ BufferedWriteStream::BufferedWriteStream(WriteStream *parentStream, uint32 bufSi
assert(_buf);
}
BufferedWriteStream::~BufferedWriteStream() {
virtual ~BufferedWriteStream() {
const bool flushResult = flushBuffer();
assert(flushResult);
@ -467,7 +468,7 @@ BufferedWriteStream::~BufferedWriteStream() {
delete[] _buf;
}
uint32 BufferedWriteStream::write(const void *dataPtr, uint32 dataSize) {
virtual uint32 write(const void *dataPtr, uint32 dataSize) {
// check if we have enough space for writing to the buffer
if (_bufSize - _pos >= dataSize) {
memcpy(_buf + _pos, dataPtr, dataSize);
@ -485,22 +486,21 @@ uint32 BufferedWriteStream::write(const void *dataPtr, uint32 dataSize) {
return dataSize;
}
bool BufferedWriteStream::flushBuffer() {
const uint32 bytesToWrite = _pos;
virtual bool flush() { return flushBuffer(); }
if (bytesToWrite) {
_pos = 0;
if (_parentStream->write(_buf, bytesToWrite) != bytesToWrite)
return false;
}
return true;
}
};
} // End of nameless namespace
WriteStream *wrapBufferedWriteStream(WriteStream *parentStream, uint32 bufSize, DisposeAfterUse::Flag disposeParentStream) {
if (parentStream)
return new BufferedWriteStream(parentStream, bufSize, disposeParentStream);
if (parentStream) {
switch (disposeParentStream) {
case DisposeAfterUse::YES:
return new BufferedWriteStream<DisposeAfterUse::YES>(parentStream, bufSize);
case DisposeAfterUse::NO:
return new BufferedWriteStream<DisposeAfterUse::NO>(parentStream, bufSize);
}
}
return 0;
}