SphereWithOwnerBoostNetObjectPrioritizer.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Iris/ReplicationSystem/Prioritization/SphereNetObjectPrioritizer.h"
#include "UObject/ObjectPtr.h"
#include "SphereWithOwnerBoostNetObjectPrioritizer.generated.h"
 
UCLASS(Transient, Config=Engine, MinimalAPI)
class USphereWithOwnerBoostNetObjectPrioritizerConfig : public USphereNetObjectPrioritizerConfig
{
    GENERATED_BODY()
 
public:
    UPROPERTY(Config)
    float OwnerPriorityBoost = 2.0f;
};
 
UCLASS(Transient, MinimalAPI)
class USphereWithOwnerBoostNetObjectPrioritizer : public USphereNetObjectPrioritizer
{
    GENERATED_BODY()
 
protected:
    // UNetObjectPrioritizer interface
    IRISCORE_API virtual void Init(FNetObjectPrioritizerInitParams& Params) override;
    IRISCORE_API virtual void Deinit() override;
    IRISCORE_API virtual bool AddObject(uint32 ObjectIndex, FNetObjectPrioritizerAddObjectParams& Params) override;
    IRISCORE_API virtual void RemoveObject(uint32 ObjectIndex, const FNetObjectPrioritizationInfo& Info) override;
    IRISCORE_API virtual void UpdateObjects(FNetObjectPrioritizerUpdateParams&) override;
    IRISCORE_API virtual void Prioritize(FNetObjectPrioritizationParams&) override;
 
protected:
    using ConnectionId = uint16;
    enum : unsigned
    {
        OwningConnectionsChunkSize = 2U*1024U,
        InvalidConnectionID = 0U,
    };
 
    struct FOwnerBoostBatchParams : public Super::FBatchParams
    {
        uint32* OwnedObjectsLocalIndices;
        uint32 OwnedObjectCount;
        float OwnerPriorityBoost;
    };
 
    void PrepareBatch(FOwnerBoostBatchParams& BatchParams, const FNetObjectPrioritizationParams& PrioritizationParams, uint32 ObjectIndexStartOffset);
    void PrioritizeBatch(FOwnerBoostBatchParams& BatchParams);
    void FinishBatch(const FOwnerBoostBatchParams& BatchParams, FNetObjectPrioritizationParams& PrioritizationParams, uint32 ObjectIndexStartOffset);
    void BoostOwningConnectionPriorities(FOwnerBoostBatchParams& BatchParams) const;
    void SetupBatchParams(FOwnerBoostBatchParams& OutBatchParams, const FNetObjectPrioritizationParams& PrioritizationParams, uint32 MaxBatchObjectCount, FMemStackBase& Mem);
 
    uint32 AllocOwningConnection();
    void FreeOwningConnection(uint32 Index);
 
    uint32 GetOwningConnection(const FObjectLocationInfo& Info) const;
 
    TChunkedArray<ConnectionId, OwningConnectionsChunkSize> OwningConnections;
    UE::Net::FNetBitArray AssignedOwningConnectionIndices;
 
private:
    TObjectPtr<const UReplicationSystem> ReplicationSystem = nullptr;
};
 
inline uint32 USphereWithOwnerBoostNetObjectPrioritizer::GetOwningConnection(const FObjectLocationInfo& Info) const
{
    return OwningConnections[Info.GetLocationIndex()];
}