AutoMorpherDecompiled/Assets/@Eden_Tools/Eden_AutoMorpher/Script/BodyPoseToClothApplier.cs

// Warning: Some assembly references could not be resolved automatically. This might lead to incorrect decompilation of some parts,
// for ex. property getter/setter access. To get optimal decompilation results, please manually add the missing references to the list of loaded assemblies.
// EdenAutoMorpherScript, Version=0.0.0.0, Culture=neutral, PublicKeyToken=null
// Eden.AutoMorpher.BodyPoseToClothApplier
using System.Collections.Generic;
using System.Linq;
using Eden.AutoMorpher;
using UnityEngine;

public class BodyPoseToClothApplier
{
	private class BoneMapping
	{
		public Transform proxyBone;

		public Transform clothBone;

		public int depth;

		public HumanBodyBones? humanBone;
	}

	public void ApplyBodyPoseToClothes(Transform bodyProxyRoot, Transform clothesRoot, Dictionary<Transform, BoneMatchUtil.BoneRootLocalData> clothToBodyMatched, Dictionary<Transform, Transform> sourceToProxy, bool copyPosition = true, bool copyRotation = false, bool copyScale = true)
	{
		if (bodyProxyRoot == null)
		{
			throw new AutoMorpherException("Body Proxy Root is Missing", "[BodyToClothPoseApplier] ApplyDeformedProxyPoseToCloth\n - bodyProxyRoot is null");
		}
		if (clothesRoot == null)
		{
			throw new AutoMorpherException("Clothes Root is Missing", "[BodyToClothPoseApplier] ApplyDeformedProxyPoseToCloth\n - clothesRoot is null");
		}
		if (clothToBodyMatched == null || clothToBodyMatched.Count == 0)
		{
			throw new AutoMorpherException("Cloth To Body Matched Map is Missing", "[BodyToClothPoseApplier] ApplyDeformedProxyPoseToCloth\n - clothToBodyMatched is null or empty");
		}
		if (sourceToProxy == null)
		{
			throw new AutoMorpherException("Source To Proxy Map is Missing", "[BodyToClothPoseApplier] ApplyDeformedProxyPoseToCloth\n - sourceToProxy is null");
		}
		clothesRoot.localScale = bodyProxyRoot.localScale;
		Animator component = bodyProxyRoot.GetComponent<Animator>();
		if (component == null)
		{
			throw new AutoMorpherException("Body Proxy Animator is Missing", "[BodyToClothPoseApplier] ApplyDeformedProxyPoseToCloth\n - bodyProxyRoot has no Animator");
		}
		Transform transform = component.transform;
		Dictionary<Transform, int> dictionary = BuildDepthMap(transform);
		Dictionary<Transform, Transform> dictionary2 = new Dictionary<Transform, Transform>();
		List<BoneMapping> list = new List<BoneMapping>();
		HashSet<Transform> hashSet = new HashSet<Transform>();
		foreach (KeyValuePair<Transform, BoneMatchUtil.BoneRootLocalData> item in clothToBodyMatched)
		{
			Transform key = item.Key;
			BoneMatchUtil.BoneRootLocalData value = item.Value;
			if (key == null || value == null)
			{
				continue;
			}
			Transform t = value.t;
			if (t == null || !sourceToProxy.TryGetValue(t, out var value2) || value2 == null)
			{
				continue;
			}
			HumanBodyBones hBone = value.hBone;
			if (hashSet.Add(key))
			{
				int depth = 999;
				if (dictionary != null && dictionary.TryGetValue(value2, out var value3))
				{
					depth = value3;
				}
				list.Add(new BoneMapping
				{
					proxyBone = value2,
					clothBone = key,
					depth = depth,
					humanBone = ((hBone != HumanBodyBones.LastBone) ? new HumanBodyBones?(hBone) : ((HumanBodyBones?)null))
				});
			}
			if (!dictionary2.ContainsKey(value2))
			{
				dictionary2.Add(value2, key);
			}
		}
		MirrorBoneMarkers(transform, dictionary2, dictionary, list);
		foreach (BoneMapping item2 in (from m in list
			where m != null && m.proxyBone != null && m.clothBone != null
			orderby m.depth
			select m).ToList())
		{
			if (!(item2.proxyBone == null) && !(item2.clothBone == null))
			{
				if (copyPosition)
				{
					item2.clothBone.position = item2.proxyBone.position;
				}
				if (copyRotation)
				{
					item2.clothBone.rotation = item2.proxyBone.rotation;
				}
				if (copyScale)
				{
					item2.clothBone.localScale = item2.proxyBone.localScale;
				}
			}
		}
	}

