PropertyReplicationFragment.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
#include "Iris/ReplicationSystem/ReplicationFragment.h"
#include "Templates/RefCounting.h"
#include "Templates/UniquePtr.h"
 
namespace UE::Net
{
    class FPropertyReplicationState;
}
 
namespace UE::Net
{
 
class FPropertyReplicationFragment : public FReplicationFragment
{
public:
    IRISCORE_API FPropertyReplicationFragment(EReplicationFragmentTraits InTraits, UObject* InOwner, const FReplicationStateDescriptor* InDescriptor);
 
    IRISCORE_API ~FPropertyReplicationFragment();
 
    const FPropertyReplicationState* GetPropertyReplicationState() const { return SrcReplicationState.Get(); }
 
    void Register(FFragmentRegistrationContext& Fragments);
    
    IRISCORE_API static FPropertyReplicationFragment* CreateAndRegisterFragment(UObject* InOwner, const FReplicationStateDescriptor* InDescriptor, FFragmentRegistrationContext& Context);
 
protected:
 
    virtual void ApplyReplicatedState(FReplicationStateApplyContext& Context) const override;
 
    virtual void CollectOwner(FReplicationStateOwnerCollector* Owners) const override;
    
    virtual void CallRepNotifies(FReplicationStateApplyContext& Context) override;
    
    virtual bool PollReplicatedState(EReplicationFragmentPollFlags PollOption) override;
 
    virtual void ReplicatedStateToString(FStringBuilderBase& StringBuilder, FReplicationStateApplyContext& Context, EReplicationStateToStringFlags Flags) const override;
 
private:
    // This is the source state from which we source our state data
    TUniquePtr<FPropertyReplicationState> SrcReplicationState;
 
    // Previous applied state, only carried around if needed
    TUniquePtr<FPropertyReplicationState> PrevReplicationState;
 
    TRefCountPtr<const FReplicationStateDescriptor> ReplicationStateDescriptor;
 
    // Owner
    UObject* Owner;
};
 
}