mirror of
https://github.com/ENSL/NS.git
synced 2024-11-15 09:21:54 +00:00
106 lines
No EOL
2.2 KiB
C++
106 lines
No EOL
2.2 KiB
C++
#include "MemoryInputStream.h"
|
|
#include <memory.h>
|
|
#include <stdlib.h>
|
|
|
|
MemoryInputStream::MemoryInputStream(void) : m_pData(NULL), m_DataLen(0), m_ReadPos(0) {}
|
|
MemoryInputStream::MemoryInputStream(uchar* pData, int nLength) : m_pData(pData), m_DataLen(nLength), m_ReadPos(0) {}
|
|
|
|
MemoryInputStream::~MemoryInputStream(void) {}
|
|
|
|
void MemoryInputStream::setSource(uchar* pData, int nLength)
|
|
{
|
|
m_pData = pData;
|
|
m_DataLen = nLength;
|
|
m_ReadPos = 0;
|
|
}
|
|
|
|
void MemoryInputStream::seekStart(bool& success)
|
|
{
|
|
m_ReadPos = 0;
|
|
success = (m_pData != NULL);
|
|
}
|
|
|
|
void MemoryInputStream::seekRelative(int count, bool& success)
|
|
{
|
|
int newPos = m_ReadPos + count;
|
|
if( m_pData != NULL && newPos >= 0 && newPos <= m_DataLen )
|
|
{
|
|
m_ReadPos = newPos;
|
|
success = true;
|
|
}
|
|
else
|
|
{
|
|
success = false;
|
|
}
|
|
}
|
|
|
|
void MemoryInputStream::seekEnd(bool& success)
|
|
{
|
|
m_ReadPos = m_DataLen;
|
|
success = (m_pData != NULL);
|
|
}
|
|
|
|
int MemoryInputStream::getAvailable(bool& success)
|
|
{
|
|
success = (m_pData != NULL && m_ReadPos != m_DataLen);
|
|
return m_DataLen - m_ReadPos;
|
|
}
|
|
|
|
uchar MemoryInputStream::readUChar(bool& success)
|
|
{
|
|
if( m_pData != NULL && m_ReadPos != m_DataLen )
|
|
{
|
|
success = true;
|
|
uchar val = m_pData[m_ReadPos];
|
|
++m_ReadPos;
|
|
return val;
|
|
}
|
|
success = false;
|
|
return 0;
|
|
}
|
|
|
|
void MemoryInputStream::readUChar(uchar* buf, int count, bool& success)
|
|
{
|
|
if( m_pData != NULL )
|
|
{
|
|
int diff = m_DataLen - m_ReadPos;
|
|
int numToRead = count < diff ? count : diff;
|
|
memcpy(buf,m_pData,numToRead);
|
|
success = (numToRead == count);
|
|
return;
|
|
}
|
|
success = false;
|
|
}
|
|
|
|
void MemoryInputStream::close(bool& success)
|
|
{
|
|
m_pData = NULL;
|
|
m_DataLen = 0;
|
|
m_ReadPos = 0;
|
|
success = true;
|
|
}
|
|
/*
|
|
#include "vgui_inputstream.h"
|
|
|
|
class MemoryInputStream : public vgui::InputStream
|
|
{
|
|
public:
|
|
MemoryInputStream();
|
|
MemoryInputStream(uchar* pData, int nLength);
|
|
virtual MemoryInputStream(void);
|
|
|
|
virtual void setSource(uchar* pData, int nLength);
|
|
|
|
virtual void seekStart(bool& success);
|
|
virtual void seekRelative(int count, bool& success);
|
|
virtual void seekEnd(bool& success);
|
|
virtual int getAvailable(bool& success);
|
|
virtual uchar readUChar(bool& success);
|
|
virtual void readUChar(uchar* buf, int count, bool& success);
|
|
virtual void close(bool& success);
|
|
private:
|
|
uchar* m_pData;
|
|
int m_DataLen;
|
|
int m_ReadPos;
|
|
};
|
|
*/ |