119 lines
3.2 KiB
C++
119 lines
3.2 KiB
C++
// Copyright (c) 2012-2023 Wojciech Figat. All rights reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "ReadStream.h"
|
|
#include "Engine/Platform/Platform.h"
|
|
|
|
/// <summary>
|
|
/// Super fast advanced data reading from raw bytes without any overhead at all
|
|
/// </summary>
|
|
class FLAXENGINE_API MemoryReadStream : public ReadStream
|
|
{
|
|
private:
|
|
|
|
const byte* _buffer;
|
|
const byte* _position;
|
|
uint32 _length;
|
|
|
|
public:
|
|
|
|
/// <summary>
|
|
/// Init (empty, cannot access before Init())
|
|
/// </summary>
|
|
MemoryReadStream();
|
|
|
|
/// <summary>
|
|
/// Init
|
|
/// </summary>
|
|
/// <param name="bytes">Bytes with data to read from it (no memory cloned, using input buffer)</param>
|
|
/// <param name="length">Amount of bytes</param>
|
|
MemoryReadStream(const byte* bytes, uint32 length);
|
|
|
|
/// <summary>
|
|
/// Init
|
|
/// </summary>
|
|
/// <param name="data">Array with data to read from</param>
|
|
template<typename T, typename AllocationType = HeapAllocation>
|
|
MemoryReadStream(const Array<T, AllocationType>& data)
|
|
: MemoryReadStream(data.Get(), data.Count() * sizeof(T))
|
|
{
|
|
}
|
|
|
|
public:
|
|
|
|
/// <summary>
|
|
/// Init stream to the custom buffer location
|
|
/// </summary>
|
|
/// <param name="bytes">Bytes with data to read from it (no memory cloned, using input buffer)</param>
|
|
/// <param name="length">Amount of bytes</param>
|
|
void Init(const byte* bytes, uint32 length);
|
|
|
|
/// <summary>
|
|
/// Init stream to the custom buffer location
|
|
/// </summary>
|
|
/// <param name="data">Array with data to read from</param>
|
|
template<typename T, typename AllocationType = HeapAllocation>
|
|
FORCE_INLINE void Init(const Array<T, AllocationType>& data)
|
|
{
|
|
Init(data.Get(), data.Count() * sizeof(T));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the current handle to position in buffer.
|
|
/// </summary>
|
|
/// <returns>The position of the buffer in memory.</returns>
|
|
const byte* GetPositionHandle() const
|
|
{
|
|
return _position;
|
|
}
|
|
|
|
public:
|
|
|
|
using ReadStream::Read;
|
|
|
|
/// <summary>
|
|
/// Reads bytes without copying the data.
|
|
/// </summary>
|
|
/// <param name="bytes">The amount of bytes to read.</param>
|
|
/// <returns>The pointer to the data in memory.</returns>
|
|
void* Read(uint32 bytes)
|
|
{
|
|
ASSERT(GetLength() - GetPosition() >= bytes);
|
|
const auto result = (void*)_position;
|
|
_position += bytes;
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Reads given data type from the stream.
|
|
/// </summary>
|
|
/// <returns>The pointer to the data in memory.</returns>
|
|
template<typename T>
|
|
FORCE_INLINE T* Read()
|
|
{
|
|
return static_cast<T*>(Read(sizeof(T)));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Reads array of given data type from the stream.
|
|
/// </summary>
|
|
/// <param name="count">The amount of items to read.</param>
|
|
/// <returns>The pointer to the data in memory.</returns>
|
|
template<typename T>
|
|
FORCE_INLINE T* Read(uint32 count)
|
|
{
|
|
return static_cast<T*>(Read(sizeof(T) * count));
|
|
}
|
|
|
|
public:
|
|
|
|
// [ReadStream]
|
|
void Flush() override;
|
|
void Close() override;
|
|
uint32 GetLength() override;
|
|
uint32 GetPosition() override;
|
|
void SetPosition(uint32 seek) override;
|
|
void ReadBytes(void* data, uint32 bytes) override;
|
|
};
|