forked from 0ad/0ad
73 lines
1.8 KiB
C++
73 lines
1.8 KiB
C++
#ifndef STREAM_H__
|
|
#define STREAM_H__
|
|
|
|
/*
|
|
Stream: A system for input/output of data, particularly with chained streams
|
|
(e.g. file -> zlib decompressor -> data processor).
|
|
|
|
Similar to wxWidget's streams, but rewritten so that code can use this without
|
|
requiring wx.
|
|
|
|
TODO: This concept might not actually be any good; consider alternatives.
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
#include <limits>
|
|
|
|
namespace DatafileIO
|
|
{
|
|
class Stream
|
|
{
|
|
public:
|
|
enum whence { FROM_START, FROM_END, FROM_CURRENT };
|
|
|
|
virtual ~Stream() {};
|
|
virtual off_t Tell() const = 0;
|
|
virtual bool IsOk() const = 0;
|
|
};
|
|
|
|
class SeekableStream
|
|
{
|
|
public:
|
|
virtual void Seek(off_t pos, Stream::whence mode) = 0;
|
|
};
|
|
|
|
|
|
class InputStream : public Stream
|
|
{
|
|
public:
|
|
// Try to read up to 'size' bytes into buffer, and return the amount
|
|
// actually read.
|
|
virtual size_t Read(void* buffer, size_t size) = 0;
|
|
|
|
// Sets 'buffer' and 'size' to point to the data from the current cursor
|
|
// position up to max_size bytes (or the end of the file, whichever is
|
|
// reached first) and returns true if successful.
|
|
// If a derived stream doesn't implement these, a default (which just
|
|
// calls Read to get all the data) will be used instead. (It is implemented
|
|
// in e.g. streams that already have a pointer to all the data, so they
|
|
// don't need to do any memory copying.)
|
|
virtual bool AcquireBuffer(void*& buffer, size_t& size, size_t max_size = std::numeric_limits<size_t>::max());
|
|
virtual void ReleaseBuffer(void* buffer);
|
|
};
|
|
|
|
class OutputStream : public Stream
|
|
{
|
|
public:
|
|
virtual void Write(const void* buffer, size_t size) = 0;
|
|
};
|
|
|
|
|
|
// Specialisations to indicate that a stream allows seeking.
|
|
class SeekableInputStream : public InputStream, public SeekableStream
|
|
{
|
|
};
|
|
|
|
class SeekableOutputStream : public OutputStream, public SeekableStream
|
|
{
|
|
};
|
|
|
|
}
|
|
|
|
#endif // STREAM_H__
|