U3D性能优化之对象池(Object Pooling)

一共两个脚本(lua):CachePrefabManager和CachePrefab

require "Game.CommonPrefab.Lua.CachePrefab"

CachePrefabManager = {}

local this
local cachemap = {}
local rect_width = 1920
local rect_height = 1080

-- 动态缓存对象
local dt_cache_map = {}

function CachePrefabManager.Init()
	if not this then
		local pp = GameObject.Find("GameManager").transform
		if not IsEquals(pp) then
			print("<color=red>CachePrefabManager.Init()</color>")
		end
		this = CachePrefabManager

		local obj = GameObject.New()
		local rect = obj.gameObject:AddComponent(typeof(UnityEngine.RectTransform))
		obj.transform:SetParent(pp)
		obj.name = "CacheRootNode"
		local canvasS = GameObject.Find("Canvas"):GetComponent("RectTransform")
		local width = Screen.width
    	local height = Screen.height
    	if width / height < 1 then
			width,height = height,width
    	end
    	local matchWidthOrHeight = MainModel.GetScene_MatchWidthOrHeight(width, height)
	    if matchWidthOrHeight == 0 then--宽适配
		    height = 1080 * (1920 / width)
		    width = 1920
	    else
	    	width = 1080 * (width / height)
		    height = 1080
	    end
	 	obj.transform.localScale = Vector3.one
	 	rect.sizeDelta = {x=width, y=height}
		this.CacheRootNode = obj.transform
	else

	end
end

-- 清除标记
function CachePrefabManager.CloseCacheTag()
	if cachemap and next(cachemap) then
		for k,v in pairs(cachemap) do
			for k1,v1 in ipairs(v) do
				v1.is_in_use = false
				v1.last_in_use_time = 0
				v1.prefab:SetParent(this.CacheRootNode)
			end
		end
	end
end

