From 1ac413e1d0ddc39a41374ff9b3375e6de3e1af1c Mon Sep 17 00:00:00 2001
From: Florian Fischer <florian.fischer@muhq.space>
Date: Tue, 14 Feb 2023 09:52:22 +0100
Subject: improve state bar

---
 go/game/player.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'go/game')

diff --git a/go/game/player.go b/go/game/player.go
index cb89e1ae..78b4637e 100644
--- a/go/game/player.go
+++ b/go/game/player.go
@@ -52,6 +52,21 @@ func (p *Player) resourceGain() int {
 	}
 }
 
+func (p *Player) UpkeepGain() int {
+	return p.resourceGain() - p.UpkeepCost()
+}
+
+func (p *Player) UpkeepCost() int {
+	cost := 0
+	for _, u := range p.gameState.Units {
+		if u.Controller() != p {
+			continue
+		}
+		cost += u.Upkeep
+	}
+	return cost
+}
+
 func (p *Player) gainResource(gain int) {
 	p.Resource += gain
 }
-- 
cgit v1.2.3