在Unity中实现对象池

用途

避免频繁地创建和销毁对象

原理

预先创建若干对象

  • 在要使用时选一个未被使用的进行使用
  • 在用完要销毁时将其标记为未使用而非销毁,以供下次使用

实现

首先实现一个容器,容器中有对象,用一个标识符来标记当前对象是否被使用

/// <summary>
/// 供对象池使用的容器(Container):容器内对象的内容、标记容器内对象为已被(未被)使用
/// </summary>
/// <typeparam name="T"></typeparam>
public class ObjectPoolContainer<T>
{
    // 容器内对象
    public T Item { get; set; }
    // 容器内对象是否已被使用的标记
    public bool Used { get; private set; }

    // 标记为已被使用
    public void Consume()
    {
        Used = true;
    }

    // 标记为未被使用
    public void Release()
    {
        Used = false;
    }
}

实现一个对象池,该对象池中的对象可以是任意对象(Object)
使用时需先自行创建对象池实例,在初始化时设置对象类型和池的初始容量

使用举例

private ObjectPool<MyObj> myObjPool = new ObjectPool<MyObj>(() => new MyObj(), 10);

var myObj = myObjPool.GetItem();
 
myObjPool.ReleaseItem(myObj); 

ObjectPool.cs

using System;
using System.Collections.Generic;
using UnityEngine;

/// <summary>
/// 一般对象的池子
/// </summary>
/// <typeparam name="T"></typeparam>
public class ObjectPool<T>
{
    // 池中所有对象
    private List<ObjectPoolContainer<T>> list;
    // 已被使用的对象(用于查询)
    private Dictionary<T, ObjectPoolContainer<T>> lookup;
    private Func<T> factoryFunc;
    private int lastIndex;

    public int Count { get { return list.Count; } }
    public int CountUsedItems { get { return lookup.Count; } }

    public ObjectPool(Func<T> factoryFunc, int initialSize)
    {
        // 该委托用于返回一个(新创建的)对象实例
        this.factoryFunc = factoryFunc;

        list = new List<ObjectPoolContainer<T>>(initialSize);
        lookup = new Dictionary<T, ObjectPoolContainer<T>>(initialSize);

        Warm(initialSize);
    }

    // 初始时即生成capacity个对象
    private void Warm(int capacity)
    {
        for (int i = 0; i < capacity; i++)
        {
            CreateContainer();
        }
    }

    private ObjectPoolContainer<T> CreateContainer()
    {
        var container = new ObjectPoolContainer<T>();
        container.Item = factoryFunc();
        list.Add(container);
        return container;
    }

    public T GetItem()
    {
        ObjectPoolContainer<T> container = null;

        // 若在list中找得到还未被使用的对象,则使用该对象作为返回值
        for (int i = 0; i < list.Count; i++)
        {
            lastIndex++;
            // 最可能没使用的就是上次使用了的下一个。若上次使用的是最后1个,则这次只要从第0个开始查
            if (lastIndex > list.Count - 1)
                lastIndex = 0;

            if (list[lastIndex].Used)
                continue;
            else
            {
                container = list[lastIndex];
                break;
            }
        }

        // list中所有对象都被使用了,新创建一个对象
        if (container == null)
            container = CreateContainer();

        // 标记为已使用
        container.Consume();
        lookup.Add(container.Item, container);

        return container.Item;
    }

    public void ReleaseItem(object item)
    {
        ReleaseItem((T)item);
    }

    public void ReleaseItem(T item)
    {
        if (lookup.ContainsKey(item))
        {
            // 标记为未使用
            var container = lookup[item];
            container.Release();
            lookup.Remove(item);
        }
    }
}

实现一个用于Unity中游戏对象(GameObject)的对象池
使用时直接调用相关静态方法

使用举例

GameObjectPool.WarmPool(prefab, 10);

var go = GameObjectPool.CreateObject(prefab, positon, rotation);