-- 创建缓冲池
function CachePrefabManager.InitCachePrefab(prefabname, count, ishide)
	CachePrefabManager.Init()
	
	-- 确保初始化被执行
	CachePrefabManager.DelCachePrefab(prefabname)
	if not cachemap[prefabname] then
		local tmp = GetPrefab(prefabname)
		if not tmp then
			print(string.format("InitCachePrefab failed, GetPrefab(%s) is nil", prefabname))
			return
		end

		local pp = GameObject.New()
		pp.transform:SetParent(this.CacheRootNode)
		local rect = pp.gameObject:AddComponent(typeof(UnityEngine.RectTransform))
		rect.sizeDelta = {x=rect_width, y=rect_height}
		pp.transform.localScale = Vector3.one
		
		pp.name = prefabname
		if ishide then
			pp:SetActive(false)
		end

		local list = {}
		cachemap[prefabname] = {}
		cachemap[prefabname].parent = pp.transform
		cachemap[prefabname].useindex = 0
		cachemap[prefabname].list = list
		cachemap[prefabname].tmp = tmp

		for i = 1, count do
			local data = {}
			data.is_in_use = false
			data.last_in_use_time = 0
			data.prefabname = prefabname
			data.index = i
			data.prefab = CachePrefab.Create(prefabname, pp.transform, tmp)
			data.prefab:SetObjName(i)
			list[#list + 1] = data
		end
		dt_cache_map[prefabname] = prefabname
	end
end
-- 删除缓冲池
function CachePrefabManager.DelCachePrefab(prefabname)
	local tbl = cachemap[prefabname]
	if tbl then
		for i = 1, tbl.useindex do
			local v = tbl.list[i]
			if v then
				v.prefab:SetParent(nil)
				v.prefab:MyExit()
			end
		end
		destroy(tbl.parent.gameObject)
		tbl.tmp = nil

		cachemap[prefabname] = nil
		dt_cache_map[prefabname] = nil
	end
end

-- 是否还有缓存
function CachePrefabManager.IsBeCache(prefabname, num)
	num = num or 1
	local data = cachemap[prefabname]
	if data and next(data.list) then
		if (data.useindex+num) <= #data.list then
			return true
		else
			return false
		end
	else
		return false
	end
end

-- 取用
function CachePrefabManager.Take(prefabname, parent, step)
	if not this or not IsEquals(this.CacheRootNode) then
		CachePrefabManager.Init()
	end
	local data = cachemap[prefabname]
	if data and next(data.list) then
		if (data.useindex+1) <= #data.list then
			data.useindex = data.useindex + 1
			local pre = data.list[data.useindex]
			pre.is_in_use = true
			return pre
		else
			local tmp = data.tmp

			if step then
				CachePrefabManager.SubstepCachePrefab(prefabname, step)
				return CachePrefabManager.Take(prefabname, parent)
			end


            --临时调用,不加入到对象池中
			local data = {}
			data.is_in_use = true
			data.last_in_use_time = 0
			data.prefabname = prefabname
			data.index = -888
			if IsEquals(parent) then
				data.prefab = CachePrefab.Create(prefabname, parent, tmp)
			else
				data.prefab = CachePrefab.Create(prefabname, this.CacheRootNode, tmp)
			end
			data.prefab:SetObjName("-888")
			return data
		end
	else
		if step then
			CachePrefabManager.SubstepCachePrefab(prefabname, step, true)
			return CachePrefabManager.Take(prefabname, parent)
		end


        --临时调用,不加入到对象池中
		local data = {}
		data.is_in_use = true
		data.last_in_use_time = 0
		data.prefabname = prefabname
		data.index = -888
		if IsEquals(parent) then
			data.prefab = CachePrefab.Create(prefabname, parent)
		else
			data.prefab = CachePrefab.Create(prefabname, this.CacheRootNode)
		end
		data.prefab:SetObjName("-888")
		return data
	end
end
-- 不强制取用，缓存没有了，返回nil
function CachePrefabManager.NoForceTake(prefabname, parent)
	if CachePrefabManager.IsBeCache(prefabname, 1) then
		return CachePrefabManager.Take(prefabname, parent)
	end
end

-- 归还
function CachePrefabManager.Back(prefab)
	if not prefab then
		return
	end
	if prefab.index < 0 then
		prefab.prefab:MyExit()
		return
	end
	local data = cachemap[prefab.prefabname]
	if data and not IsEquals(data.parent) then
		cachemap[prefab.prefabname] = nil
		return
	end
	if data and next(data.list) then
		if prefab.is_in_use then
			if prefab.index ~= data.useindex then
				data.list[data.useindex],data.list[prefab.index] = data.list[prefab.index],data.list[data.useindex]
			end
			if not data.list[prefab.index] then
				return
			end
			data.list[prefab.index].index = prefab.index
			prefab.index = data.useindex
			if not IsEquals(prefab.prefab.prefabObj) then
				return
			end
			prefab.prefab:SetObjName(data.useindex)
			prefab.is_in_use = false
			prefab.prefab:SetParent(data.parent)
			data.useindex = data.useindex - 1
		end
	end
end

-- 扩充缓冲池
function CachePrefabManager.SubstepCachePrefab(prefabname, count, ishide)
	CachePrefabManager.Init()
	
	-- 确保初始化被执行
	if not cachemap[prefabname] then
		local pp = GameObject.New()
		pp.transform:SetParent(this.CacheRootNode)
		local rect = pp.gameObject:AddComponent(typeof(UnityEngine.RectTransform))
		rect.sizeDelta = {x=rect_width, y=rect_height}
		pp.transform.localScale = Vector3.one
		
		pp.name = prefabname
		if ishide then
			pp:SetActive(false)
		end
		cachemap[prefabname] = {}
		cachemap[prefabname].parent = pp.transform
		cachemap[prefabname].list = {}
		cachemap[prefabname].useindex = 0

		dt_cache_map[prefabname] = prefabname
	end

	local ppp = cachemap[prefabname].parent
	local list = cachemap[prefabname].list
	local cur_count = #list
	local tmp = cachemap[prefabname].tmp or GetPrefab(prefabname)
	for i = 1, count do
		local data = {}
		data.is_in_use = false
		data.last_in_use_time = 0
		data.prefabname = prefabname
		data.index = i + cur_count
		data.prefab = CachePrefab.Create(prefabname, ppp, tmp)
		data.prefab:SetObjName(data.index)
		list[#list + 1] = data
	end
	cachemap[prefabname].tmp = tmp
end

-- 获取缓冲池表
function CachePrefabManager.GetCacheMap()
	return dt_cache_map
end

local basefunc = require "Game.Common.basefunc"

CachePrefab = basefunc.class()
local C = CachePrefab

function C.Create(prefabname, parent, tmp)
	return C.New(prefabname, parent, tmp)
end

function C:AddMsgListener()
    for proto_name,func in pairs(self.lister) do
        Event.AddListener(proto_name, func)
    end
end

function C:MakeLister()
    self.lister = {}
end

function C:RemoveListener()
    for proto_name,func in pairs(self.lister) do
        Event.RemoveListener(proto_name, func)
    end
    self.lister = {}
end

function C:ctor(prefabname, parent, tmp)
    if tmp then
        self.prefabObj = GameObject.Instantiate(tmp, parent)
    else
        self.prefabObj = GameObject.Instantiate(GetPrefab(prefabname), parent)
    end
    self.gameObject = self.prefabObj
end

function C:MyExit()
    destroy(self.gameObject)
end

function C:GetObj()
	return self.prefabObj
end

function C:SetObjName(name)
    if IsEquals(self.prefabObj) then
        self.prefabObj.name = name
    end
end
function C:SetParent(parent)
	if IsEquals(parent) and IsEquals(self.prefabObj) then
		self.prefabObj.transform:SetParent(parent)
	end
end