Spring RTS Engine

Code: Select all

function widget:GetInfo()
  return {
    name      = "Visible Units",
    desc      = "Speedup for VisibleUnits() callin",
    author    = "Argh",
    date      = "August 23, 2009",
    license   = "Public Domain, or the least-restrictive rights of your country of residence",
    layer     = 0,
    enabled   = true -- loaded by default?
  }
end

local GetDrawFrame = Spring.GetDrawFrame
local GetVisibleUnits = Spring.GetVisibleUnits
local ALL_UNITS       = Spring.ALL_UNITS
local frame, oldframe = 0,0
WG.visible = {}

function widget:DrawWorld()
	frame = GetDrawFrame()
	if frame > oldframe then
		--Spring.Echo("running now",frame)
		WG.visible = GetVisibleUnits(ALL_UNITS,nil,false)
		oldframe = frame + 10		
	end
end

Code: Select all

function widget:DrawWorldPreUnit()
--Set up basic OpenGL parameters here
	for i=1,#WG.visible do
		local unitID = WG.visible[i]
		id = GetUnitDefID(unitID)
		if UsesReallyCoolFX[id] then
			--display lists, shaders, whatever here
		end				
	end
  	glResetState()
end

Code: Select all

function gadget:GetInfo()
  return {
    name      = "00000Visible Units (Gadget Version)",
    desc      = "Speedup for VisibleUnits() callin",
    author    = "Argh",
    date      = "December 21, 2009",
    license   = "(C) Wolfe Games, 2009, released under GPL v.2",
    layer     = 0,
    enabled   = true -- loaded by default?
  }
end

if (gadgetHandler:IsSyncedCode()) then
else

local oldframe = 0
GG.visible = {}

function gadget:DrawGenesis()
	local ALL_UNITS = Spring.ALL_UNITS
	local GetGameFrame = Spring.GetGameFrame
	local GetVisibleUnits = Spring.GetVisibleUnits
	local GetLocalTeamID = Spring.GetLocalTeamID
	local GetUnitLosState = Spring.GetUnitLosState

	local myTeam = GetLocalTeamID()
	local frame = GetGameFrame()

	if frame > oldframe then
		GG.visible = {}
		local losState
		local visTable = GetVisibleUnits(ALL_UNITS)
		for i,k in ipairs (visTable) do
			losState = Spring.GetUnitLosState(k,myTeam)
			losState = losState.los
			--Spring.Echo(losState)
			if losState == true then
				table.insert(GG.visible,k)
			end
		end
		local oldframe = frame + 5		
	end
end
end

Code: Select all

function gadget:GetInfo()
  return {
    ...
  }
end

if (gadgetHandler:IsSyncedCode()) then
	return
end

local ALL_UNITS = Spring.ALL_UNITS
local GetGameFrame = Spring.GetGameFrame
local GetLocalTeamID = Spring.GetLocalTeamID
local GetUnitLosState = Spring.GetUnitLosState
local OrigGetVisibleUnits = Spring.GetVisibleUnits

local uupdateEachSec = 5
local oldframe = 0
local bufVisibleLists = {}

local tconcat = table.concat
local function ArgumentsToTableIndex(...)
	return tconcat({...})
end

local function UpdateVisUnits(...)
	local tidx = ArgumentsToTableIndex(...)
	bufVisibleLists[tidx] = OrigGetVisibleUnits(...)
end

--// please note, that the following code needs to get executed BEFORE any other gadgets which use those functions! (-> either give the file a smart name or modify the gadgetHandler to load the gadget right at the beginning)
Spring.GetVisibleUnits = function (...)
  	local tidx = ArgumentsToTableIndex(...)
  	local bufList = bufVisibleLists[tidx]

	local frame = GetGameFrame()
  	if (not bufList)or(frame > bufList.oldframe) then
  	  	CallAsTeam(GetLocalTeamID(),UpdateVisUnits,...)
  	  	bufList = bufVisibleLists[tidx]
  	  	bufList.oldframe = frame + updateEachSec
  	end

  	return bufList
end

Code: Select all

Spring.GetVisibleUnits = function (...)
  	...
  	if (not bufList)or(frame > bufList.oldframe) then
  	  	--vvvvvvvv
  	  	CallAsTeam(GetLocalTeamID(),UpdateVisUnits,...)
  	  	...
  	end

  	return bufList
end

Code: Select all

function gadget:GetInfo()
  return {
    ...
  }
end

if (gadgetHandler:IsSyncedCode()) then
	return
end

local ALL_UNITS = Spring.ALL_UNITS
local CallAsTeam = CallAsTeam
local spGetLocalTeamID = Spring.GetLocalTeamID
local spGetTimer = Spring.GetTimer
local spDiffTimers = Spring.DiffTimers
local OrigGetVisibleUnits = Spring.GetVisibleUnits

local updateEachSec = 0.2 --//in seconds!
local oldframe = 0
local bufVisibleLists = {}

local tconcat = table.concat
local function ArgumentsToTableIndex(...)
	return tconcat({...})
end

local function UpdateVisUnits(...)
	local tidx = ArgumentsToTableIndex(...)
	bufVisibleLists[tidx] = OrigGetVisibleUnits(...)
end

--// please note, that the following code needs to get executed BEFORE any other gadgets which use those functions! (-> either give the file a smart name or modify the gadgetHandler to load the gadget right at the beginning)
Spring.GetVisibleUnits = function (...)
  	local tidx = ArgumentsToTableIndex(...)
  	local bufList = bufVisibleLists[tidx]

	local curTime = spGetTimer()
  	if (not bufList)or
  	   (spDiffTimers(curTime, bufList.oldtime) > updateEachSec)
  	then
  	  	CallAsTeam(spGetLocalTeamID(),UpdateVisUnits,...)
  	  	bufList = bufVisibleLists[tidx]
  	  	bufList.oldtime = curTime
  	end

  	return bufList
end