	private Dictionary<Transform, int> BuildDepthMap(Transform root)
	{
		Dictionary<Transform, int> dictionary = new Dictionary<Transform, int>();
		if (root == null)
		{
			return dictionary;
		}
		Transform[] componentsInChildren = root.GetComponentsInChildren<Transform>(includeInactive: true);
		foreach (Transform transform in componentsInChildren)
		{
			if (!(transform == null))
			{
				dictionary[transform] = GetDepthFromRoot(transform, root);
			}
		}
		return dictionary;
	}

	private int GetDepthFromRoot(Transform t, Transform root)
	{
		int num = 0;
		Transform transform = t;
		while (transform != null && transform != root)
		{
			num++;
			transform = transform.parent;
		}
		return num;
	}

	private void MirrorBoneMarkers(Transform proxyArmature, Dictionary<Transform, Transform> proxyToCloth, Dictionary<Transform, int> proxyDepthMap, List<BoneMapping> mappings)
	{
		if (proxyArmature == null || proxyToCloth == null || mappings == null)
		{
			return;
		}
		TempBoneMarker[] componentsInChildren = proxyArmature.GetComponentsInChildren<TempBoneMarker>(includeInactive: true);
		if (componentsInChildren == null || componentsInChildren.Length == 0)
		{
			return;
		}
		HashSet<Transform> hashSet = new HashSet<Transform>();
		for (int i = 0; i < mappings.Count; i++)
		{
			BoneMapping boneMapping = mappings[i];
			if (boneMapping != null && boneMapping.proxyBone != null)
			{
				hashSet.Add(boneMapping.proxyBone);
			}
		}
		foreach (TempBoneMarker item in (from bm in componentsInChildren
			where bm != null && bm.transform != null
			orderby (proxyDepthMap != null && proxyDepthMap.TryGetValue(bm.transform, out var value3)) ? value3 : int.MaxValue
			select bm).ToList())
		{
			Transform transform = item.transform;
			if (transform == null)
			{
				continue;
			}
			Transform originalParent = item.originalParent;
			if (originalParent == null || !proxyToCloth.TryGetValue(originalParent, out var value) || value == null)
			{
				continue;
			}
			List<Transform> list = CollectDirectChilds(value);
			Transform transform2 = FindMarkerInChild(value);
			TempBoneMarker tempBoneMarker = null;
			if (transform2 == null)
			{
				transform2 = new GameObject(transform.name).transform;
				transform2.SetParent(value, worldPositionStays: true);
				tempBoneMarker = transform2.gameObject.AddComponent<TempBoneMarker>();
			}
			else
			{
				tempBoneMarker = transform2.GetComponent<TempBoneMarker>();
			}
			transform2.position = transform.position;
			transform2.rotation = transform.rotation;
			transform2.localScale = transform.localScale;
			tempBoneMarker.originalParent = value;
			tempBoneMarker.additionalInfo = item.additionalInfo;
			tempBoneMarker.wrappedChildNames = item.wrappedChildNames;
			for (int num = 0; num < list.Count; num++)
			{
				Transform transform3 = list[num];
				if (!(transform3 == null) && !(transform3 == transform2))
				{
					transform3.SetParent(transform2, worldPositionStays: true);
				}
			}
			if (!proxyToCloth.ContainsKey(transform))
			{
				proxyToCloth.Add(transform, transform2);
			}
			if (hashSet.Add(transform))
			{
				int depth = 999;
				if (proxyDepthMap != null && proxyDepthMap.TryGetValue(transform, out var value2))
				{
					depth = value2;
				}
				mappings.Add(new BoneMapping
				{
					proxyBone = transform,
					clothBone = transform2,
					depth = depth,
					humanBone = null
				});
			}
		}
	}

	private Transform FindMarkerInChild(Transform parent)
	{
		if (parent == null)
		{
			return null;
		}
		for (int i = 0; i < parent.childCount; i++)
		{
			Transform child = parent.GetChild(i);
			if (!(child == null) && child.GetComponent<TempBoneMarker>() != null)
			{
				return child;
			}
		}
		return null;
	}

	private List<Transform> CollectDirectChilds(Transform parentTransform)
	{
		int childCount = parentTransform.childCount;
		List<Transform> list = new List<Transform>(childCount);
		for (int i = 0; i < childCount; i++)
		{
			list.Add(parentTransform.GetChild(i));
		}
		return list;
	}
}