/// <summary>
/// Instantiate prefabs into the Dictionary for later retrieval.
/// </summary>
/// <param name="prefab">Prefab we want to register.</param>
/// <param name="count">Number of prefab we want to register into the pool.</param>
public void Register(PooledBehaviour prefab, int count = 0)
{
//Util.Log(string.Format("Registering {0} of {1}.", count, prefab.gameObject.name));
if (!pools.ContainsKey(prefab.gameObject.name))
{
pools.Add(prefab.name, new Stack <PooledBehaviour>());
}
for (int i = 0; i < count; i++)
{
PooledBehaviour pb = (Instantiate(prefab.gameObject)).GetComponent <PooledBehaviour>();
pb.gameObject.name = prefab.gameObject.name;
pb.gameObject.SetActive(false);
Util.Parent(pb.gameObject, this.gameObject);
ids.Add(pb.gameObject.GetInstanceID());
pools[prefab.name].Push(pb);
}
//Util.Log(string.Format("Instantiated {0} of {1}", pools[prefab.name].Count, prefab.name));
}
/// <summary>
/// Return a PooledBehavior to the pool.
/// </summary>
/// <param name="pb">PooledBehavior we want to return to the pool.</param>
public void Return(PooledBehaviour pb)
{
Util.Assert(
pools.ContainsKey(pb.gameObject.name),
string.Format("Unable to find {0} in Pool. Did you use Instantiate?", pb.gameObject.name));
// Prevents returning of already returned gameobjects.
if (!ids.Contains(pb.gameObject.GetInstanceID()))
{
pb.Reset();
pb.gameObject.SetActive(false);
ids.Add(pb.gameObject.GetInstanceID());
pools[pb.gameObject.name].Push(pb);
Util.Parent(pb.gameObject, this.gameObject);
}
else
{
//Util.Log(string.Format("Duplicate ID detected: {0}. Did you already return this behavior?", pb.GetInstanceID()));
}
}
/// <summary>
/// Pop an unused PooledBehavior from the approrpriate stack.
/// If the Stack is empty, we Instantiate the object.
/// </summary>
/// <typeparam name="T">Derived type of the pooledbehavior we're retrieving</typeparam>
/// <param name="script">Script on the prefab we want to retrieve</param>
/// <returns>PooledBehavior derived script attached to a resetted gameObject.</returns>
public T Get <T>(T script) where T : PooledBehaviour
{
Util.Assert(
pools.ContainsKey(script.gameObject.name),
string.Format("Unable to find {0} in Pool. Did you register the prefab?", script.gameObject.name)
);
PooledBehaviour pb = null;
if (pools[script.gameObject.name].Count == 0)
{
//Util.Log(string.Format("Pool ran out of {0}. Instantiating...", script.gameObject.name));
pb = (Instantiate(script.gameObject)).GetComponent <PooledBehaviour>();
pb.gameObject.name = script.gameObject.name;
}
else
{
//Util.Log(string.Format("Getting {0} from a pool of {1}.", script.gameObject.name, pools[script.gameObject.name].Count));
pb = pools[script.gameObject.name].Pop();
ids.Remove(pb.gameObject.GetInstanceID());
}
pb.gameObject.SetActive(true);
//Util.Log("Retrieved " + pb.GetInstanceID());
return(pb.GetComponent <T>());
}
