Spring RTS Engine

Code: Select all

function gadget:GetInfo()
	return {
		name = "supplies",
		desc = "manage fuel and ammo supplies",
		author = "thesleepless",
		date = "Dec 29, 2008",
		license = "Public Domain",
		layer = 0,
		enabled = true
	}
end

if(not gadgetHandler:IsSyncedCode()) then
	return
else
	local ammo = {}
	local fuel = {}
	local outOfFuel = {}
	local outOfAmmo = {}
	local maxSpeeds = {}
	local myTeam = Spring.GetMyTeamID()
end

function gadget:Initialize()
end

function gadget:UnitCreated(unitID, unitDefID, teamID)
	local ud = UnitDefs[unitDefID]
	local unitName = ud.name
	ammo[unitID] = ud.startingAmmo
	fuel[unitID] = ud.startingFuel
end

function gadget:UnitDestroyed(unitID, _, teamID)
	if(ammo[unitID]) then
		ammo[unitID] = nil
	end
	if(fuel[unitID]) then
		fuel[unitID] = nil
	end
end

-- supplies units around the supply unit witn supplies
function UnitSupplyUnits(sUnitID)
	-- get all friendly units within radius
	local posx, posy, posz = Spring.GetUnitPosition(sUnitID)
	local ud = UnitDefs[Spring.GetUnitDefID(unitID)]
	local units = Spring.GetUnitsInCylinder(posx, posz, ud.supplyRadius)
	for unitID,val in pairs(units) do
		if(Spring.GetUnitTeam(unitID) == Spring.GetUnitTeam(sUnitID)) then
			if(ammo[unitID] < ud.fuelCapacity and ammo[sUnitID] >= 1) then
				ammo[unitID] = ammo[unitID] + 1
				ammo[sUnitID] = ammo[sUnitID] - 1
			end
			if(ud.canMove) then
				if(fuel[unitID] < ud.fuelCapacity and fuel[sUnitID] >= 1) then
					if(outOfFuel[unitID]) then
						outOfFuel[unitID] = nil
						Spring.SetUnitCOBValue("MAX_SPEED", maxSpeeds[unitID])
					end
					fuel[unitID] = fuel[unitID] + 1
					fuel[sUnitID] = fuel[sUnitID] - 1
				end
			end
		end
	end
end

function gadet:GameFrame(n)
	if n%30 ~= 0 then
		return
	end
	-- subtract fuel based on movement
	local units = Spring.GetAllUnits()
	for i=1, #units do
		local unitID = units[i]
		local ud = UnitDefs[Spring.GetUnitDefID(unitID)]
		if(ud.canMove) then
			if(not fuel[unitID]) then
				fuel[unitID] = 0
			end
			-- get speed of unit
			local sx, sy, sz = Spring.GetUnitVelocity(unitID)
			local speed = math.sqrt(sx * sx + sy * sy + sz * sz)
			if(fuel[unitID] > 0) then
				fuel[unitID] = fuel[unitID] - (ud.fuelUsage * speed * 30)
			end
			if( not outOfFuel[unitID] and fuel[unitID] <= 0) then
				fuel[unitID] = 0
				outOfFuel[unitID] = true
				maxSpeeds[unitID] = Spring.GetUnitCOBValue(unitID, "MAX_SPEED")
				Spring.SetUnitCOBValue(unitID, "MAX_SPEED", 0)
				if(Spring.GetUnitTeam(unitID) == myTeam) then
					Spring.Echo(ud.name .. " is out of fuel")
				end
			end
		end
	end
end

function gadget:ProjectileCreated(projID, ownerID)
	if(ammo[ownerID]) then
		ammo[ownerID] = ammo[ownerID] - 1
	end
	if(ammo[ownerID] <= 0) then
		outOfAmmo[ownerID] = true
		local ud = UnitDefs[Spring.GetUnitDefID(ownerID)]
		if(Spring.GetUnitTeam(ownerID) == myTeam) then
			Spring.Echo(ud.name .. " is out of ammo")
		end
		-- need to diable the unit from firing somehow
	end
end

Code: Select all

         local sx, sy, sz = Spring.GetUnitVelocity(unitID)
         local speed = math.sqrt(sx * sx + sy * sy + sz * sz)
         if(fuel[unitID] > 0) then
            fuel[unitID] = fuel[unitID] - (ud.fuelUsage * speed * 30)
         end

Code: Select all

lua_IamUsingFuelNow(arg) { return(0); }

StartMoving()
call-script lua_IamUsingFuelNow();
;