You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
42fe03350e
#rb jb #rnx #preflight 6204ce8d000261ff89f8933e [CL 18933162 by Martin Ridgers in ue5-main branch]
126 lines
3.2 KiB
C++
126 lines
3.2 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "CoreTypes.h"
|
|
#include "HAL/UnrealMemory.h"
|
|
#include "Templates/UnrealTemplate.h"
|
|
|
|
namespace UE {
|
|
namespace Trace {
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
class FStreamReader
|
|
{
|
|
public:
|
|
template <typename Type>
|
|
Type const* GetPointer();
|
|
template <typename Type>
|
|
Type const* GetPointerUnchecked();
|
|
const uint8* GetPointer(uint32 Size);
|
|
const uint8* GetPointerUnchecked();
|
|
void Advance(uint32 Size);
|
|
bool IsEmpty() const;
|
|
int32 GetRemaining() const;
|
|
bool CanMeetDemand() const;
|
|
bool Backtrack(const uint8* To);
|
|
struct FMark* SaveMark() const;
|
|
void RestoreMark(struct FMark* Mark);
|
|
|
|
protected:
|
|
uint8* Buffer = nullptr;
|
|
uint32 DemandHint = 0;
|
|
uint32 Cursor = 0;
|
|
uint32 End = 0;
|
|
};
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
template <typename Type>
|
|
Type const* FStreamReader::GetPointer()
|
|
{
|
|
return (Type const*)GetPointer(sizeof(Type));
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
template <typename Type>
|
|
Type const* FStreamReader::GetPointerUnchecked()
|
|
{
|
|
return (Type const*)GetPointerUnchecked();
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
inline const uint8* FStreamReader::GetPointerUnchecked()
|
|
{
|
|
return Buffer + Cursor;
|
|
}
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
class FStreamBuffer
|
|
: public FStreamReader
|
|
{
|
|
public:
|
|
FStreamBuffer() = default;
|
|
FStreamBuffer(uint32_t InitalBufferSize);
|
|
~FStreamBuffer();
|
|
FStreamBuffer(FStreamBuffer&& Rhs) noexcept;
|
|
FStreamBuffer(const FStreamBuffer&) = default;
|
|
FStreamBuffer& operator = (FStreamBuffer&& Rhs) noexcept;
|
|
FStreamBuffer& operator = (const FStreamBuffer&) = delete;
|
|
template <typename Lambda>
|
|
int32 Fill(Lambda&& Source);
|
|
void Append(const uint8* Data, uint32 Size);
|
|
uint8* Append(uint32 Size);
|
|
|
|
protected:
|
|
void Consolidate();
|
|
uint32 BufferSize = 0;
|
|
};
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
inline FStreamBuffer::~FStreamBuffer()
|
|
{
|
|
FMemory::Free(Buffer);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
inline FStreamBuffer::FStreamBuffer(FStreamBuffer&& Rhs) noexcept
|
|
{
|
|
Swap(BufferSize, Rhs.BufferSize);
|
|
Swap(Buffer, Rhs.Buffer);
|
|
Swap(Cursor, Rhs.Cursor);
|
|
Swap(End, Rhs.End);
|
|
Swap(DemandHint, Rhs.DemandHint);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
inline FStreamBuffer& FStreamBuffer::operator = (FStreamBuffer&& Rhs) noexcept
|
|
{
|
|
Swap(BufferSize, Rhs.BufferSize);
|
|
Swap(Buffer, Rhs.Buffer);
|
|
Swap(Cursor, Rhs.Cursor);
|
|
Swap(End, Rhs.End);
|
|
Swap(DemandHint, Rhs.DemandHint);
|
|
return *this;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
template <typename Lambda>
|
|
inline int32 FStreamBuffer::Fill(Lambda&& Source)
|
|
{
|
|
Consolidate();
|
|
|
|
uint8* Dest = Buffer + End;
|
|
int32 ReadSize = Source(Dest, BufferSize - End);
|
|
if (ReadSize > 0)
|
|
{
|
|
End += ReadSize;
|
|
}
|
|
|
|
return ReadSize;
|
|
}
|
|
|
|
} // namespace Trace
|
|
} // namespace UE
|