ObjectPoller.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Net/Core/NetBitArray.h"
#include "Iris/Core/NetChunkedArray.h"
 
// Forward declarations
class UObjectReplicationBridge;
 
namespace UE::Net
{
    class FNetRefHandle;
 
    namespace Private
    {
        typedef uint32 FInternalNetRefIndex;
 
        class FReplicationSystemInternal;
        class FNetRefHandleManager;
        class FNetStatsContext;
    }
}
 
namespace UE::Net::Private
{
 
class FObjectPoller
{
    friend class FReplicationPollTask;
 
public:
 
    struct FPreUpdateAndPollStats
    {
        uint32 PreUpdatedObjectCount = 0;
        uint32 PolledObjectCount = 0;
        uint32 SkippedObjectCount = 0;
        uint32 PolledReferencesObjectCount = 0;
 
        void Accumulate(const FPreUpdateAndPollStats& StatsToAdd)
        {
            PreUpdatedObjectCount += StatsToAdd.PreUpdatedObjectCount;
            PolledObjectCount += StatsToAdd.PolledObjectCount;
            PolledReferencesObjectCount += StatsToAdd.PolledReferencesObjectCount;
        }
    };
 
    struct FInitParams
    {
        FReplicationSystemInternal* ReplicationSystemInternal = nullptr;
        UObjectReplicationBridge* ObjectReplicationBridge = nullptr;
    };
 
public:
 
    FObjectPoller(const FInitParams& InitParams);
 
    const FPreUpdateAndPollStats& GetPollStats() const { return PollStats; }
 
    void PollAndCopyObjects(const FNetBitArrayView& ObjectsConsideredForPolling);
 
    void PollAndCopySingleObject(FInternalNetRefIndex ObjectIndex);
 
private:
 
    void ForcePollObject(FInternalNetRefIndex ObjectIndex, FNetStatsContext* InNetStatsContext, FPreUpdateAndPollStats& InPollStats);
 
    void PushModelPollObject(FInternalNetRefIndex ObjectIndex, FNetStatsContext* InNetStatsContext, FPreUpdateAndPollStats& InPollStats);
 
private:
 
    UObjectReplicationBridge* ObjectReplicationBridge;
    FReplicationSystemInternal* ReplicationSystemInternal;
 
    FNetRefHandleManager& LocalNetRefHandleManager;
    FNetStatsContext* NetStatsContext = nullptr;
    const TNetChunkedArray<TObjectPtr<UObject>>& ReplicatedInstances;
 
    const FNetBitArrayView AccumulatedDirtyObjects;
 
    FNetBitArrayView DirtyObjectsToQuantize;
    FNetBitArrayView DirtyObjectsThisFrame;
    FNetBitArrayView GarbageCollectionAffectedObjects;
 
    FPreUpdateAndPollStats PollStats;
 
    bool bUsePerPropertyDirtyTracking = false;
};
 
} // end namespace UE::Net::Private