ef2-sdk/dlls/game/archive.cpp
2003-11-05 00:00:00 +00:00

1027 lines
19 KiB
C++

//-----------------------------------------------------------------------------
//
// $Logfile:: /EF2/Code/DLLs/game/archive.cpp $
// $Revision:: 12 $
// $Author:: Singlis $
// $Date:: 9/26/03 2:35p $
//
// Copyright (C) 1997 by Ritual Entertainment, Inc.
// All rights reserved.
//
// This source is may not be distributed and/or modified without
// expressly written permission by Ritual Entertainment, Inc.
//
//
// DESCRIPTION:
// Class for archiving objects
//
#include "_pch_cpp.h"
#include "archive.h"
enum
{
ARC_NULL, ARC_Vector, ARC_Vec3, ARC_Vec4, ARC_Integer, ARC_Unsigned, ARC_Byte, ARC_Char, ARC_Short, ARC_UnsignedShort,
ARC_Float, ARC_Double, ARC_Boolean, ARC_String, ARC_Raw, ARC_Object, ARC_ObjectPointer,
ARC_SafePointer, ARC_Event, ARC_EventPointer, ARC_Quat, ARC_Entity, ARC_Bool,
ARC_NUMTYPES
};
static const char *typenames[] =
{
"NULL", "vector", "vec3", "vec4", "int", "unsigned", "byte", "char", "short", "unsigned short",
"float", "double", "qboolean", "string", "raw data", "object", "objectpointer",
"safepointer", "event", "eventpointer", "quaternion", "entity", "bool"
};
#define ArchiveHeader ( *( int * )"EF2" )
#define ArchiveVersion 2 // This must be changed any time the format changes!
#define ArchiveInfo "EF2 Archive Version 2" // This must be changed any time the format changes!
CLASS_DECLARATION( Class, FileRead, NULL )
{
{ NULL, NULL }
};
FileRead::FileRead()
{
length = 0;
buffer = NULL;
pos = 0;
}
FileRead::~FileRead()
{
Close();
}
void FileRead::Close( void )
{
if ( buffer )
{
gi.Free( ( void * )buffer );
buffer = NULL;
}
filename = "";
length = 0;
pos = 0;
}
const char *FileRead::Filename( void )
{
return filename.c_str();
}
size_t FileRead::Length( void )
{
return length;
}
size_t FileRead::Pos( void )
{
return pos - buffer;
}
qboolean FileRead::Seek( size_t newpos )
{
if ( !buffer )
{
return false;
}
if ( newpos > length )
{
return false;
}
pos = buffer + newpos;
return true;
}
qboolean FileRead::Open( const char *name )
{
byte *tempbuf;
assert( name );
assert( !buffer );
Close();
if ( !name )
{
return false;
}
length = gi.FS_ReadFile( name, ( void ** )&tempbuf, true );
if ( length == ( size_t )( -1 ) )
{
return false;
}
// create our own space
buffer = ( byte * )gi.Malloc( length );
// copy the file over to our space
memcpy( buffer, tempbuf, length );
// free the file
gi.FS_FreeFile( tempbuf );
filename = name;
pos = buffer;
return true;
}
qboolean FileRead::Read( void *dest, size_t size )
{
assert( dest );
assert( buffer );
assert( pos );
if ( !dest )
{
return false;
}
if ( size == 0 )
{
return false;
}
if ( ( pos + size ) > ( buffer + length ) )
{
return false;
}
memcpy( dest, pos, size );
pos += size;
return true;
}
CLASS_DECLARATION( Class, Archiver, NULL )
{
{ NULL, NULL }
};
Archiver::Archiver()
{
file = 0;
fileerror = false;
harderror = true;
archivemode = ARCHIVE_READ;
numclassespos = 0;
assert( ( sizeof( typenames ) / sizeof( typenames[ 0 ] ) ) == ARC_NUMTYPES );
}
Archiver::~Archiver()
{
if ( file )
{
Close();
}
readfile.Close();
}
void Archiver::FileError( const char *fmt, ... )
{
va_list argptr;
char text[ 1024 ];
va_start( argptr, fmt );
vsprintf( text, fmt, argptr );
va_end( argptr );
fileerror = true;
Close();
if ( archivemode == ARCHIVE_READ )
{
if ( harderror )
{
gi.Error( ERR_DROP, "Error while loading %s : %s\n", filename.c_str(), text );
}
else
{
gi.WPrintf( "Error while loading %s : %s\n", filename.c_str(), text );
}
}
else
{
if ( harderror )
{
gi.Error( ERR_DROP, "Error while writing to %s : %s\n", filename.c_str(), text );
}
else
{
gi.WPrintf( "Error while writing to %s : %s\n", filename.c_str(), text );
}
}
}
void Archiver::Close( void )
{
if ( file )
{
if ( archivemode == ARCHIVE_WRITE )
{
int numobjects;
// write out the number of classpointers
gi.FS_FSeek( file, numclassespos, FS_SEEK_SET );
numclassespos = gi.FS_FTell( file );
numobjects = classpointerList.NumObjects();
ArchiveInteger( &numobjects );
}
gi.FS_FCloseFile( file );
file = 0;
}
readfile.Close();
if ( archivemode == ARCHIVE_READ )
{
int i, num;
Class * classptr;
pointer_fixup_t *fixup;
num = fixupList.NumObjects();
for( i = 1; i <= num; i++ )
{
fixup = fixupList.ObjectAt( i );
classptr = classpointerList.ObjectAt( fixup->index );
// Make sure the ptr isn't screwed up
assert( !( ( classptr == NULL ) && ( fixup->index != ARCHIVE_NULL_POINTER ) ) );
if ( fixup->type == pointer_fixup_normal )
{
Class ** fixupptr;
fixupptr = ( Class ** )fixup->ptr;
*fixupptr = classptr;
}
else if ( fixup->type == pointer_fixup_safe )
{
SafePtrBase * fixupptr;
fixupptr = ( SafePtrBase * )fixup->ptr;
fixupptr->InitSafePtr( classptr );
}
delete fixup;
}
fixupList.FreeObjectList();
classpointerList.FreeObjectList();
}
}
/****************************************************************************************
File Read/Write functions
*****************************************************************************************/
qboolean Archiver::Read( const char *name, qboolean file_harderror )
{
unsigned header;
unsigned version;
str info;
int num;
int i;
Class *null;
harderror = file_harderror;
assert( name );
if ( !name )
{
FileError( "NULL pointer for filename in Archiver::Read.\n" );
return false;
}
fileerror = false;
archivemode = ARCHIVE_READ;
filename = name;
if ( !readfile.Open( filename.c_str() ) )
{
if ( file_harderror )
{
FileError( "Couldn't open file." );
}
return false;
}
ArchiveUnsigned( &header );
if ( header != ArchiveHeader )
{
readfile.Close();
FileError( "Not a valid archive." );
return false;
}
ArchiveUnsigned( &version );
if ( version > ArchiveVersion )
{
readfile.Close();
FileError( "Archive is from version %.2f. Check http://www.ritual.com for an update.", version );
return false;
}
if ( version < ArchiveVersion )
{
readfile.Close();
FileError( "Archive is out of date." );
return false;
}
ArchiveString( &info );
gi.DPrintf( "%s\n", info.c_str() );
// setup out class pointers
ArchiveInteger( &num );
classpointerList.Resize( num );
null = NULL;
for( i = 1; i <= num; i++ )
{
classpointerList.AddObject( null );
}
return true;
}
qboolean Archiver::Create( const char *name, qboolean file_harderror )
{
unsigned header;
unsigned version;
str info;
int numZero = 0;
harderror = file_harderror;
assert( name );
if ( !name )
{
FileError( "NULL pointer for filename in Archiver::Create.\n" );
return false;
}
fileerror = false;
archivemode = ARCHIVE_WRITE;
filename = name;
file = gi.FS_FOpenFileWrite( filename.c_str() );
if ( !file )
{
FileError( "Couldn't open file." );
return false;
}
header = ArchiveHeader;
ArchiveUnsigned( &header );
version = ArchiveVersion;
ArchiveUnsigned( &version );
info = ArchiveInfo;
ArchiveString( &info );
numclassespos = gi.FS_FTell( file );
ArchiveInteger( &numZero );
return true;
}
inline void Archiver::CheckRead( void )
{
assert( archivemode == ARCHIVE_READ );
if ( !fileerror && ( archivemode != ARCHIVE_READ ) )
{
FileError( "File read during a write operation." );
}
}
inline void Archiver::CheckWrite( void )
{
assert( archivemode == ARCHIVE_WRITE );
if ( !fileerror && ( archivemode != ARCHIVE_WRITE ) )
{
FileError( "File write during a read operation." );
}
}
inline size_t Archiver::ReadSize( void )
{
size_t s;
s = 0;
if ( !fileerror )
{
readfile.Read( &s, sizeof( s ) );
}
return s;
}
inline void Archiver::CheckSize( int type, size_t size )
{
size_t s;
if ( !fileerror )
{
s = ReadSize();
if ( size != s )
{
FileError( "Invalid data size of %d on %s.", s, typenames[ type ] );
}
}
}
inline void Archiver::WriteSize( size_t size )
{
gi.FS_Write( &size, sizeof( size ), file );
}
inline int Archiver::ReadType( void )
{
int t;
if ( !fileerror )
{
readfile.Read( &t, sizeof( t ) );
return t;
}
return ARC_NULL;
}
inline void Archiver::WriteType( int type )
{
gi.FS_Write( &type, sizeof( type ), file );
}
inline void Archiver::CheckType( int type )
{
int t;
assert( ( type >= 0 ) && ( type < ARC_NUMTYPES ) );
if ( !fileerror )
{
t = ReadType();
if ( t != type )
{
FileError( "Expecting %s", typenames[ type ] );
}
}
}
/****************************************************************************************
File Archive functions
*****************************************************************************************/
//#define ARCHIVE_USE_TYPES 1
inline void Archiver::ArchiveData( int type, void *data, size_t size )
{
if ( archivemode == ARCHIVE_READ )
{
#ifndef NDEBUG
CheckRead();
#endif
#ifdef ARCHIVE_USE_TYPES
CheckType( type );
#endif
if ( !fileerror && size )
{
readfile.Read( data, size );
}
}
else
{
#ifndef NDEBUG
CheckWrite();
#endif
#ifdef ARCHIVE_USE_TYPES
WriteType( type );
#endif
if ( !fileerror && size )
{
gi.FS_Write( data, size, file );
}
}
}
#define ARCHIVE( func, type ) \
void Archiver::Archive##func \
( \
type * v \
) \
\
{ \
ArchiveData( ARC_##func, v, sizeof( type ) ); \
}
ARCHIVE( Vector, Vector );
ARCHIVE( Integer, int );
ARCHIVE( Unsigned, unsigned );
ARCHIVE( Byte, byte );
ARCHIVE( Char, char );
ARCHIVE( Short, short );
ARCHIVE( UnsignedShort, unsigned short );
ARCHIVE( Float, float );
ARCHIVE( Double, double );
ARCHIVE( Boolean, qboolean );
ARCHIVE( Quat, Quat );
ARCHIVE( Bool, bool );
void Archiver::ArchiveVec3( vec3_t vec )
{
ArchiveData( ARC_Vec3, vec, sizeof( vec3_t ) );
}
void Archiver::ArchiveVec4( vec4_t vec )
{
ArchiveData( ARC_Vec4, vec, sizeof( vec4_t ) );
}
void Archiver::ArchiveObjectPointer( Class ** ptr )
{
int index = 0;
if ( archivemode == ARCHIVE_READ )
{
pointer_fixup_t *fixup;
ArchiveData( ARC_ObjectPointer, &index, sizeof( index ) );
// Check for a NULL pointer
assert( ptr );
if ( !ptr )
{
FileError( "NULL pointer in ArchiveObjectPointer." );
}
//
// see if the variable was NULL
//
if ( index == ARCHIVE_NULL_POINTER )
{
*ptr = NULL;
}
else
{
// init the pointer with NULL until we can fix it
*ptr = NULL;
fixup = new pointer_fixup_t;
fixup->ptr = ( void ** )ptr;
fixup->index = index;
fixup->type = pointer_fixup_normal;
fixupList.AddObject( fixup );
}
}
else
{
if ( *ptr )
{
index = classpointerList.AddUniqueObject( *ptr );
}
else
{
index = ARCHIVE_NULL_POINTER;
}
ArchiveData( ARC_ObjectPointer, &index, sizeof( index ) );
}
}
void Archiver::ArchiveSafePointer( SafePtrBase * ptr )
{
int index = 0;
if ( archivemode == ARCHIVE_READ )
{
pointer_fixup_t *fixup;
ArchiveData( ARC_SafePointer, &index, sizeof( &index ) );
// Check for a NULL pointer
assert( ptr );
if ( !ptr )
{
FileError( "NULL pointer in ReadSafePointer." );
}
//
// see if the variable was NULL
//
if ( index == ARCHIVE_NULL_POINTER )
{
ptr->InitSafePtr( NULL );
}
else
{
// init the pointer with NULL until we can fix it
ptr->InitSafePtr( NULL );
// Add new fixup
fixup = new pointer_fixup_t;
fixup->ptr = ( void ** )ptr;
fixup->index = index;
fixup->type = pointer_fixup_safe;
fixupList.AddObject( fixup );
}
}
else
{
if ( ptr->Pointer() )
{
Class * obj;
obj = ptr->Pointer();
index = classpointerList.AddUniqueObject( obj );
}
else
{
index = ARCHIVE_NULL_POINTER;
}
ArchiveData( ARC_SafePointer, &index, sizeof( index ) );
}
}
void Archiver::ArchiveEvent( Event * ev )
{
if ( archivemode == ARCHIVE_READ )
{
#ifndef NDEBUG
CheckRead();
#endif
#ifdef ARCHIVE_USE_TYPES
CheckType( ARC_Event );
#endif
if ( !fileerror )
{
ev->Archive( *this );
}
}
else
{
#ifndef NDEBUG
CheckWrite();
#endif
#ifdef ARCHIVE_USE_TYPES
WriteType( ARC_Event );
#endif
//FIXME!!!! Make this handle null events
if ( ev == NULL )
{
NullEvent.Archive( *this );
}
else
{
ev->Archive( *this );
}
}
}
void Archiver::ArchiveEventPointer( Event ** ev )
{
int index;
if ( archivemode == ARCHIVE_READ )
{
#ifndef NDEBUG
CheckRead();
#endif
#ifdef ARCHIVE_USE_TYPES
CheckType( ARC_EventPointer );
#endif
ArchiveInteger( &index );
if ( !fileerror )
{
if ( index == ARCHIVE_POINTER_VALID )
{
*ev = new Event;
(*ev)->Archive( *this );
}
else
{
(*ev) = NULL;
}
}
}
else
{
#ifndef NDEBUG
CheckWrite();
#endif
if ( *ev )
{
index = ARCHIVE_POINTER_VALID;
}
else
{
index = ARCHIVE_NULL_POINTER;
}
#ifdef ARCHIVE_USE_TYPES
WriteType( ARC_EventPointer );
#endif
ArchiveInteger( &index );
if ( *ev )
{
(*ev)->Archive( *this );
}
}
}
void Archiver::ArchiveRaw( void *data, size_t size )
{
ArchiveData( ARC_Raw, data, size );
}
void Archiver::ArchiveString( str * string )
{
if ( archivemode == ARCHIVE_READ )
{
size_t s;
char *data;
#ifndef NDEBUG
CheckRead();
#endif
#ifdef ARCHIVE_USE_TYPES
CheckType( ARC_String );
#endif
if ( !fileerror )
{
s = ReadSize();
if ( !fileerror )
{
data = new char[ s + 1 ];
if ( data )
{
if ( s )
{
readfile.Read( data, s );
}
data[ s ] = 0;
*string = data;
delete [] data;
}
}
}
}
else
{
#ifndef NDEBUG
CheckWrite();
#endif
#ifdef ARCHIVE_USE_TYPES
WriteType( ARC_String );
#endif
WriteSize( string->length() );
//gi.FS_Write( string->m_data->data, string->length(), file );
gi.FS_Write( string->c_str(), string->length(), file );
}
}
Class * Archiver::ReadObject( void )
{
ClassDef *cls;
Class *obj;
str classname;
long objstart;
long endpos;
int index;
size_t size;
qboolean isent;
int type;
CheckRead();
type = ReadType();
if ( ( type != ARC_Object ) && ( type != ARC_Entity ) )
{
FileError( "Expecting %s or %s", typenames[ ARC_Object ], typenames[ ARC_Entity ] );
}
size = ReadSize();
ArchiveString( &classname );
cls = getClass( classname.c_str() );
if ( !cls )
{
FileError( "Invalid class %s.", classname.c_str() );
}
isent = checkInheritance( &Entity::ClassInfo, cls );
if ( type == ARC_Entity )
{
if ( !isent )
{
FileError( "Non-Entity class object '%s' saved as an Entity based object.", classname.c_str() );
}
ArchiveInteger( &level.spawn_entnum );
//
// make sure to setup spawnflags properly
//
ArchiveInteger( &level.spawnflags );
}
else if ( isent )
{
FileError( "Entity class object '%s' saved as non-Entity based object.", classname.c_str() );
}
ArchiveInteger( &index );
objstart = readfile.Pos();
obj = ( Class * )cls->newInstance();
if ( !obj )
{
FileError( "Failed to on new instance of class %s.", classname.c_str() );
}
else
{
obj->Archive( *this );
}
if ( !fileerror )
{
endpos = readfile.Pos();
if ( ( endpos - objstart ) > size )
{
FileError( "Object read past end of object's data" );
}
else if ( ( endpos - objstart ) < size )
{
FileError( "Object didn't read entire data from file" );
}
}
//
// register this pointer with our list
//
classpointerList.AddObjectAt( index, obj );
return obj;
}
void Archiver::ArchiveObject( Class *obj )
{
str classname;
int index;
size_t size;
qboolean isent;
if ( archivemode == ARCHIVE_READ )
{
ClassDef *cls;
long objstart;
long endpos;
int type;
CheckRead();
type = ReadType();
if ( ( type != ARC_Object ) && ( type != ARC_Entity ) )
{
FileError( "Expecting %s or %s", typenames[ ARC_Object ], typenames[ ARC_Entity ] );
}
size = ReadSize();
ArchiveString( &classname );
cls = getClass( classname.c_str() );
if ( !cls )
{
FileError( "Invalid class %s.", classname.c_str() );
}
if ( obj->classinfo() != cls )
{
FileError( "Archive has a '%s' object, but was expecting a '%s' object.", classname.c_str(), obj->getClassname() );
}
isent = obj->isSubclassOf( Entity );
if ( type == ARC_Entity )
{
int entnum;
if ( !isent )
{
FileError( "Non-Entity class object '%s' saved as an Entity based object.", classname.c_str() );
}
ArchiveInteger( &entnum );
( ( Entity * )obj )->SetEntNum( entnum );
//
// make sure to setup spawnflags properly
//
ArchiveInteger( &level.spawnflags );
}
else if ( isent )
{
FileError( "Entity class object '%s' saved as non-Entity based object.", classname.c_str() );
}
ArchiveInteger( &index );
objstart = readfile.Pos();
obj->Archive( *this );
if ( !fileerror )
{
endpos = readfile.Pos();
if ( ( endpos - objstart ) > size )
{
FileError( "Object read past end of object's data" );
}
else if ( ( endpos - objstart ) < size )
{
FileError( "Object didn't read entire data from file" );
}
}
//
// register this pointer with our list
//
classpointerList.AddObjectAt( index, obj );
}
else
{
long sizepos;
long objstart = 0;
long endpos;
assert( obj );
if ( !obj )
{
FileError( "NULL object in WriteObject" );
}
isent = obj->isSubclassOf( Entity );
CheckWrite();
if ( isent )
{
WriteType( ARC_Entity );
}
else
{
WriteType( ARC_Object );
}
sizepos = gi.FS_FTell( file );
size = 0;
WriteSize( size );
classname = obj->getClassname();
ArchiveString( &classname );
if ( isent )
{
// Write out the entity number
ArchiveInteger( &( ( Entity * )obj )->entnum );
//
// make sure to setup spawnflags properly
//
ArchiveInteger( &( ( Entity * )obj )->spawnflags );
}
// write out pointer index for this class pointer
index = classpointerList.AddUniqueObject( obj );
ArchiveInteger( &index );
if ( !fileerror )
{
objstart = gi.FS_FTell( file );
obj->Archive( *this );
}
if ( !fileerror )
{
endpos = gi.FS_FTell( file );
size = endpos - objstart;
gi.FS_FSeek( file, sizepos, FS_SEEK_SET );
WriteSize( size );
if ( !fileerror )
{
gi.FS_FSeek( file, endpos, FS_SEEK_SET );
}
}
}
}