BufferReader.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreTypes.h"
#include "Misc/AssertionMacros.h"
#include "HAL/UnrealMemory.h"
#include "Serialization/Archive.h"
#include "Containers/UnrealString.h"
 
class FBufferReaderBase : public FArchive
{
public:
    FBufferReaderBase(void* Data, int64 Size, bool bInFreeOnClose, bool bIsPersistent = false)
        : ReaderData(Data)
        , ReaderPos(0)
        , ReaderSize(Size)
        , bFreeOnClose(bInFreeOnClose)
    {
        this->SetIsLoading(true);
        this->SetIsPersistent(bIsPersistent);
    }
 
    // Abstract to force implementors to call Close() themselves, since it's a virtual function.
    virtual ~FBufferReaderBase() = 0;
 
    virtual bool Close() override
    {
        if (bFreeOnClose)
        {
            FMemory::Free(ReaderData);
            ReaderData = nullptr;
        }
        return !IsError();
    }
    void Serialize(void* Data, int64 Num) final
    {
        check(ReaderPos >= 0);
        check(ReaderPos + Num <= ReaderSize);
        FMemory::Memcpy(Data, (uint8*)ReaderData + ReaderPos, Num);
        ReaderPos += Num;
    }
    int64 Tell() final
    {
        return ReaderPos;
    }
    int64 TotalSize() final
    {
        return ReaderSize;
    }
    void Seek(int64 InPos) final 
    {
        check(InPos >= 0);
        check(InPos <= ReaderSize);
        ReaderPos = InPos;
    }
    bool AtEnd() final 
    {
        return ReaderPos >= ReaderSize;
    }
    virtual FString GetArchiveName() const { return TEXT("FBufferReaderBase"); }
protected:
    void*       ReaderData;
    int64       ReaderPos;
    int64       ReaderSize;
    bool    bFreeOnClose;
};
 
inline FBufferReaderBase::~FBufferReaderBase()
{
}
 
class FBufferReader final : public FBufferReaderBase
{
public:
    FBufferReader( void* Data, int64 Size, bool bInFreeOnClose, bool bIsPersistent = false )
        : FBufferReaderBase(Data, Size, bInFreeOnClose, bIsPersistent)
    {
    }
 
    virtual ~FBufferReader() override
    {
        Close();
    }
 
    virtual FString GetArchiveName() const { return TEXT("FBufferReader"); }
};