ChangeMaskCache.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreTypes.h"
#include "Containers/Array.h"
#include "Net/Core/NetBitArray.h"
 
namespace UE::Net::Private
{
 
struct FChangeMaskCache
{
    // Objects that were copied this frame
    struct FCachedInfo
    {
        uint32 InternalIndex;
        uint32 StorageOffset : 24U;
        uint32 bMarkSubObjectOwnerDirty : 1U;
        uint32 bHasDirtyChangeMask : 1;
    };
    TArray<FCachedInfo, TInlineAllocator<1>> Indices;
    TArray<uint32, TInlineAllocator<1>> Storage;
 
    inline void PrepareCache(uint32 IndexCount, uint32 StorageSize);
 
    inline void ResetCache();
    
    inline void EmptyCache();
 
    inline FCachedInfo& AddChangeMaskForObject(uint32 InternalIndex, uint32 BitCount);
 
    inline FCachedInfo& AddSubObjectOwnerDirty(uint32 InternalIndex);
 
    inline FCachedInfo& AddEmptyChangeMaskForObject(uint32 InternalIndex);
 
    inline uint32* GetChangeMaskStorage(const FCachedInfo& Info);
 
    inline void PopLastEntry();
};
 
void FChangeMaskCache::PrepareCache(uint32 IndexCount, uint32 StorageSize)
{
    Indices.Reset(IndexCount);
    Storage.Reset(StorageSize);
}
 
void FChangeMaskCache::ResetCache()
{
    Indices.Reset();
    Storage.Reset();
}
 
void FChangeMaskCache::EmptyCache()
{
    Indices.Empty();
    Storage.Empty();
}
 
FChangeMaskCache::FCachedInfo& FChangeMaskCache::AddChangeMaskForObject(uint32 InternalIndex, uint32 BitCount)
{
    const uint32 StorageIndex = Storage.Num();
    const uint32 WordCount = FNetBitArrayView::CalculateRequiredWordCount(BitCount);
 
    FCachedInfo Info;
    Info.InternalIndex = InternalIndex;
    Info.StorageOffset = StorageIndex;
    Info.bMarkSubObjectOwnerDirty = 0U;
    Info.bHasDirtyChangeMask = 0U;
 
    Storage.AddZeroed(WordCount);
 
    return Indices.Add_GetRef(Info);
}
 
FChangeMaskCache::FCachedInfo& FChangeMaskCache::AddEmptyChangeMaskForObject(uint32 InternalIndex)
{
    FCachedInfo Info;
    Info.InternalIndex = InternalIndex;
    Info.StorageOffset = 0U;
    Info.bMarkSubObjectOwnerDirty = 0U;
    Info.bHasDirtyChangeMask = 0U;
 
    return Indices.Add_GetRef(Info);
}
 
uint32* FChangeMaskCache::GetChangeMaskStorage(const FCachedInfo& Info)
{
    return &Storage.GetData()[Info.StorageOffset];
}
 
void FChangeMaskCache::PopLastEntry()
{
    const FCachedInfo& Info = Indices.Last();
 
    Storage.SetNum(Info.StorageOffset, EAllowShrinking::No);
    Indices.Pop(EAllowShrinking::No);
}
 
inline FChangeMaskCache::FCachedInfo& FChangeMaskCache::AddSubObjectOwnerDirty(uint32 InternalIndex)
{
    FCachedInfo Info;
    Info.InternalIndex = InternalIndex;
    Info.StorageOffset = 0U;
    Info.bMarkSubObjectOwnerDirty = 1U;
    Info.bHasDirtyChangeMask = 0U;
 
    return Indices.Add_GetRef(Info);
}
 
}