diff --git a/Assets/GWConquest/Scripts/UI/BalanceUI.cs b/Assets/GWConquest/Scripts/UI/BalanceUI.cs index 28969ad..b6f6e51 100644 --- a/Assets/GWConquest/Scripts/UI/BalanceUI.cs +++ b/Assets/GWConquest/Scripts/UI/BalanceUI.cs @@ -24,6 +24,11 @@ namespace GWConquest int totalSupplies = 0; int totalFuel = 0; + float balanceFood = 0; + float balanceSupplies = 0; + float balanceFuel = 0; + float balanceCredits = 0; + foreach(var d in District.AllDistricts) { if(d.ControllingPlayer == currentPlayer) @@ -31,6 +36,25 @@ namespace GWConquest totalFood += d.Inventory.GetItemAmount("Food"); totalSupplies += d.Inventory.GetItemAmount("Supplies"); totalFuel += d.Inventory.GetItemAmount("Fuel"); + + foreach(var (item, amount) in d.ItemProductionList) + { + switch(item) + { + case "Food": + balanceFood += amount; + break; + case "Supplies": + balanceSupplies += amount; + break; + case "Fuel": + balanceFuel += amount; + break; + case "Credits": + balanceCredits += amount; + break; + } + } } } @@ -41,6 +65,11 @@ namespace GWConquest FoodDisplay.Total = totalFood; SuppliesDisplay.Total = totalSupplies; FuelDisplay.Total = totalFuel; + + FoodDisplay.Balance = Mathf.RoundToInt(balanceFood); + SuppliesDisplay.Balance = Mathf.RoundToInt(balanceSupplies); + FuelDisplay.Balance = Mathf.RoundToInt(balanceFuel); + CreditsDisplay.Balance = Mathf.RoundToInt(balanceCredits); } }