AssetRipper_AssetRipper/Source/AssetRipper.Processing/PrefabOutlining/GameObjectCloner.cs

using AssetRipper.Assets;
using AssetRipper.Assets.Cloning;
using AssetRipper.Assets.Collections;
using AssetRipper.SourceGenerated;
using AssetRipper.SourceGenerated.Classes.ClassID_1;
using AssetRipper.SourceGenerated.Extensions;

namespace AssetRipper.Processing.PrefabOutlining;

public static class GameObjectCloner
{
	public static IGameObject Clone(IGameObject source, ProcessedAssetCollection processedCollection)
	{
		Dictionary<IUnityObjectBase, IUnityObjectBase> clonedAssetDictionary = [];
		foreach (IUnityObjectBase asset in source.FetchHierarchy())
		{
			IUnityObjectBase clonedAsset = processedCollection.CreateAsset(asset.ClassID, AssetFactory.Create);
			clonedAssetDictionary.Add(asset, clonedAsset);
		}
		ClonedAssetResolver resolver = new ClonedAssetResolver(clonedAssetDictionary);
		foreach ((IUnityObjectBase asset, IUnityObjectBase clonedAsset) in clonedAssetDictionary)
		{
			PPtrConverter converter = new PPtrConverter(asset.Collection, clonedAsset.Collection, resolver);
			clonedAsset.CopyValues(asset, converter);
		}
		return (IGameObject)clonedAssetDictionary[source];
	}

	private sealed class ClonedAssetResolver : IAssetResolver
	{
		private readonly Dictionary<IUnityObjectBase, IUnityObjectBase> cache;

		public ClonedAssetResolver(Dictionary<IUnityObjectBase, IUnityObjectBase> cache)
		{
			this.cache = cache;
		}

		public T? Resolve<T>(IUnityObjectBase? asset) where T : IUnityObjectBase
		{
			if (asset is null)
			{
				return default;
			}
			else if (cache.TryGetValue(asset, out IUnityObjectBase? clonedAsset))
			{
				return TryCast<T>(clonedAsset);
			}
			else
			{
				return TryCast<T>(asset);
			}
		}

		private static T? TryCast<T>(IUnityObjectBase asset) where T : IUnityObjectBase
		{
			return asset is T t ? t : default;
		}
	}
}