ddnet/src/engine/shared/datafile.h

112 lines
2.5 KiB
C
Raw Normal View History

2010-11-20 10:37:14 +00:00
/* (c) Magnus Auvinen. See licence.txt in the root of the distribution for more information. */
/* If you are missing that file, acquire a complete release at teeworlds.com. */
2010-05-29 07:25:38 +00:00
#ifndef ENGINE_SHARED_DATAFILE_H
#define ENGINE_SHARED_DATAFILE_H
2019-12-08 22:14:56 +00:00
#include <engine/storage.h>
#include <base/hash.h>
#include <base/system.h>
2020-10-26 16:05:57 +00:00
#include <zlib.h>
2010-05-29 07:25:38 +00:00
// raw datafile access
class CDataFileReader
{
struct CDatafile *m_pDataFile;
2010-05-29 07:25:38 +00:00
void *GetDataImpl(int Index, int Swap);
int GetFileDataSize(int Index);
int GetExternalItemType(int InternalType);
int GetInternalItemType(int ExternalType);
2010-05-29 07:25:38 +00:00
public:
CDataFileReader() :
m_pDataFile(0) {}
2010-05-29 07:25:38 +00:00
~CDataFileReader() { Close(); }
2010-05-29 07:25:38 +00:00
bool IsOpen() const { return m_pDataFile != 0; }
2010-10-06 21:07:35 +00:00
bool Open(class IStorage *pStorage, const char *pFilename, int StorageType);
2010-05-29 07:25:38 +00:00
bool Close();
2010-05-29 07:25:38 +00:00
void *GetData(int Index);
void *GetDataSwapped(int Index); // makes sure that the data is 32bit LE ints when saved
int GetDataSize(int Index);
void UnloadData(int Index);
void *GetItem(int Index, int *pType, int *pID);
int GetItemSize(int Index) const;
2010-05-29 07:25:38 +00:00
void GetType(int Type, int *pStart, int *pNum);
int FindItemIndex(int Type, int ID);
void *FindItem(int Type, int ID);
int NumItems() const;
int NumData() const;
2010-05-29 07:25:38 +00:00
void Unload();
SHA256_DIGEST Sha256() const;
unsigned Crc() const;
int MapSize() const;
IOHANDLE File();
2010-05-29 07:25:38 +00:00
};
// write access
class CDataFileWriter
{
struct CDataInfo
{
int m_UncompressedSize;
int m_CompressedSize;
void *m_pCompressedData;
};
2010-05-29 07:25:38 +00:00
struct CItemInfo
{
int m_Type;
int m_ID;
2010-05-29 07:25:38 +00:00
int m_Size;
int m_Next;
int m_Prev;
void *m_pData;
};
struct CItemTypeInfo
{
int m_Num;
int m_First;
int m_Last;
};
enum
{
MAX_ITEM_TYPES = 0x10000,
MAX_ITEMS = 1024,
MAX_DATAS = 1024,
MAX_EXTENDED_ITEM_TYPES = 64,
};
2010-05-29 07:25:38 +00:00
IOHANDLE m_File;
int m_NumItems;
int m_NumDatas;
int m_NumItemTypes;
int m_NumExtendedItemTypes;
CItemTypeInfo *m_pItemTypes;
CItemInfo *m_pItems;
CDataInfo *m_pDatas;
int m_aExtendedItemTypes[MAX_EXTENDED_ITEM_TYPES];
int GetExtendedItemTypeIndex(int Type);
2010-05-29 07:25:38 +00:00
public:
CDataFileWriter();
~CDataFileWriter();
void Init();
2019-12-08 22:14:56 +00:00
bool OpenFile(class IStorage *pStorage, const char *pFilename, int StorageType = IStorage::TYPE_SAVE);
2020-10-27 17:57:14 +00:00
bool Open(class IStorage *pStorage, const char *pFilename, int StorageType = IStorage::TYPE_SAVE);
2020-10-26 16:05:57 +00:00
int AddData(int Size, void *pData, int CompressionLevel = Z_DEFAULT_COMPRESSION);
2010-05-29 07:25:38 +00:00
int AddDataSwapped(int Size, void *pData);
int AddItem(int Type, int ID, int Size, void *pData);
2010-05-29 07:25:38 +00:00
int Finish();
};
#endif