GHZMRSupportXR/Assets/TriLib/TriLibCore/Scripts/AssetUnloader.cs

using System.Collections.Generic;
using UnityEngine;
namespace TriLibCore
{
    /// <summary>Represents a Class to destroy every Asset (Textures, Materials, Meshes) loaded by TriLib for this GameObject.</summary>
    public class AssetUnloader : MonoBehaviour
    {
        /// <summary>
        /// Assets Allocation List.
        /// </summary>
        public List<Object> Allocations;

        private int _id;
        /// <summary>The Asset Unloader unique identifier.</summary>
        public int Id
        {
            get => _id;
            set
            {
                _id = value;
                Register();
            }
        }

        private static int _lastId;

        private static readonly Dictionary<int, int> AssetUnloaders = new Dictionary<int, int>();

        /// <summary>Gets the next allocation Identifier.</summary>
        /// <returns>The Allocation Identifier.</returns>
        public static int GetNextId()
        {
            return _lastId++;
        }

        private void Register()
        {
            if (!AssetUnloaders.ContainsKey(_id))
            {
                AssetUnloaders[_id] = 0;
            }
            else
            {
                AssetUnloaders[_id]++;
            }
        }

        private void Start()
        {
            Register();
            ChangeLayerRecursively(transform);
        }

        public void ChangeLayerRecursively(Transform parentTransform)
        {
            // 修改父物体的层级
            parentTransform.gameObject.layer = LayerMask.NameToLayer("Arrow");

            // 遍历所有子物体，并递归调用ChangeLayerRecursively方法
            for (int i = 0; i < parentTransform.childCount; i++)
            {
                Transform childTransform = parentTransform.GetChild(i);
                ChangeLayerRecursively(childTransform);
            }
        }

        private void OnDestroy()
        {
            if (AssetUnloaders.TryGetValue(_id, out var value))
            {
                if (--value <= 0)
                {
                    foreach (var allocation in Allocations)
                    {
                        if (allocation == null)
                        {
                            continue;
                        }
                        Destroy(allocation);
                    }
                    AssetUnloaders.Remove(_id);
                }
                else
                {
                    AssetUnloaders[_id] = value;
                }
            }
        }
    }
}