XRLib/Assets/Scripts/Simulator/Model/EntityManager.cs

using Cysharp.Threading.Tasks;
using System;
using System.Collections.Generic;
using UnityEngine;
using UVC.Core;
using UVC.Pool;

public class EntityManager : SingletonScene<EntityManager>
{
    private readonly Dictionary<string, string> prefabPaths = new Dictionary<string, string>()
    {
        {"box","prefabs/Box"}
    };

    private Dictionary<string, Entity> entityDatas = new Dictionary<string, Entity>();

    public Vector3 ObjectSize;
    public event Action<Entity> OnEntityDestroyed;
    public event Action<Entity> OnEntityTransferred;

    private GameObjectPool<Entity>? entityPool;
    public GameObjectPool<Entity> EntityPool
    {
        get
        {
            if (entityPool == null)
            {
                Debug.LogError("Pool is not initialized. Please call InitializePoolAsync first.");
            }
            return entityPool!;
        }
    }

    protected override void Init()
    {
        InitializeEntityPoolAsync().Forget();
    }

    private async UniTask InitializeEntityPoolAsync()
    {
        if (entityPool != null) return;

        var prefab = await Resources.LoadAsync<GameObject>(prefabPaths["box"]) as GameObject;
        if (prefab == null)
        {
            Debug.LogError($"Prefab not found at path: {prefabPaths["box"]}");
            return;
        }
        entityPool = new GameObjectPool<Entity>(prefab, transform);
        ObjectSize = prefab.GetComponent<MeshFilter>().sharedMesh.bounds.size;
    }

    public Entity SpawnEntity(string entityName, string prefabName)
    {
        var entity = entityPool.GetItem($"{entityName}");
        entity.name = entityName;
        SetEntity(entity, prefabName);
        if (!entityDatas.ContainsKey(entityName))
        {
            entityDatas.Add(entityName, entity);
        }
        else
        {
            entityDatas[entityName] = entity;
        }
        return entity;
    }

    public List<Entity> SpawnEntites(List<string> entityNames)
    {
        List<Entity> entities = new List<Entity>();
        foreach (var entityName in entityNames)
        {
            var entity = entityPool.GetItem($"{entityName}");
            entity.name = entityName;
            if (!entityDatas.ContainsKey(entityName))
            {
                entityDatas.Add(entityName, entity);
                entities.Add(entity);
            }
        }
        return entities;
    }

    public Entity GetEntity(string entityName, string prefabName = null)
    {
        if (entityDatas.ContainsKey(entityName))
        {
            var entity = entityDatas[entityName];
            OnEntityTransferred?.Invoke(entity);
            return entity;
        }
        else
        {
            return SpawnEntity(entityName, prefabName);
        }
    }

    public List<Entity> GetEntities(List<string> entityNames, List<string> prefabNames = null)
    {
        List<Entity> entities = new List<Entity>();
        foreach (var entityName in entityNames)
        {
            var entity = GetEntity(entityName);
            entities.Add(entity);
        }
        return entities;
    }

    public void SetEntity(Entity entity, string prefabName)
    {
        var prefabData = PrefabManager.Instance.GetPrefab(prefabName);
        if (!string.IsNullOrEmpty(prefabName) && prefabData != null)
        {
            string colorHex = $"{prefabData.color}";
            if (ColorUtility.TryParseHtmlString(colorHex, out var color))
            {
                entity.SetColor(color);
            }
        }
    }

    public void DestroyEnity(List<string> entityNames)
    {
        foreach (var entityName in entityNames)
        {
            if (entityDatas.ContainsKey(entityName))
            {
                var entity = entityDatas[entityName];
                OnEntityTransferred?.Invoke(entity);
                OnEntityDestroyed?.Invoke(entity);
                Destroy(entity.gameObject);
                entityDatas.Remove(entityName);
            }
        }
    }
}