UniVRM/Assets/VRM10_Samples/ClothSample/RotateParticle/Runtime/RotateParticleSpringboneRuntime.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using SphereTriangle;
using UniGLTF;
using UniGLTF.SpringBoneJobs.Blittables;
using UnityEngine;
using UniVRM10;

namespace RotateParticle
{
    /// <summary>
    /// SpcriptedImporter 経由の import 向け。
    /// NativeArray の確保や DontDestroyOnLoad を回避。
    /// </summary>
    public class RotateParticleSpringboneRuntime : IVrm10SpringBoneRuntime
    {
        RotateParticleSystem _system;

        public void Dispose()
        {
        }

        public async Task InitializeAsync(Vrm10Instance instance, IAwaitCaller awaitCaller)
        {
            await awaitCaller.NextFrame();
            // if (_setup != null)
            // {
            //     _setup(instance);
            // }

            var animator = instance.GetComponent<Animator>();
            if (animator == null)
            {
                Debug.LogWarning("no animator");
                return;
            }

            var avatar = animator.avatar;
            if (!avatar.isHuman)
            {
                Debug.LogWarning("not humanoid");
                return;
            }

            _system = new RotateParticleSystem();
            _system.Env.DragForce = 0.6f;
            _system.Env.Stiffness = 0.07f;

            foreach (var (group, head, tail, radius) in Capsules)
            {
                _system.AddColliderIfNotExists(group,
                    animator.GetBoneTransform(head), animator.GetBoneTransform(tail), radius);
            }

            // skirt
            {
                if (TryAddGroup(CollisionGroupMask.Group01, "Skirt", animator, HumanBodyBones.Hips, new[] { "skirt", "ｽｶｰﾄ", "スカート" }, StrandConnectionType.ClothLoop,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }
            {
                if (TryAddGroupChildChild(CollisionGroupMask.Group01, "Skirt", animator, HumanBodyBones.Hips, new[] { "skirt", "ｽｶｰﾄ", "スカート" }, new string[] { }, StrandConnectionType.ClothLoop, false,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }
            {
                if (TryAddGroup(CollisionGroupMask.Group02, "髪", animator, HumanBodyBones.Head, new[] { "髪", "hair" }, StrandConnectionType.Strand,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }
            {
                if (TryAddGroup(CollisionGroupMask.Group01, "裾", animator, HumanBodyBones.Hips, new[] { "裾" }, StrandConnectionType.Cloth,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }
            {
                if (TryAddGroupChildChild(CollisionGroupMask.Group02, "左袖", animator, HumanBodyBones.LeftUpperArm, new[] { "袖" }, new[] { "ひじ袖" }, StrandConnectionType.ClothLoop,
                    false,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }
            {
                if (TryAddGroupChildChild(CollisionGroupMask.Group02, "左ひじ袖", animator, HumanBodyBones.LeftLowerArm, new[] { "袖" }, new string[] { }, StrandConnectionType.ClothLoop,
                    false,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }
            {
                if (TryAddGroupChildChild(CollisionGroupMask.Group02, "右袖", animator, HumanBodyBones.RightUpperArm, new[] { "袖" }, new[] { "ひじ袖" }, StrandConnectionType.ClothLoop,
                    false,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }
            {
                if (TryAddGroupChildChild(CollisionGroupMask.Group02, "右ひじ袖", animator, HumanBodyBones.RightLowerArm, new[] { "袖" }, new string[] { }, StrandConnectionType.ClothLoop,
                    false,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }
            {
                if (TryAddGroup(CollisionGroupMask.Group01 | CollisionGroupMask.Group02, "マント", animator, HumanBodyBones.Chest, new[] { "マント" }, StrandConnectionType.Cloth,
                    out var g))
                {
                    _system._strandGroups.Add(g);
                }
            }

            _system.Initialize();
        }

        static (string group, HumanBodyBones head, HumanBodyBones tail, float radius)[] Capsules = new[]
        {
            ("Leg", HumanBodyBones.LeftUpperLeg, HumanBodyBones.LeftLowerLeg, 0.06f),
            ("Leg", HumanBodyBones.LeftLowerLeg, HumanBodyBones.LeftFoot, 0.05f),
            ("Leg", HumanBodyBones.RightUpperLeg, HumanBodyBones.RightLowerLeg, 0.06f),
            ("Leg", HumanBodyBones.RightLowerLeg, HumanBodyBones.RightFoot, 0.05f),

            ("Arm", HumanBodyBones.LeftUpperArm, HumanBodyBones.LeftLowerArm, 0.03f),
            ("Arm", HumanBodyBones.LeftLowerArm, HumanBodyBones.LeftHand, 0.03f),
            ("Arm", HumanBodyBones.LeftHand, HumanBodyBones.LeftMiddleProximal, 0.02f),
            ("Arm", HumanBodyBones.RightUpperArm, HumanBodyBones.RightLowerArm, 0.03f),
            ("Arm", HumanBodyBones.RightLowerArm, HumanBodyBones.RightHand, 0.03f),
            ("Arm", HumanBodyBones.RightHand, HumanBodyBones.RightMiddleProximal, 0.02f),
        };

        // T GetOrAddComponent<T>() where T : Component
        // {
        //     var t = GetComponent<T>();
        //     if (t != null)
        //     {
        //         return t;
        //     }
        //     return gameObject.AddComponent<T>();
        // }

        bool TryAddGroup(CollisionGroupMask mask, string name, Animator animator, HumanBodyBones humanBone, string[] targets, StrandConnectionType type,
            out StrandGroup group)
        {
            var bone = animator.GetBoneTransform(humanBone);
            if (bone == null)
            {
                Debug.LogWarning($"{humanBone} not found");
                group = default;
                return false;
            }

            List<Transform> transforms = new();
            foreach (Transform child in bone)
            {
                foreach (var target in targets)
                {
                    if (child.name.ToLower().Contains(target.ToLower()))
                    {
                        transforms.Add(child);
                        break;
                    }
                }
            }
            if (transforms.Count == 0)
            {
                // Debug.LogWarning($"{string.Join(',', targets)} not found");
                group = default;
                return false;
            }

            group = new StrandGroup
            {
                Name = name,
                CollisionMask = mask,
                DefaultStrandRaius = 0.02f,
                Connection = type
            };
            group.Roots.AddRange(transforms);

            // sort
            group.Roots.Sort(new TransformSort(bone.position));
            return true;
        }

        bool TryAddGroupChildChild(CollisionGroupMask mask, string name, Animator animator, HumanBodyBones humanBone,
            string[] targets, string[] excludes,
            StrandConnectionType type,
            bool sort,
            out StrandGroup group)
        {
            var bone = animator.GetBoneTransform(humanBone);
            if (bone == null)
            {
                Debug.LogWarning($"{humanBone} not found");
                group = default;
                return false;
            }

            List<Transform> transforms = new();
            foreach (Transform child in bone)
            {
                foreach (Transform childchild in child)
                {
                    if (excludes.Any(x => childchild.name.ToLower().Contains(x.ToLower())))
                    {
                        continue;
                    }

                    foreach (var target in targets)
                    {
                        if (childchild.name.ToLower().Contains(target.ToLower()))
                        {
                            transforms.Add(childchild);
                            break;
                        }
                    }
                }
            }
            if (transforms.Count == 0)
            {
                // Debug.LogWarning($"{string.Join(',', targets)} not found");
                group = default;
                return false;
            }

            group = new StrandGroup
            {
                Name = name,
                CollisionMask = mask,
                DefaultStrandRaius = 0.02f,
                Connection = type
            };
            group.Roots.AddRange(transforms);

            // sort
            if (sort)
            {
                group.Roots.Sort(new TransformSort(bone.position));
            }
            return true;
        }

        public void Process()
        {
            _system.Process(Time.deltaTime);
        }

        public bool ReconstructSpringBone()
        {
            return false;
        }

        public void RestoreInitialTransform()
        {
        }

        public void SetJointLevel(Transform joint, BlittableJointMutable jointSettings)
        {
        }

        public void SetModelLevel(Transform modelRoot, BlittableModelLevel modelSettings)
        {
        }
    }
}