GameObjectPool.ReleaseObject(go);

GameObjectPool.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

/// <summary>
/// Unity中GameObject的池子
///     与一般Object的区别是:GameObject可直接出现在场景(Scene)中
/// </summary>
public class GameObjectPool : Singleton<GameObjectPool>
{
    public bool shouldShowLog;
    public Transform root;

    // 预制体-对象池
    private Dictionary<GameObject, ObjectPool<GameObject>> prefabLookup;
    // 实例-对象池(隶属于同一预制体的实例对应的对象池是一样的)
    private Dictionary<GameObject, ObjectPool<GameObject>> instanceLookup;

    private bool hasRefreshedLog = false;

    private void Awake()
    {
        Init();
    }

    private void Update()
    {
        if (shouldShowLog && hasRefreshedLog)
        {
            ShowLog();
            hasRefreshedLog = false;
        }
    }

    private void Init()
    {
        prefabLookup = new Dictionary<GameObject, ObjectPool<GameObject>>();
        instanceLookup = new Dictionary<GameObject, ObjectPool<GameObject>>();
    }

    public void WarmPoolNonStatic(GameObject prefab, int size, bool isActive)
    {
        var pool = new ObjectPool<GameObject>(() => { return InstantiatePrefab(prefab, isActive); }, size);
        prefabLookup[prefab] = pool;

        hasRefreshedLog = true;
    }

    public GameObject CreateObjectNonStatic(GameObject prefab)
    {
        return CreateObjectNonStatic(prefab, Vector3.zero, Quaternion.identity);
    }

    public GameObject CreateObjectNonStatic(GameObject prefab, Vector3 position, Quaternion rotation)
    {
        if (!prefabLookup.ContainsKey(prefab))
            WarmPool(prefab, 1, true);

        var pool = prefabLookup[prefab];

        var go = pool.GetItem();
        go.transform.position = position;
        go.transform.rotation = rotation;
        go.SetActive(true);

        instanceLookup.Add(go, pool);
        hasRefreshedLog = true;
        return go;
    }

    public void ReleaseObjectNonStatic(GameObject go)
    {
        go.SetActive(false);

        if (instanceLookup.ContainsKey(go))
        {
            instanceLookup[go].ReleaseItem(go);
            instanceLookup.Remove(go);
            hasRefreshedLog = true;
        }
    }

    private GameObject InstantiatePrefab(GameObject prefab, bool isActive)
    {
        var go = Instantiate(prefab) as GameObject;
        if (isActive)
            go.SetActive(true);
        else
            go.SetActive(false);

        if (root != null)
            go.transform.parent = root;
        return go;
    }

    public void ShowLog()
    {
        foreach (var item in prefabLookup)
        {
            Debug.Log(string.Format("“游戏对象池”  预制体名称:{0}  {1}个在被使用,共有{2}个", item.Key.name, item.Value.CountUsedItems, item.Value.Count));
        }
    }

    #region 静态接口(供外部直接调用)
    public static void WarmPool(GameObject prefab, int size, bool isActive = false)
    {
        Instance.WarmPoolNonStatic(prefab, size, isActive);
    }

    public static GameObject CreateObject(GameObject prefab)
    {
        return Instance.CreateObjectNonStatic(prefab);
    }

    public static GameObject CreateObject(GameObject prefab, Vector3 position, Quaternion rotation)
    {
        return Instance.CreateObjectNonStatic(prefab, position, rotation);
    }

    public static void ReleaseObject(GameObject go)
    {
        Instance.ReleaseObjectNonStatic(go);
    }
    #endregion
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 160,165评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,720评论 1 298
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,849评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,245评论 0 213
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,596评论 3 288
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,747评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,977评论 2 315
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,708评论 0 204
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,448评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,657评论 2 249
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,141评论 1 261
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,493评论 3 258
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,153评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,108评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,890评论 0 198
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,799评论 2 277
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,685评论 2 272

推荐阅读更多精彩内容