58 lines
1.9 KiB
C#
58 lines
1.9 KiB
C#
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
|
|
public static class GameObjectPool
|
|
{
|
|
private static GameObject PoolAttach;
|
|
private static Dictionary<GameObject, Queue<GameObject>> s_poolMap = new Dictionary<GameObject, Queue<GameObject>>();
|
|
private static Dictionary<GameObject, GameObject> s_instanceToSrc = new Dictionary<GameObject, GameObject>();
|
|
public static GameObject GetInstance(GameObject prefabSrc, Transform parent)
|
|
{
|
|
if (PoolAttach == null)
|
|
{
|
|
PoolAttach = new GameObject("[GameObjectPool]");
|
|
PoolAttach.transform.localScale = Vector3.zero;
|
|
GameObject.DontDestroyOnLoad(PoolAttach);
|
|
}
|
|
|
|
if (!s_poolMap.ContainsKey(prefabSrc)) s_poolMap[prefabSrc] = new Queue<GameObject>();
|
|
|
|
GameObject instance = null;
|
|
var pool = s_poolMap[prefabSrc];
|
|
if (pool.Count > 0)
|
|
{
|
|
instance = pool.Dequeue();
|
|
instance.transform.SetParent(parent, true);
|
|
instance.transform.localScale = prefabSrc.transform.localScale;
|
|
instance.transform.localRotation = prefabSrc.transform.localRotation;
|
|
instance.transform.localPosition = prefabSrc.transform.localPosition;
|
|
}
|
|
else
|
|
{
|
|
instance = GameObject.Instantiate(prefabSrc, parent);
|
|
s_instanceToSrc[instance] = prefabSrc;
|
|
}
|
|
|
|
//instance.SetActive(true);
|
|
return instance;
|
|
}
|
|
|
|
public static void Release(GameObject instance)
|
|
{
|
|
s_instanceToSrc.TryGetValue(instance, out var src);
|
|
if (src != null && s_poolMap.TryGetValue(src, out var pool))
|
|
{
|
|
pool.Enqueue(instance);
|
|
//instance.SetActive(false);
|
|
if (instance == null) return;
|
|
if (PoolAttach == null) return;
|
|
|
|
instance.transform.SetParent(PoolAttach.transform, true);
|
|
}
|
|
else
|
|
{
|
|
GameObject.Destroy(instance);
|
|
}
|
|
}
|
|
}
|