Added shieldball healthbars widget

LuaUI/Widgets/gui_shieldball_healthbars.lua

@@ -0,0 +1,283 @@
+function widget:GetInfo()
+	return {
+		name         = "Shieldball healthbars UI",
+		desc         = "Give a healthbar to shieldballs. Version 1.0",
+		author       = "dyth68",
+		date         = "2024",
+		license      = "PD", -- should be compatible with Spring
+		layer        = 11,
+		enabled      = true
+	}
+end
+
+local UPDATE_FRAME=7
+local glCallList = gl.CallList
+local glPushMatrix = gl.PushMatrix
+local glTranslate = gl.Translate
+local glRotate = gl.Rotate
+local glScale = gl.Scale
+local glPopMatrix = gl.PopMatrix
+local glCallList = gl.CallList
+local glCreateList = gl.CreateList
+local glDeleteList = gl.DeleteList
+local glVertex = gl.Vertex
+local glPolygonMode = gl.PolygonMode
+local glBeginEnd = gl.BeginEnd
+local glLineWidth = gl.LineWidth
+local glColor = gl.Color
+local spGetUnitPosition = Spring.GetUnitPosition
+local spGetUnitShieldState = Spring.GetUnitShieldState
+local spGetUnitsInCylinder = Spring.GetUnitsInCylinder
+local spGetUnitDefID = Spring.GetUnitDefID
+local spGetUnitHealth = Spring.GetUnitHealth
+
+local Optics = VFS.Include("LuaRules/Gadgets/Include/Optics.lua")
+---------------------------------
+local function printThing(theTable, theKey, indent)
+	local indent = indent or ""
+	local theKey = theKey or "key"
+	if (type(theTable) == "table") then
+		Spring.Echo(indent .. theKey .. ":")
+		for a, b in pairs(theTable) do
+			printThing(b, tostring(a), indent .. "  ")
+		end
+	else
+		Spring.Echo(indent .. tostring(theKey) .. ": " .. tostring(theTable))
+	end
+end
+---------------------------------
+local shieldUnitDefs = {}
+
+
+for unitDefID=1, #UnitDefs do
+	if UnitDefs[unitDefID].shieldWeaponDef then
+		shieldUnitDefs[unitDefID] = true
+	end
+end
+---------------------------------
+
+local BASE_FONT_SIZE = 192
+local healthBarOffsetZ = 300
+local healthBarHeight = 30
+local healthBarWidth = 200
+local healthBarMargin = 5
+local healthBarRoundiness = 5
+---------------------------------
+
+-- Map from teamID to array of balls, each ball with location, curr power and total power
+local shieldBalls = {}
+local drawBallHealthbarList
+
+local font = gl.LoadFont("FreeSansBold.otf", BASE_FONT_SIZE, 0, 0)
+
+local function DrawHullVertices(hull)
+	for j = 1, #hull do
+		glVertex(hull[j].x, hull[j].y, hull[j].z)
+	end
+end
+
+local function DrawBallHealthbar()
+	for teamID, teamBalls in pairs(shieldBalls) do
+		for _, shieldBall in pairs(teamBalls) do
+			glColor(0.8, 0.8, 0.8, 0.5)
+			glPolygonMode(GL.FRONT_AND_BACK, GL.FILL)
+			local healthBarHeightAboveTheZero = shieldBall.highestTopOfShield + 40
+			local healthBarOffsetZ = shieldBall.zStdDev * 2 + 60
+			-- The outer bar is actually an Octagon to give the impression of rounded corners
+			local healthBarRoundedCorners = {
+				{x = shieldBall.x - healthBarWidth + healthBarRoundiness, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + healthBarHeight)},
+				{x = shieldBall.x + healthBarWidth - healthBarRoundiness, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + healthBarHeight)},
+				{x = shieldBall.x + healthBarWidth, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + healthBarHeight - healthBarRoundiness)},
+				{x = shieldBall.x + healthBarWidth, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + healthBarRoundiness)},
+				{x = shieldBall.x + healthBarWidth - healthBarRoundiness, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ)},
+				{x = shieldBall.x - healthBarWidth + healthBarRoundiness, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ)},
+				{x = shieldBall.x - healthBarWidth, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + healthBarRoundiness)},
+				{x = shieldBall.x - healthBarWidth, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + healthBarHeight - healthBarRoundiness)},
+			}
+			glBeginEnd(GL.TRIANGLE_FAN, DrawHullVertices, healthBarRoundedCorners)
+			glColor(0.5, 0.5, 1.0, 0.5)
+			local innerBarWidthWhenFull = healthBarWidth - healthBarMargin
+			local innerBarHeight = healthBarHeight - healthBarMargin
+			local proportionFull = shieldBall.currShield / shieldBall.totShield
+			local innerBarWidth = 2 * innerBarWidthWhenFull * proportionFull
+			local innerHealthBarCorners = {
+				{ x = shieldBall.x - innerBarWidthWhenFull, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + innerBarHeight)},
+				{ x = shieldBall.x - innerBarWidthWhenFull + innerBarWidth, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + innerBarHeight)},
+				{ x = shieldBall.x - innerBarWidthWhenFull + innerBarWidth, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + healthBarMargin)},
+				{ x = shieldBall.x - innerBarWidthWhenFull, y = healthBarHeightAboveTheZero, z = shieldBall.z - (healthBarOffsetZ + healthBarMargin)},
+			}
+			glBeginEnd(GL.TRIANGLE_FAN, DrawHullVertices, innerHealthBarCorners)
+			local fontSize = BASE_FONT_SIZE / 10
+			--glScale(fontSize / BASE_FONT_SIZE, fontSize / BASE_FONT_SIZE, fontSize / BASE_FONT_SIZE)
+			glPushMatrix()
+			glTranslate(shieldBall.x, healthBarHeightAboveTheZero, shieldBall.z - healthBarOffsetZ - innerBarHeight/2 - healthBarMargin/2)
+			glRotate(-90, 1, 0, 0)
+			local shieldHpText = tostring(math.floor(shieldBall.currShield)) .. " / " .. tostring(shieldBall.totShield)
+			font:Begin()
+				font:SetTextColor(0.2, 0.2, 0.2, 0.6)
+				font:Print(shieldHpText, 0, 0, fontSize, "cv")
+			font:End()
+			glPopMatrix()
+		end
+	end
+end
+---------------------------------
+
+ -- TODO: Use link data
+local function shieldsAreTouching(shield1, shield2)
+	if not shield2 then
+		return false
+	end
+	local xDiff = shield1.x - shield2.x
+	local zDiff = shield1.z - shield2.z
+	local yDiff = shield1.y - shield2.y
+	local sumRadius = shield1.shieldRadius + shield2.shieldRadius
+	return xDiff <= sumRadius and zDiff <= sumRadius and (xDiff*xDiff + yDiff*yDiff + zDiff*zDiff) < sumRadius*sumRadius
+end
+
+---------------------------------
+
+local shieldBallsIdsByTeam = {}
+local allShieldUnitsByTeam = {}
+
+local function updateClustering()
+	for _, teamID in pairs(Spring.GetTeamList()) do
+		local allUnits = Spring.GetTeamUnits(teamID)
+		local allShieldUnits = {}
+		for _,unitID in pairs(allUnits) do
+			local unitDefID = spGetUnitDefID(unitID)
+			if unitDefID and shieldUnitDefs[unitDefID] then
+				local _, _, _, _, buildProgress = spGetUnitHealth(unitID)
+				if buildProgress == 1 then
+					local shieldWep = WeaponDefs[UnitDefs[unitDefID].shieldWeaponDef]
+					local x,y,z = spGetUnitPosition(unitID)
+					if x then
+						allShieldUnits[unitID] = {
+							shieldMaxCharge  = shieldWep.shieldPower,
+							shieldRadius = shieldWep.shieldRadius,
+							x = x,
+							y = y,
+							z = z
+						}
+					end
+					--printThing(allShieldUnits[unitID], "shieldDef", "")
+				end
+			end
+		end
+		local unitLocations = {}
+		local unitNeighborsMatrix = {}
+		for unitID, shieldProps in pairs(allShieldUnits) do
+			local x,y,z = spGetUnitPosition(unitID)
+			unitLocations[#unitLocations + 1] = {
+				x = x,
+				z = z,
+				fID = unitID,
+			}
+			local unitsInRange = spGetUnitsInCylinder(x, z, shieldProps.shieldRadius*2)
+			unitNeighborsMatrix[unitID] = {}
+			for _, unitInRange in ipairs(unitsInRange) do
+				if shieldsAreTouching(shieldProps, allShieldUnits[unitInRange]) then
+					unitNeighborsMatrix[unitID][unitInRange] = true
+					if not unitNeighborsMatrix[unitInRange] then
+						unitNeighborsMatrix[unitInRange] = {}
+					end
+					unitNeighborsMatrix[unitInRange][unitID] = true
+				end
+			end
+		end
+		--printThing(unitLocations, "unitLocations", "")
+		--printThing(unitNeighborsMatrix, "unitNeighborsMatrix", "")
+		local opticsObject = Optics.new(unitLocations, unitNeighborsMatrix, 2, false)
+		opticsObject:Run()
+		shieldBallsIdsByTeam[teamID] = opticsObject:Clusterize(700)
+		allShieldUnitsByTeam[teamID] = allShieldUnits
+	end
+end
+
+---------------------------------
+local function updateCurrentShieldBalls()
+	for _, teamID in pairs(Spring.GetTeamList()) do
+		local shieldBallsIds = shieldBallsIdsByTeam[teamID] or {}
+		local allShieldUnits = allShieldUnitsByTeam[teamID] or {}
+		local ballsInTeam = {}
+		for i = 1, #shieldBallsIds do
+			local thisBall = shieldBallsIds[i]
+			local totalCurrentShield = 0
+			local totalMaxShield = 0
+			local x_avg, z_avg = 0, 0
+			local numUnits = 0
+			local highestTopOfShield = 0
+			local memberPositionsByUnitID = {}
+			for j = 1, #thisBall.members do
+				local unitID = thisBall.members[j]
+				local x,y,z = spGetUnitPosition(unitID)
+				if x then
+					local shieldProps = allShieldUnits[unitID]
+					totalMaxShield = totalMaxShield + shieldProps.shieldMaxCharge
+					local enabled, currPower = spGetUnitShieldState(unitID)
+					x_avg = x_avg + x
+					z_avg = z_avg + z
+					numUnits = numUnits + 1
+					if enabled then
+						totalCurrentShield = totalCurrentShield + currPower
+						highestTopOfShield = math.max(highestTopOfShield, y + shieldProps.shieldRadius)
+					end
+				end
+				memberPositionsByUnitID[unitID] = {x = x, y = y, z = z}
+			end
+			if numUnits > 0 then
+				x_avg = x_avg / numUnits
+				z_avg = z_avg / numUnits
+			end
+
+			local xStdDev = 0
+			local zStdDev = 0
+			for j = 1, #thisBall.members do
+				local unitID = thisBall.members[j]
+				local x = memberPositionsByUnitID[unitID].x
+				local z = memberPositionsByUnitID[unitID].z
+				if x then
+					xStdDev = xStdDev + (x - x_avg) * (x - x_avg)
+					zStdDev = zStdDev + (z - z_avg) * (z - z_avg)
+				end
+			end
+			xStdDev = math.sqrt(xStdDev / numUnits)
+			zStdDev = math.sqrt(zStdDev / numUnits)
+
+			local ballData = {
+				currShield = totalCurrentShield,
+				totShield = totalMaxShield,
+				x = x_avg,
+				z = z_avg,
+				highestTopOfShield = highestTopOfShield,
+				xStdDev = xStdDev,
+				zStdDev = zStdDev,
+				numUnits = numUnits,
+			}
+			ballsInTeam[#ballsInTeam + 1] = ballData
+		end
+
+
+		shieldBalls[teamID] = ballsInTeam
+		drawBallHealthbarList = glCreateList(DrawBallHealthbar)
+	end
+end
+---------------------------------
+
+function widget:GameFrame(n)
+	if (n%UPDATE_FRAME==0) then
+		updateClustering()
+	end
+	updateCurrentShieldBalls()
+end
+
+function widget:Initialize()
+	updateClustering()
+	updateCurrentShieldBalls()
+end
+
+function widget:DrawWorld()
+	if drawBallHealthbarList then
+		glCallList(drawBallHealthbarList)
+	end
+end