From 544eb03f7e102e99f61b7708230f5967f65b28d0 Mon Sep 17 00:00:00 2001
From: Wojtek Figat <wojtek@figat.pl>
Date: Mon, 3 Jul 2023 13:06:08 +0200
Subject: [PATCH] Add network replication support for in-built object/asset
 reference holders

---
 Source/Engine/Serialization/ReadStream.h  | 28 +++++++++++++++++++++++
 Source/Engine/Serialization/Stream.h      |  8 +++++++
 Source/Engine/Serialization/WriteStream.h | 20 ++++++++++++++++
 3 files changed, 56 insertions(+)

diff --git a/Source/Engine/Serialization/ReadStream.h b/Source/Engine/Serialization/ReadStream.h
index 9d3215442..26b536c6d 100644
--- a/Source/Engine/Serialization/ReadStream.h
+++ b/Source/Engine/Serialization/ReadStream.h
@@ -159,6 +159,34 @@ public:
         Read(ptr);
         v = ptr;
     }
+    template<typename T>
+    FORCE_INLINE void Read(SoftObjectReference<T>& v)
+    {
+        uint32 id[4];
+        ReadBytes(id, sizeof(id));
+        v.Set(*(Guid*)id);
+    }
+    template<typename T>
+    FORCE_INLINE void Read(AssetReference<T>& v)
+    {
+        uint32 id[4];
+        ReadBytes(id, sizeof(id));
+        v = (T*)::LoadAsset(*(Guid*)id, T::TypeInitializer);
+    }
+    template<typename T>
+    FORCE_INLINE void Read(WeakAssetReference<T>& v)
+    {
+        uint32 id[4];
+        ReadBytes(id, sizeof(id));
+        v = (T*)::LoadAsset(*(Guid*)id, T::TypeInitializer);
+    }
+    template<typename T>
+    FORCE_INLINE void Read(SoftAssetReference<T>& v)
+    {
+        uint32 id[4];
+        ReadBytes(id, sizeof(id));
+        v.Set(*(Guid*)id);
+    }
 
     /// <summary>
     /// Read data array
diff --git a/Source/Engine/Serialization/Stream.h b/Source/Engine/Serialization/Stream.h
index 83f0eaf21..e477eee95 100644
--- a/Source/Engine/Serialization/Stream.h
+++ b/Source/Engine/Serialization/Stream.h
@@ -17,6 +17,14 @@ class ISerializable;
 class ScriptingObject;
 template<typename T>
 class ScriptingObjectReference;
+template<typename T>
+class SoftObjectReference;
+template<typename T>
+class AssetReference;
+template<typename T>
+class WeakAssetReference;
+template<typename T>
+class SoftAssetReference;
 
 /// <summary>
 /// Base class for all data streams (memory streams, file streams etc.)
diff --git a/Source/Engine/Serialization/WriteStream.h b/Source/Engine/Serialization/WriteStream.h
index 92ac211ca..95bd87df9 100644
--- a/Source/Engine/Serialization/WriteStream.h
+++ b/Source/Engine/Serialization/WriteStream.h
@@ -176,6 +176,26 @@ public:
     {
         Write(v.Get());
     }
+    template<typename T>
+    FORCE_INLINE void Write(const SoftObjectReference<T>& v)
+    {
+        Write(v.Get());
+    }
+    template<typename T>
+    FORCE_INLINE void Write(const AssetReference<T>& v)
+    {
+        Write(v.Get());
+    }
+    template<typename T>
+    FORCE_INLINE void Write(const WeakAssetReference<T>& v)
+    {
+        Write(v.Get());
+    }
+    template<typename T>
+    FORCE_INLINE void Write(const SoftAssetReference<T>& v)
+    {
+        Write(v.Get());
+    }
 
     template<typename T, typename AllocationType = HeapAllocation>
     void Write(const Array<T, AllocationType>& data)