You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
569 lines
22 KiB
569 lines
22 KiB
14 years ago
|
/*==LICENSE==*
|
||
|
|
||
|
CyanWorlds.com Engine - MMOG client, server and tools
|
||
|
Copyright (C) 2011 Cyan Worlds, Inc.
|
||
|
|
||
|
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 3 of the License, or
|
||
|
(at your option) any later version.
|
||
|
|
||
|
This program is distributed in the hope that it will be useful,
|
||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
GNU General Public License for more details.
|
||
|
|
||
|
You should have received a copy of the GNU General Public License
|
||
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
||
|
You can contact Cyan Worlds, Inc. by email legal@cyan.com
|
||
|
or by snail mail at:
|
||
|
Cyan Worlds, Inc.
|
||
|
14617 N Newport Hwy
|
||
|
Mead, WA 99021
|
||
|
|
||
|
*==LICENSE==*/
|
||
|
#ifndef hsStream_Defined
|
||
|
#define hsStream_Defined
|
||
|
|
||
|
#include <stdarg.h> // Included for GCC 3.2.2+
|
||
|
|
||
|
#include "hsTypes.h"
|
||
|
#include "hsMemory.h"
|
||
|
|
||
|
namespace hsPackFileSys {
|
||
|
struct FileEntry;
|
||
|
}
|
||
|
|
||
|
// Define this for use of Streams with Logging (commonly used w/ a packet sniffer)
|
||
|
// These streams log their reads to an event list
|
||
|
//#define STREAM_LOGGER
|
||
|
|
||
|
#ifndef STREAM_LOGGER
|
||
|
#define hsReadOnlyLoggingStream hsReadOnlyStream
|
||
|
#define LogRead(byteCount, buffer, desc) Read(byteCount, buffer)
|
||
|
#define LogReadSafeString() ReadSafeString();
|
||
|
#define LogReadSafeStringLong() ReadSafeStringLong();
|
||
|
#define LogSkip(deltaByteCount, desc) Skip(deltaByteCount)
|
||
|
#define LogReadSwap(value, desc) ReadSwap(value)
|
||
|
#define LogReadSwapArray(count, values, desc) ReadSwap(count, values)
|
||
|
#define LogSubStreamStart(desc) LogVoidFunc()
|
||
|
#define LogSubStreamPushDesc(desc) LogVoidFunc()
|
||
|
#define LogSubStreamEnd() LogVoidFunc()
|
||
|
#define LogStringString(s) LogVoidFunc()
|
||
|
#endif
|
||
|
|
||
|
class hsStream {
|
||
|
public:
|
||
|
enum {
|
||
|
kEolnCode = '\n',
|
||
|
kComment = '#'
|
||
|
};
|
||
|
enum VDB_Type {// Virtual Database type
|
||
|
kVDB_GroupObject,
|
||
|
kVDB_Mesh
|
||
|
};
|
||
|
protected:
|
||
|
UInt32 fBytesRead;
|
||
|
UInt32 fPosition;
|
||
|
|
||
|
hsBool IsTokenSeparator(char c);
|
||
|
public:
|
||
|
hsStream() : fBytesRead(0), fPosition(0) {}
|
||
|
virtual ~hsStream();
|
||
|
|
||
|
virtual hsBool Open(const char *, const char * = "rb")=0;
|
||
|
virtual hsBool Open(const wchar *, const wchar * = L"rb")=0;
|
||
|
virtual hsBool Close()=0;
|
||
|
virtual hsBool AtEnd();
|
||
|
virtual UInt32 Read(UInt32 byteCount, void * buffer) = 0;
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer) = 0;
|
||
|
virtual void Skip(UInt32 deltaByteCount) = 0;
|
||
|
virtual void Rewind() = 0;
|
||
|
virtual void FastFwd();
|
||
|
virtual UInt32 GetPosition() const;
|
||
|
virtual void SetPosition(UInt32 position);
|
||
|
virtual void Truncate();
|
||
|
virtual void Flush() {}
|
||
|
|
||
|
#ifdef STREAM_LOGGER
|
||
|
// Logging Reads & Skips
|
||
|
virtual UInt32 LogRead(UInt32 byteCount, void * buffer, const char* desc) { return Read(byteCount,buffer); }
|
||
|
virtual char* LogReadSafeString() { return ReadSafeString(); }
|
||
|
virtual char* LogReadSafeStringLong() { return ReadSafeStringLong(); }
|
||
|
virtual void LogSkip(UInt32 deltaByteCount, const char* desc) { Skip(deltaByteCount); }
|
||
|
|
||
|
// Stream Notes for Logging
|
||
|
virtual void LogStringString(const char* s) { }
|
||
|
virtual void LogSubStreamStart(const char* desc) { }
|
||
|
virtual void LogSubStreamEnd() { }
|
||
|
virtual void LogSubStreamPushDesc(const char* desc) { }
|
||
|
#endif
|
||
|
void LogVoidFunc() { }
|
||
|
|
||
|
// Optimization for small Reads
|
||
|
virtual UInt8 ReadByte();
|
||
|
virtual hsBool Read4Bytes(void *buffer); // Reads 4 bytes, return true if success
|
||
|
virtual hsBool Read8Bytes(void *buffer); // Reads 8 bytes, return true if success
|
||
|
virtual hsBool Read12Bytes(void *buffer); // Reads 12 bytes, return true if success
|
||
|
|
||
|
virtual UInt32 GetEOF();
|
||
|
UInt32 GetSizeLeft();
|
||
|
virtual void CopyToMem(void* mem);
|
||
|
virtual hsBool IsCompressed() { return false; }
|
||
|
|
||
|
UInt32 WriteString(const char cstring[]);
|
||
|
UInt32 WriteFmt(const char * fmt, ...);
|
||
|
UInt32 WriteFmtV(const char * fmt, va_list av);
|
||
|
|
||
|
UInt32 WriteSafeStringLong(const char *string); // uses 4 bytes for length
|
||
|
UInt32 WriteSafeWStringLong(const wchar_t *string);
|
||
|
char * ReadSafeStringLong();
|
||
|
wchar_t * ReadSafeWStringLong();
|
||
|
|
||
|
UInt32 WriteSafeString(const char *string); // uses 2 bytes for length
|
||
|
UInt32 WriteSafeWString(const wchar_t *string);
|
||
|
char * ReadSafeString();
|
||
|
wchar_t * ReadSafeWString();
|
||
|
|
||
|
hsBool GetToken(char *s, UInt32 maxLen=UInt32(-1), const char beginComment=kComment, const char endComment=kEolnCode);
|
||
|
hsBool ReadLn(char* s, UInt32 maxLen=UInt32(-1), const char beginComment=kComment, const char endComment=kEolnCode);
|
||
|
|
||
|
bool Readbool();
|
||
|
hsBool ReadBool();
|
||
|
void ReadBool(int count, hsBool values[]);
|
||
|
UInt16 ReadSwap16();
|
||
|
void ReadSwap16(int count, UInt16 values[]);
|
||
|
UInt32 ReadSwap32();
|
||
|
void ReadSwap32(int count, UInt32 values[]);
|
||
|
UInt32 ReadUnswap32();
|
||
|
|
||
|
void Writebool(bool value);
|
||
|
void WriteBool(hsBool value);
|
||
|
void WriteBool(int count, const hsBool values[]);
|
||
|
void WriteByte(UInt8 value);
|
||
|
void WriteSwap16(UInt16 value);
|
||
|
void WriteSwap16(int count, const UInt16 values[]);
|
||
|
void WriteSwap32(UInt32 value);
|
||
|
void WriteSwap32(int count, const UInt32 values[]);
|
||
|
void WriteUnswap32(UInt32 value);
|
||
|
|
||
|
|
||
|
/* Overloaded Begin (8 & 16 & 32 int)*/
|
||
|
/* yes, swapping an 8 bit value does nothing, just useful*/
|
||
|
void ReadSwap(bool* value) { *value = this->ReadByte() ? true : false; }
|
||
|
void ReadSwap(UInt8* value) { *value = this->ReadByte(); }
|
||
|
void ReadSwap(int count, UInt8 values[]) { this->Read(count, values); }
|
||
|
void ReadSwap(UInt16* value) { *value = this->ReadSwap16(); }
|
||
|
void ReadSwap(int count, UInt16 values[]) { this->ReadSwap16(count, values); }
|
||
|
void ReadSwap(UInt32* value) { *value = this->ReadSwap32(); }
|
||
|
void ReadSwap(int count, UInt32 values[]) { this->ReadSwap32(count, values); }
|
||
|
#ifdef STREAM_LOGGER
|
||
|
// Begin LogReadSwaps
|
||
|
virtual void LogReadSwap(bool* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwap(UInt8* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, UInt8 values[], const char* desc) { this->ReadSwap(count, values); }
|
||
|
virtual void LogReadSwap(UInt16* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, UInt16 values[], const char* desc) { this->ReadSwap(count, values); }
|
||
|
virtual void LogReadSwap(UInt32* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, UInt32 values[], const char* desc) { this->ReadSwap(count, values); }
|
||
|
// End LogReadSwaps
|
||
|
#endif
|
||
|
void WriteSwap(bool value) { this->Write(1,&value); }
|
||
|
void WriteSwap(UInt8 value) { this->Write(1,&value); }
|
||
|
void WriteSwap(int count, const UInt8 values[]) { this->Write(count, values); }
|
||
|
void WriteSwap(UInt16 value) { this->WriteSwap16(value); }
|
||
|
void WriteSwap(int count, const UInt16 values[]) { this->WriteSwap16(count, values); }
|
||
|
void WriteSwap(UInt32 value) { this->WriteSwap32(value); }
|
||
|
void WriteSwap(int count, const UInt32 values[]) { this->WriteSwap32(count, values); }
|
||
|
void ReadSwap(Int8* value) { *value = this->ReadByte(); }
|
||
|
void ReadSwap(int count, Int8 values[]) { this->Read(count, values); }
|
||
|
void ReadSwap(char* value) { *value = (char)this->ReadByte(); }
|
||
|
void ReadSwap(int count, char values[]) { this->Read(count, values); }
|
||
|
void ReadSwap(Int16* value) { *value = (Int16)this->ReadSwap16(); }
|
||
|
void ReadSwap(int count, Int16 values[]) { this->ReadSwap16(count, (UInt16*)values); }
|
||
|
void ReadSwap(Int32* value) { *value = (Int32)this->ReadSwap32(); }
|
||
|
void ReadSwap(int count, Int32 values[]) { this->ReadSwap32(count, (UInt32*)values); }
|
||
|
void ReadSwap(int* value) { *value = (Int32)this->ReadSwap32(); }
|
||
|
void ReadSwap(int count, int values[]) { this->ReadSwap32(count, (UInt32*)values); }
|
||
|
#ifdef STREAM_LOGGER
|
||
|
// Begin LogReadSwaps
|
||
|
virtual void LogReadSwap(Int8* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, Int8 values[], const char* desc) { this->ReadSwap(count, values); }
|
||
|
virtual void LogReadSwap(char* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, char values[], const char* desc) { this->ReadSwap(count, values); }
|
||
|
virtual void LogReadSwap(Int16* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, Int16 values[], const char* desc) { this->ReadSwap(count, (UInt16*)values); }
|
||
|
virtual void LogReadSwap(Int32* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, Int32 values[], const char* desc) { this->ReadSwap(count, (UInt32*)values); }
|
||
|
virtual void LogReadSwap(int* value, const char* desc) { this->ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, int values[], const char* desc) { this->ReadSwap(count, (UInt32*)values); }
|
||
|
// End LogReadSwaps
|
||
|
#endif
|
||
|
void WriteSwap(Int8 value) { this->Write(1,&value); }
|
||
|
void WriteSwap(int count, const Int8 values[]) { this->Write(count, values); }
|
||
|
void WriteSwap(char value) { this->Write(1,(UInt8*)&value); }
|
||
|
void WriteSwap(int count, const char values[]) { this->Write(count, (UInt8*)values); }
|
||
|
void WriteSwap(Int16 value) { this->WriteSwap16((UInt16)value); }
|
||
|
void WriteSwap(int count, const Int16 values[]) { this->WriteSwap16(count, (UInt16*)values); }
|
||
|
void WriteSwap(Int32 value) { this->WriteSwap32((UInt32)value); }
|
||
|
void WriteSwap(int count, const Int32 values[]) { this->WriteSwap32(count, (UInt32*)values); }
|
||
|
void WriteSwap(int value) { this->WriteSwap32((UInt32)value); }
|
||
|
void WriteSwap(int count, const int values[]) { this->WriteSwap32(count, (UInt32*)values); }
|
||
|
/* Overloaded End */
|
||
|
|
||
|
|
||
|
#if HS_CAN_USE_FLOAT
|
||
|
float ReadSwapFloat();
|
||
|
void ReadSwapFloat(int count, float values[]);
|
||
|
double ReadSwapDouble();
|
||
|
void ReadSwapDouble(int count, double values[]);
|
||
|
float ReadUnswapFloat();
|
||
|
void WriteSwapFloat(float value);
|
||
|
void WriteSwapFloat(int count, const float values[]);
|
||
|
void WriteSwapDouble(double value);
|
||
|
void WriteSwapDouble(int count, const double values[]);
|
||
|
void WriteUnswapFloat(float value);
|
||
|
|
||
|
|
||
|
/* Overloaded Begin (Float)*/
|
||
|
void ReadSwap(float* value) { *value = ReadSwapFloat(); }
|
||
|
void ReadSwap(int count, float values[]) { ReadSwapFloat(count, values); }
|
||
|
void ReadSwap(double* value) { *value = ReadSwapDouble(); }
|
||
|
void ReadSwap(int count, double values[]) { ReadSwapDouble(count, values); }
|
||
|
#ifdef STREAM_LOGGER
|
||
|
// Begin LogReadSwaps
|
||
|
virtual void LogReadSwap(float* value, const char* desc) { ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, float values[], const char* desc) { ReadSwap(count, values); }
|
||
|
virtual void LogReadSwap(double* value, const char* desc) { ReadSwap(value); }
|
||
|
virtual void LogReadSwapArray(int count, double values[], const char* desc) { ReadSwap(count, values); }
|
||
|
// End LogReadSwaps
|
||
|
#endif
|
||
|
void WriteSwap(float value) { WriteSwapFloat(value); }
|
||
|
void WriteSwap(int count, const float values[]) { WriteSwapFloat(count, values); }
|
||
|
void WriteSwap(double value) { WriteSwapDouble(value); }
|
||
|
void WriteSwap(int count, const double values[]) { WriteSwapDouble(count, values); }
|
||
|
/* Overloaded End */
|
||
|
#endif
|
||
|
|
||
|
#if HS_SCALAR_IS_FIXED
|
||
|
hsFixed ReadSwapScalar() { return (hsFixed)this->ReadSwap32(); }
|
||
|
void ReadSwapScalar(int count, hsFixed values[])
|
||
|
{
|
||
|
this->ReadSwap32(count, (UInt32*)values);
|
||
|
}
|
||
|
hsFixed ReadUnswapScalar() { return (hsFixed)this->ReadUnswap32(); }
|
||
|
|
||
|
|
||
|
void WriteSwapScalar(hsFixed value) { this->WriteSwap32(value); }
|
||
|
void WriteSwapScalar(int count, const hsFixed values[])
|
||
|
{
|
||
|
this->WriteSwap32(count, (UInt32*)values);
|
||
|
}
|
||
|
void WriteUnswapScalar(hsFixed value) { this->WriteUnswap32(value); }
|
||
|
|
||
|
|
||
|
/* Overloaded Begin (Scalar) */
|
||
|
void ReadSwap(hsFixed* value) { this->ReadSwap((UInt32*)value); }
|
||
|
void ReadSwap(int count, hsFixed values[]) { this->ReadSwap(count, (UInt32*)values); }
|
||
|
void WriteSwap(hsFixed value) { this->WriteSwap((UInt32)value); }
|
||
|
void WriteSwap(int count, const hsFixed values[]) { this->WriteSwap(count, (UInt32*)values); }
|
||
|
/* Overloaded End */
|
||
|
|
||
|
#else
|
||
|
float ReadSwapScalar() { return (float)this->ReadSwapFloat(); }
|
||
|
void ReadSwapScalar(int count, float values[])
|
||
|
{
|
||
|
this->ReadSwapFloat(count, (float*)values);
|
||
|
}
|
||
|
float ReadUnswapScalar() { return (float)this->ReadUnswapFloat(); }
|
||
|
void WriteSwapScalar(float value) { this->WriteSwapFloat(value); }
|
||
|
void WriteSwapScalar(int count, const float values[])
|
||
|
{
|
||
|
this->WriteSwapFloat(count, (float*)values);
|
||
|
}
|
||
|
void WriteUnswapScalar(float value) { this->WriteUnswapFloat(value); }
|
||
|
#endif
|
||
|
|
||
|
void WriteSwapAtom(UInt32 tag, UInt32 size);
|
||
|
UInt32 ReadSwapAtom(UInt32* size);
|
||
|
|
||
|
|
||
|
/* Overloaded Begin (Atom)*/
|
||
|
void WriteSwap(UInt32* tag, UInt32 size) { WriteSwapAtom(*tag, size); }
|
||
|
void ReadSwap(UInt32* tag, UInt32 *size) { *tag = ReadSwapAtom(size); }
|
||
|
/* Overloaded End */
|
||
|
virtual void VirtualSetPosition(UInt32 pos, VDB_Type ){ SetPosition(pos); };
|
||
|
virtual hsPackFileSys::FileEntry *GetFileEntry() { return nil; } // Streams from Packfiles can return a FileEntry
|
||
|
|
||
|
};
|
||
|
|
||
|
class hsStreamable {
|
||
|
public:
|
||
|
virtual void Read(hsStream* stream) = 0;
|
||
|
virtual void Write(hsStream* stream) = 0;
|
||
|
virtual UInt32 GetStreamSize() = 0;
|
||
|
};
|
||
|
|
||
|
class hsFileStream: public hsStream
|
||
|
{
|
||
|
UInt32 fRef;
|
||
|
#if HS_BUILD_FOR_PS2
|
||
|
enum {
|
||
|
kBufferSize = 2*1024
|
||
|
};
|
||
|
UInt32 fFileSize;
|
||
|
char fBuffer[kBufferSize];
|
||
|
Int32 fVirtualFilePointer;
|
||
|
Int32 fBufferBase; // offset to top of fBuffer
|
||
|
hsBool fBufferIsEmpty;
|
||
|
hsBool fWriteBufferUsed; // In write mode. fBuffer must be flush, when file was closed.
|
||
|
#endif
|
||
|
|
||
|
public:
|
||
|
hsFileStream();
|
||
|
virtual ~hsFileStream();
|
||
|
virtual hsBool Open(const char *name, const char *mode = "rb");
|
||
|
virtual hsBool Open(const wchar *name, const wchar *mode = L"rb");
|
||
|
virtual hsBool Close();
|
||
|
|
||
|
virtual hsBool AtEnd();
|
||
|
virtual UInt32 Read(UInt32 byteCount, void* buffer);
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer);
|
||
|
virtual void Skip(UInt32 deltaByteCount);
|
||
|
virtual void Rewind();
|
||
|
virtual void Truncate();
|
||
|
|
||
|
virtual UInt32 GetFileRef();
|
||
|
virtual void SetFileRef(UInt32 refNum);
|
||
|
};
|
||
|
|
||
|
#if !HS_BUILD_FOR_PS2
|
||
|
#if !(HS_BUILD_FOR_REFERENCE)
|
||
|
|
||
|
class hsUNIXStream: public hsStream
|
||
|
{
|
||
|
FILE* fRef;
|
||
|
char* fBuff;
|
||
|
|
||
|
public:
|
||
|
hsUNIXStream(): fRef(0), fBuff(nil) {}
|
||
|
~hsUNIXStream();
|
||
|
virtual hsBool Open(const char* name, const char* mode = "rb");
|
||
|
virtual hsBool Open(const wchar *name, const wchar *mode = L"rb");
|
||
|
virtual hsBool Close();
|
||
|
|
||
|
virtual hsBool AtEnd();
|
||
|
virtual UInt32 Read(UInt32 byteCount, void* buffer);
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer);
|
||
|
virtual void SetPosition(UInt32 position);
|
||
|
virtual void Skip(UInt32 deltaByteCount);
|
||
|
virtual void Rewind();
|
||
|
virtual void FastFwd();
|
||
|
virtual void Truncate();
|
||
|
virtual void Flush();
|
||
|
|
||
|
FILE* GetFILE() { return fRef; }
|
||
|
void SetFILE(FILE* file) { fRef = file; }
|
||
|
|
||
|
virtual UInt32 GetEOF();
|
||
|
};
|
||
|
|
||
|
// Small substream class: give it a base stream, an offset and a length, and it'll
|
||
|
// treat all ops as if you had a chunk from the base stream as a separate, vanilla
|
||
|
// stream of the given length.
|
||
|
|
||
|
class plReadOnlySubStream: public hsStream
|
||
|
{
|
||
|
hsStream *fBase;
|
||
|
UInt32 fOffset, fLength;
|
||
|
|
||
|
void IFixPosition( void );
|
||
|
|
||
|
public:
|
||
|
plReadOnlySubStream(): fBase( nil ), fOffset( 0 ), fLength( 0 ) {}
|
||
|
~plReadOnlySubStream();
|
||
|
|
||
|
virtual hsBool Open(const char *, const char *) { hsAssert(0, "plReadOnlySubStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Open(const wchar *, const wchar *) { hsAssert(0, "plReadOnlySubStream::Open NotImplemented"); return false; }
|
||
|
void Open( hsStream *base, UInt32 offset, UInt32 length );
|
||
|
virtual hsBool Close() { fBase = nil; fOffset = 0; fLength = 0; return true; }
|
||
|
virtual hsBool AtEnd();
|
||
|
virtual UInt32 Read(UInt32 byteCount, void* buffer);
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer);
|
||
|
virtual void Skip(UInt32 deltaByteCount);
|
||
|
virtual void Rewind();
|
||
|
virtual void FastFwd();
|
||
|
virtual void Truncate();
|
||
|
|
||
|
virtual UInt32 GetEOF();
|
||
|
};
|
||
|
|
||
|
#endif
|
||
|
#endif
|
||
|
|
||
|
class hsRAMStream : public hsStream {
|
||
|
hsAppender fAppender;
|
||
|
hsAppenderIterator fIter;
|
||
|
public:
|
||
|
hsRAMStream();
|
||
|
hsRAMStream(UInt32 chunkSize);
|
||
|
virtual ~hsRAMStream();
|
||
|
|
||
|
virtual hsBool Open(const char *, const char *) { hsAssert(0, "hsRAMStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Open(const wchar *, const wchar *) { hsAssert(0, "hsRAMStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Close() { hsAssert(0, "hsRAMStream::Close NotImplemented"); return false; }
|
||
|
|
||
|
|
||
|
virtual hsBool AtEnd();
|
||
|
virtual UInt32 Read(UInt32 byteCount, void * buffer);
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer);
|
||
|
virtual void Skip(UInt32 deltaByteCount);
|
||
|
virtual void Rewind();
|
||
|
virtual void Truncate();
|
||
|
|
||
|
virtual UInt32 GetEOF();
|
||
|
virtual void CopyToMem(void* mem);
|
||
|
|
||
|
void Reset(); // clears the buffers
|
||
|
};
|
||
|
|
||
|
class hsNullStream : public hsStream {
|
||
|
public:
|
||
|
|
||
|
virtual hsBool Open(const char *, const char *) { return true; }
|
||
|
virtual hsBool Open(const wchar *, const wchar *) { return true; }
|
||
|
virtual hsBool Close() { return true; }
|
||
|
|
||
|
virtual UInt32 Read(UInt32 byteCount, void * buffer); // throw's exception
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer);
|
||
|
virtual void Skip(UInt32 deltaByteCount);
|
||
|
virtual void Rewind();
|
||
|
virtual void Truncate();
|
||
|
|
||
|
UInt32 GetBytesWritten() const { return fBytesRead; }
|
||
|
void Reset( ) { fBytesRead = 0; }
|
||
|
};
|
||
|
|
||
|
// read only mem stream
|
||
|
class hsReadOnlyStream : public hsStream {
|
||
|
protected:
|
||
|
char* fStart;
|
||
|
char* fData;
|
||
|
char* fStop;
|
||
|
public:
|
||
|
hsReadOnlyStream(int size, const void* data) { Init(size, data); }
|
||
|
hsReadOnlyStream() {}
|
||
|
|
||
|
virtual void Init(int size, const void* data) { fStart=((char*)data); fData=((char*)data); fStop=((char*)data + size); }
|
||
|
virtual hsBool Open(const char *, const char *) { hsAssert(0, "hsReadOnlyStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Open(const wchar *, const wchar *) { hsAssert(0, "hsReadOnlyStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Close() { hsAssert(0, "hsReadOnlyStream::Close NotImplemented"); return false; }
|
||
|
virtual hsBool AtEnd();
|
||
|
virtual UInt32 Read(UInt32 byteCount, void * buffer);
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer); // throws exception
|
||
|
virtual void Skip(UInt32 deltaByteCount);
|
||
|
virtual void Rewind();
|
||
|
virtual void Truncate();
|
||
|
virtual UInt32 GetBytesRead() const { return fBytesRead; }
|
||
|
virtual UInt32 GetEOF() { return (UInt32)(fStop-fStart); }
|
||
|
virtual void CopyToMem(void* mem);
|
||
|
};
|
||
|
|
||
|
// write only mem stream
|
||
|
class hsWriteOnlyStream : public hsReadOnlyStream {
|
||
|
public:
|
||
|
hsWriteOnlyStream(int size, const void* data) : hsReadOnlyStream(size, data) {}
|
||
|
hsWriteOnlyStream() {}
|
||
|
|
||
|
virtual hsBool Open(const char *, const char *) { hsAssert(0, "hsWriteOnlyStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Open(const wchar *, const wchar *) { hsAssert(0, "hsWriteOnlyStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Close() { hsAssert(0, "hsWriteOnlyStream::Close NotImplemented"); return false; }
|
||
|
virtual UInt32 Read(UInt32 byteCount, void * buffer); // throws exception
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer);
|
||
|
virtual UInt32 GetBytesRead() const { return 0; }
|
||
|
virtual UInt32 GetBytesWritten() const { return fBytesRead; }
|
||
|
};
|
||
|
|
||
|
// circular queue stream
|
||
|
class hsQueueStream : public hsStream {
|
||
|
private:
|
||
|
char* fQueue;
|
||
|
UInt32 fReadCursor;
|
||
|
UInt32 fWriteCursor;
|
||
|
UInt32 fSize;
|
||
|
|
||
|
public:
|
||
|
hsQueueStream(Int32 size);
|
||
|
~hsQueueStream();
|
||
|
|
||
|
virtual hsBool Open(const char *, const char *) { hsAssert(0, "hsQueueStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Open(const wchar *, const wchar *) { hsAssert(0, "hsQueueStream::Open NotImplemented"); return false; }
|
||
|
virtual hsBool Close() { hsAssert(0, "hsQueueStream::Close NotImplemented"); return false; }
|
||
|
|
||
|
virtual UInt32 Read(UInt32 byteCount, void * buffer);
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer);
|
||
|
virtual void Skip(UInt32 deltaByteCount);
|
||
|
virtual void Rewind();
|
||
|
virtual void FastFwd();
|
||
|
virtual hsBool AtEnd();
|
||
|
|
||
|
UInt32 GetSize() { return fSize; }
|
||
|
const char* GetQueue() { return fQueue; }
|
||
|
UInt32 GetReadCursor() { return fReadCursor; }
|
||
|
UInt32 GetWriteCursor() { return fWriteCursor; }
|
||
|
};
|
||
|
|
||
|
class hsBufferedStream : public hsStream
|
||
|
{
|
||
|
FILE* fRef;
|
||
|
UInt32 fFileSize;
|
||
|
|
||
|
enum { kBufferSize = 2*1024 };
|
||
|
char fBuffer[kBufferSize];
|
||
|
// If the buffer is empty, this is zero. Otherwise it is the size of the
|
||
|
// buffer (if we read a full block), or something less than that if we read
|
||
|
// a partial block at the end of the file.
|
||
|
UInt32 fBufferLen;
|
||
|
|
||
|
hsBool fWriteBufferUsed;
|
||
|
|
||
|
#ifdef HS_DEBUGGING
|
||
|
// For doing statistics on how efficient we are
|
||
|
int fBufferHits, fBufferMisses;
|
||
|
UInt32 fBufferReadIn, fBufferReadOut, fReadDirect, fLastReadPos;
|
||
|
char* fFilename;
|
||
|
const char* fCloseReason;
|
||
|
#endif
|
||
|
|
||
|
public:
|
||
|
hsBufferedStream();
|
||
|
virtual ~hsBufferedStream();
|
||
|
|
||
|
virtual hsBool Open(const char* name, const char* mode = "rb");
|
||
|
virtual hsBool Open(const wchar* name, const wchar* mode = L"rb");
|
||
|
virtual hsBool Close();
|
||
|
|
||
|
virtual hsBool AtEnd();
|
||
|
virtual UInt32 Read(UInt32 byteCount, void* buffer);
|
||
|
virtual UInt32 Write(UInt32 byteCount, const void* buffer);
|
||
|
virtual void Skip(UInt32 deltaByteCount);
|
||
|
virtual void Rewind();
|
||
|
virtual void Truncate();
|
||
|
virtual UInt32 GetEOF();
|
||
|
|
||
|
FILE* GetFileRef();
|
||
|
void SetFileRef(FILE* file);
|
||
|
|
||
|
// Something optional for when we're doing stats. Will log the reason why
|
||
|
// the file was closed. Really just for plRegistryPageNode.
|
||
|
void SetCloseReason(const char* reason)
|
||
|
{
|
||
|
#ifdef HS_DEBUGGING
|
||
|
fCloseReason = reason;
|
||
|
#endif
|
||
|
}
|
||
|
};
|
||
|
|
||
|
#endif
|