NetObjectConnectionFilter.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Containers/ContainersFwd.h"
#include "Iris/ReplicationSystem/Filtering/NetObjectFilter.h"
#include "Iris/ReplicationSystem/NetRefHandle.h"
#include "Net/Core/NetBitArray.h"
#include "UObject/StrongObjectPtr.h"
#include "NetObjectConnectionFilter.generated.h"
 
namespace UE::Net::Private
{
    class FNetRefHandleManager;
}
 
UCLASS(transient, MinimalAPI)
class UNetObjectConnectionFilterConfig : public UNetObjectFilterConfig
{
    GENERATED_BODY()
 
public:
    UPROPERTY(Config)
    uint16 MaxObjectCount = 4096;
};
 
UCLASS(transient, MinimalAPI) 
class UNetObjectConnectionFilter : public UNetObjectFilter
{
    GENERATED_BODY()
 
public:
    IRISCORE_API void SetReplicateToConnection(UE::Net::FNetRefHandle RefHandle, uint32 ConnectionId, UE::Net::ENetFilterStatus FilterStatus);
 
protected:
    struct FFilteringInfo : public FNetObjectFilteringInfo
    {
        void SetLocalObjectIndex(uint16 Index) { Data[0] = Index; }
        uint16 GetLocalObjectIndex() const { return Data[0]; }
    };
 
    struct FPerConnectionInfo
    {
        UE::Net::FNetBitArray ReplicationEnabledObjects;
    };
 
    // UNetObjectFilter interface
    IRISCORE_API virtual void OnInit(const FNetObjectFilterInitParams&) override;
    IRISCORE_API virtual void OnDeinit() override;
    virtual void OnMaxInternalNetRefIndexIncreased(uint32 NewMaxInternalIndex) override {}
    IRISCORE_API virtual void AddConnection(uint32 ConnectionId) override;
    IRISCORE_API virtual void RemoveConnection(uint32 ConnectionId) override;
    IRISCORE_API virtual bool AddObject(uint32 ObjectIndex, FNetObjectFilterAddObjectParams&) override;
    IRISCORE_API virtual void RemoveObject(uint32 ObjectIndex, const FNetObjectFilteringInfo&) override;
    IRISCORE_API virtual void PreFilter(FNetObjectPreFilteringParams&) override;
    IRISCORE_API virtual void Filter(FNetObjectFilteringParams&) override;
 
    TStrongObjectPtr<UNetObjectConnectionFilterConfig> Config;
    TArray<uint32> LocalToNetRefIndex;
    TArray<FPerConnectionInfo> PerConnectionInfos;
    UE::Net::FNetBitArray UsedLocalInfoIndices;
 
    uint32 MaxInternalIndex = 0;
    bool bObjectRemoved = false;
};