69 lines
1.8 KiB
C#
69 lines
1.8 KiB
C#
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using UnityEngine;
|
|
|
|
public class ObjectPool<T> : MonoBehaviour where T : Component
|
|
{
|
|
[SerializeField] private int numObjects;
|
|
[SerializeField] private T prefab;
|
|
[SerializeField] private int numToSpawnPerFrame = 10;
|
|
|
|
private List<T> _pool;
|
|
private List<GameObject> _poolGameObjects;
|
|
|
|
public event Action<T> OnObjectFound;
|
|
public event Action Initialised;
|
|
|
|
public (T, int) GetObject()
|
|
{
|
|
T foundObject = null;
|
|
int idx = -1;
|
|
if (_pool == null) return (null, -1);
|
|
|
|
for (var index = _pool.Count - 1; index >= 0; index--)
|
|
{
|
|
if (!_poolGameObjects[index].activeSelf)
|
|
{
|
|
idx = index;
|
|
foundObject = _pool[index];
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (foundObject == null)
|
|
Debug.LogWarning("No more objects in pool!");
|
|
else
|
|
OnObjectFound?.Invoke(foundObject);
|
|
|
|
return (foundObject, idx);
|
|
}
|
|
|
|
public void ReturnObject(T obj, int idx)
|
|
{
|
|
_pool.RemoveAt(idx);
|
|
_pool.Add(obj);
|
|
|
|
_poolGameObjects.RemoveAt(idx);
|
|
_poolGameObjects.Add(obj.gameObject);
|
|
}
|
|
|
|
public IEnumerator Start()
|
|
{
|
|
_pool = new List<T>(numObjects);
|
|
_poolGameObjects = new List<GameObject>(numObjects);
|
|
for (int i = 0; i < numObjects; ++i)
|
|
{
|
|
var instance = Instantiate(prefab, transform);
|
|
instance.gameObject.SetActive(false);
|
|
_pool.Add(instance);
|
|
_poolGameObjects.Add(instance.gameObject);
|
|
|
|
if (i % numToSpawnPerFrame == 0)
|
|
yield return null;
|
|
}
|
|
|
|
Initialised?.Invoke();
|
|
}
|
|
} |