431 lines
23 KiB
C++
431 lines
23 KiB
C++
/*
|
|
Copyright (c) 2013 Khaled Mammou - Advanced Micro Devices, Inc.
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
*/
|
|
|
|
|
|
#pragma once
|
|
#ifndef O3DGC_BINARY_STREAM_H
|
|
#define O3DGC_BINARY_STREAM_H
|
|
|
|
#include "o3dgcCommon.h"
|
|
#include "o3dgcVector.h"
|
|
|
|
namespace o3dgc
|
|
{
|
|
const unsigned long O3DGC_BINARY_STREAM_DEFAULT_SIZE = 4096;
|
|
const unsigned long O3DGC_BINARY_STREAM_BITS_PER_SYMBOL0 = 7;
|
|
const unsigned long O3DGC_BINARY_STREAM_MAX_SYMBOL0 = (1 << O3DGC_BINARY_STREAM_BITS_PER_SYMBOL0) - 1;
|
|
const unsigned long O3DGC_BINARY_STREAM_BITS_PER_SYMBOL1 = 6;
|
|
const unsigned long O3DGC_BINARY_STREAM_MAX_SYMBOL1 = (1 << O3DGC_BINARY_STREAM_BITS_PER_SYMBOL1) - 1;
|
|
const unsigned long O3DGC_BINARY_STREAM_NUM_SYMBOLS_UINT32 = (32+O3DGC_BINARY_STREAM_BITS_PER_SYMBOL0-1) /
|
|
O3DGC_BINARY_STREAM_BITS_PER_SYMBOL0;
|
|
|
|
//!
|
|
class BinaryStream
|
|
{
|
|
public:
|
|
//! Constructor.
|
|
BinaryStream(unsigned long size = O3DGC_BINARY_STREAM_DEFAULT_SIZE)
|
|
{
|
|
m_endianness = SystemEndianness();
|
|
m_stream.Allocate(size);
|
|
};
|
|
//! Destructor.
|
|
~BinaryStream(void){};
|
|
|
|
void WriteFloat32(float value, O3DGCStreamType streamType)
|
|
{
|
|
if (streamType == O3DGC_STREAM_TYPE_ASCII)
|
|
{
|
|
WriteFloat32ASCII(value);
|
|
}
|
|
else
|
|
{
|
|
WriteFloat32Bin(value);
|
|
}
|
|
}
|
|
void WriteUInt32(unsigned long position, unsigned long value, O3DGCStreamType streamType)
|
|
{
|
|
if (streamType == O3DGC_STREAM_TYPE_ASCII)
|
|
{
|
|
WriteUInt32ASCII(position, value);
|
|
}
|
|
else
|
|
{
|
|
WriteUInt32Bin(position, value);
|
|
}
|
|
}
|
|
void WriteUInt32(unsigned long value, O3DGCStreamType streamType)
|
|
{
|
|
if (streamType == O3DGC_STREAM_TYPE_ASCII)
|
|
{
|
|
WriteUInt32ASCII(value);
|
|
}
|
|
else
|
|
{
|
|
WriteUInt32Bin(value);
|
|
}
|
|
}
|
|
void WriteUChar(unsigned int position, unsigned char value, O3DGCStreamType streamType)
|
|
{
|
|
if (streamType == O3DGC_STREAM_TYPE_ASCII)
|
|
{
|
|
WriteUInt32ASCII(position, value);
|
|
}
|
|
else
|
|
{
|
|
WriteUInt32Bin(position, value);
|
|
}
|
|
}
|
|
void WriteUChar(unsigned char value, O3DGCStreamType streamType)
|
|
{
|
|
if (streamType == O3DGC_STREAM_TYPE_ASCII)
|
|
{
|
|
WriteUCharASCII(value);
|
|
}
|
|
else
|
|
{
|
|
WriteUChar8Bin(value);
|
|
}
|
|
}
|
|
float ReadFloat32(unsigned long & position, O3DGCStreamType streamType) const
|
|
{
|
|
float value;
|
|
if (streamType == O3DGC_STREAM_TYPE_ASCII)
|
|
{
|
|
value = ReadFloat32ASCII(position);
|
|
}
|
|
else
|
|
{
|
|
value = ReadFloat32Bin(position);
|
|
}
|
|
return value;
|
|
}
|
|
unsigned long ReadUInt32(unsigned long & position, O3DGCStreamType streamType) const
|
|
{
|
|
unsigned long value;
|
|
if (streamType == O3DGC_STREAM_TYPE_ASCII)
|
|
{
|
|
value = ReadUInt32ASCII(position);
|
|
}
|
|
else
|
|
{
|
|
value = ReadUInt32Bin(position);
|
|
}
|
|
return value;
|
|
}
|
|
unsigned char ReadUChar(unsigned long & position, O3DGCStreamType streamType) const
|
|
{
|
|
unsigned char value;
|
|
if (streamType == O3DGC_STREAM_TYPE_ASCII)
|
|
{
|
|
value = ReadUCharASCII(position);
|
|
}
|
|
else
|
|
{
|
|
value = ReadUChar8Bin(position);
|
|
}
|
|
return value;
|
|
}
|
|
|
|
void WriteFloat32Bin(unsigned long position, float value)
|
|
{
|
|
assert(position < m_stream.GetSize() - 4);
|
|
unsigned char * ptr = (unsigned char *) (&value);
|
|
if (m_endianness == O3DGC_BIG_ENDIAN)
|
|
{
|
|
m_stream[position++] = ptr[3];
|
|
m_stream[position++] = ptr[2];
|
|
m_stream[position++] = ptr[1];
|
|
m_stream[position ] = ptr[0];
|
|
}
|
|
else
|
|
{
|
|
m_stream[position++] = ptr[0];
|
|
m_stream[position++] = ptr[1];
|
|
m_stream[position++] = ptr[2];
|
|
m_stream[position ] = ptr[3];
|
|
}
|
|
}
|
|
void WriteFloat32Bin(float value)
|
|
{
|
|
unsigned char * ptr = (unsigned char *) (&value);
|
|
if (m_endianness == O3DGC_BIG_ENDIAN)
|
|
{
|
|
m_stream.PushBack(ptr[3]);
|
|
m_stream.PushBack(ptr[2]);
|
|
m_stream.PushBack(ptr[1]);
|
|
m_stream.PushBack(ptr[0]);
|
|
}
|
|
else
|
|
{
|
|
m_stream.PushBack(ptr[0]);
|
|
m_stream.PushBack(ptr[1]);
|
|
m_stream.PushBack(ptr[2]);
|
|
m_stream.PushBack(ptr[3]);
|
|
}
|
|
}
|
|
void WriteUInt32Bin(unsigned long position, unsigned long value)
|
|
{
|
|
assert(position < m_stream.GetSize() - 4);
|
|
unsigned char * ptr = (unsigned char *) (&value);
|
|
if (m_endianness == O3DGC_BIG_ENDIAN)
|
|
{
|
|
m_stream[position++] = ptr[3];
|
|
m_stream[position++] = ptr[2];
|
|
m_stream[position++] = ptr[1];
|
|
m_stream[position ] = ptr[0];
|
|
}
|
|
else
|
|
{
|
|
m_stream[position++] = ptr[0];
|
|
m_stream[position++] = ptr[1];
|
|
m_stream[position++] = ptr[2];
|
|
m_stream[position ] = ptr[3];
|
|
}
|
|
}
|
|
void WriteUInt32Bin(unsigned long value)
|
|
{
|
|
unsigned char * ptr = (unsigned char *) (&value);
|
|
if (m_endianness == O3DGC_BIG_ENDIAN)
|
|
{
|
|
m_stream.PushBack(ptr[3]);
|
|
m_stream.PushBack(ptr[2]);
|
|
m_stream.PushBack(ptr[1]);
|
|
m_stream.PushBack(ptr[0]);
|
|
}
|
|
else
|
|
{
|
|
m_stream.PushBack(ptr[0]);
|
|
m_stream.PushBack(ptr[1]);
|
|
m_stream.PushBack(ptr[2]);
|
|
m_stream.PushBack(ptr[3]);
|
|
}
|
|
}
|
|
void WriteUChar8Bin(unsigned int position, unsigned char value)
|
|
{
|
|
m_stream[position] = value;
|
|
}
|
|
void WriteUChar8Bin(unsigned char value)
|
|
{
|
|
m_stream.PushBack(value);
|
|
}
|
|
float ReadFloat32Bin(unsigned long & position) const
|
|
{
|
|
unsigned long value = ReadUInt32Bin(position);
|
|
float fvalue = *((float *)(&value));
|
|
return fvalue;
|
|
}
|
|
unsigned long ReadUInt32Bin(unsigned long & position) const
|
|
{
|
|
assert(position < m_stream.GetSize() - 4);
|
|
unsigned long value = 0;
|
|
if (m_endianness == O3DGC_BIG_ENDIAN)
|
|
{
|
|
value += (m_stream[position++]<<24);
|
|
value += (m_stream[position++]<<16);
|
|
value += (m_stream[position++]<<8);
|
|
value += (m_stream[position++]);
|
|
}
|
|
else
|
|
{
|
|
value += (m_stream[position++]);
|
|
value += (m_stream[position++]<<8);
|
|
value += (m_stream[position++]<<16);
|
|
value += (m_stream[position++]<<24);
|
|
}
|
|
return value;
|
|
}
|
|
unsigned char ReadUChar8Bin(unsigned long & position) const
|
|
{
|
|
return m_stream[position++];
|
|
}
|
|
|
|
void WriteFloat32ASCII(float value)
|
|
{
|
|
unsigned long uiValue = *((unsigned long *)(&value));
|
|
WriteUInt32ASCII(uiValue);
|
|
}
|
|
void WriteUInt32ASCII(unsigned long position, unsigned long value)
|
|
{
|
|
assert(position < m_stream.GetSize() - O3DGC_BINARY_STREAM_NUM_SYMBOLS_UINT32);
|
|
unsigned long value0 = value;
|
|
for(unsigned long i = 0; i < O3DGC_BINARY_STREAM_NUM_SYMBOLS_UINT32; ++i)
|
|
{
|
|
m_stream[position++] = (value0 & O3DGC_BINARY_STREAM_MAX_SYMBOL0);
|
|
value0 >>= O3DGC_BINARY_STREAM_BITS_PER_SYMBOL0;
|
|
}
|
|
}
|
|
void WriteUInt32ASCII(unsigned long value)
|
|
{
|
|
for(unsigned long i = 0; i < O3DGC_BINARY_STREAM_NUM_SYMBOLS_UINT32; ++i)
|
|
{
|
|
m_stream.PushBack(value & O3DGC_BINARY_STREAM_MAX_SYMBOL0);
|
|
value >>= O3DGC_BINARY_STREAM_BITS_PER_SYMBOL0;
|
|
}
|
|
}
|
|
void WriteIntASCII(long value)
|
|
{
|
|
WriteUIntASCII(IntToUInt(value));
|
|
}
|
|
void WriteUIntASCII(unsigned long value)
|
|
{
|
|
if (value >= O3DGC_BINARY_STREAM_MAX_SYMBOL0)
|
|
{
|
|
m_stream.PushBack(O3DGC_BINARY_STREAM_MAX_SYMBOL0);
|
|
value -= O3DGC_BINARY_STREAM_MAX_SYMBOL0;
|
|
unsigned char a, b;
|
|
do
|
|
{
|
|
a = ((value & O3DGC_BINARY_STREAM_MAX_SYMBOL1) << 1);
|
|
b = ( (value >>= O3DGC_BINARY_STREAM_BITS_PER_SYMBOL1) > 0);
|
|
a += b;
|
|
m_stream.PushBack(a);
|
|
} while (b);
|
|
}
|
|
else
|
|
{
|
|
m_stream.PushBack((unsigned char) value);
|
|
}
|
|
}
|
|
void WriteUCharASCII(unsigned char value)
|
|
{
|
|
assert(value <= O3DGC_BINARY_STREAM_MAX_SYMBOL0);
|
|
m_stream.PushBack(value);
|
|
}
|
|
float ReadFloat32ASCII(unsigned long & position) const
|
|
{
|
|
unsigned long value = ReadUInt32ASCII(position);
|
|
float fvalue = *((float *)(&value));
|
|
return fvalue;
|
|
}
|
|
unsigned long ReadUInt32ASCII(unsigned long & position) const
|
|
{
|
|
assert(position < m_stream.GetSize() - O3DGC_BINARY_STREAM_NUM_SYMBOLS_UINT32);
|
|
unsigned long value = 0;
|
|
unsigned long shift = 0;
|
|
for(unsigned long i = 0; i < O3DGC_BINARY_STREAM_NUM_SYMBOLS_UINT32; ++i)
|
|
{
|
|
value += (m_stream[position++] << shift);
|
|
shift += O3DGC_BINARY_STREAM_BITS_PER_SYMBOL0;
|
|
}
|
|
return value;
|
|
}
|
|
long ReadIntASCII(unsigned long & position) const
|
|
{
|
|
return UIntToInt(ReadUIntASCII(position));
|
|
}
|
|
unsigned long ReadUIntASCII(unsigned long & position) const
|
|
{
|
|
unsigned long value = m_stream[position++];
|
|
if (value == O3DGC_BINARY_STREAM_MAX_SYMBOL0)
|
|
{
|
|
long x;
|
|
unsigned long i = 0;
|
|
do
|
|
{
|
|
x = m_stream[position++];
|
|
value += ( (x>>1) << i);
|
|
i += O3DGC_BINARY_STREAM_BITS_PER_SYMBOL1;
|
|
} while (x & 1);
|
|
}
|
|
return value;
|
|
}
|
|
unsigned char ReadUCharASCII(unsigned long & position) const
|
|
{
|
|
return m_stream[position++];
|
|
}
|
|
O3DGCErrorCode Save(const char * const fileName)
|
|
{
|
|
FILE * fout = fopen(fileName, "wb");
|
|
if (!fout)
|
|
{
|
|
return O3DGC_ERROR_CREATE_FILE;
|
|
}
|
|
fwrite(m_stream.GetBuffer(), 1, m_stream.GetSize(), fout);
|
|
fclose(fout);
|
|
return O3DGC_OK;
|
|
}
|
|
O3DGCErrorCode Load(const char * const fileName)
|
|
{
|
|
FILE * fin = fopen(fileName, "rb");
|
|
if (!fin)
|
|
{
|
|
return O3DGC_ERROR_OPEN_FILE;
|
|
}
|
|
fseek(fin, 0, SEEK_END);
|
|
unsigned long size = ftell(fin);
|
|
m_stream.Allocate(size);
|
|
rewind(fin);
|
|
unsigned int nread = (unsigned int) fread((void *) m_stream.GetBuffer(), 1, size, fin);
|
|
m_stream.SetSize(size);
|
|
if (nread != size)
|
|
{
|
|
return O3DGC_ERROR_READ_FILE;
|
|
}
|
|
fclose(fin);
|
|
return O3DGC_OK;
|
|
}
|
|
O3DGCErrorCode LoadFromBuffer(unsigned char * buffer, unsigned long bufferSize)
|
|
{
|
|
m_stream.Allocate(bufferSize);
|
|
memcpy(m_stream.GetBuffer(), buffer, bufferSize);
|
|
m_stream.SetSize(bufferSize);
|
|
return O3DGC_OK;
|
|
}
|
|
unsigned long GetSize() const
|
|
{
|
|
return m_stream.GetSize();
|
|
}
|
|
const unsigned char * GetBuffer(unsigned long position) const
|
|
{
|
|
return m_stream.GetBuffer() + position;
|
|
}
|
|
unsigned char * GetBuffer(unsigned long position)
|
|
{
|
|
return (m_stream.GetBuffer() + position);
|
|
}
|
|
unsigned char * GetBuffer()
|
|
{
|
|
return m_stream.GetBuffer();
|
|
}
|
|
void GetBuffer(unsigned long position, unsigned char * & buffer) const
|
|
{
|
|
buffer = (unsigned char *) (m_stream.GetBuffer() + position); // fix me: ugly!
|
|
}
|
|
void SetSize(unsigned long size)
|
|
{
|
|
m_stream.SetSize(size);
|
|
};
|
|
void Allocate(unsigned long size)
|
|
{
|
|
m_stream.Allocate(size);
|
|
}
|
|
|
|
private:
|
|
Vector<unsigned char> m_stream;
|
|
O3DGCEndianness m_endianness;
|
|
};
|
|
|
|
}
|
|
#endif // O3DGC_BINARY_STREAM_H
|
|
|