ObjectPool.cs

Go to the documentation of this file.

using UnityEngine;
using System.Collections.Generic;

namespace HKI.Core.Pooling
{
    [AddComponentMenu("HKI/Misc/Object Pool")]
    public class ObjectPool : MonoBehaviour
    {
        // Public variables
        [SerializeField] GameObject Prefab = null;

        [SerializeField] bool Prewarm = true;

        [SerializeField] int InstanctiatedAmountOnPrewarm = 1;

        // Private variables
        Queue<GameObject> objectQueue = new Queue<GameObject>();
        List<GameObject> pooledObjects = new List<GameObject>();

        // Wake function
        void Awake()
        {
            if(Prewarm)
            {
                for(int i = 0; i < InstanctiatedAmountOnPrewarm; i++)
                    InstanctiateObject();
            }
        }

        // Interface function
        public GameObject GetNewPooledObject()
        {
            if(objectQueue.Count == 0)
                InstanctiateObject();

            return objectQueue.Dequeue();
        }

        public void GiveBack(GameObject obj)
        {
            if(obj != null)
                objectQueue.Enqueue(obj);
        }

        public void GiveBackActiveObject(GameObject obj)
        {
            if (obj != null)
            {
                obj.SetActive(false);
                objectQueue.Enqueue(obj);
            }
        }

        // Helper functions
        void InstanctiateObject()
        {
            GameObject obj = Instantiate<GameObject>(Prefab);
            obj.SetActive(false);

            pooledObjects.Add(obj);
            objectQueue.Enqueue(obj);
        }
    }
}