mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
72c181a399
Much of the UI was using the old budget which wasn't removed from Game like it should have been when Coaltion was introduced. The UI displayed (and in some cases pulled from) the starting budget rather than the real budget.
116 lines
3.4 KiB
Python
116 lines
3.4 KiB
Python
import itertools
|
|
from typing import Optional
|
|
|
|
from PySide2.QtWidgets import (
|
|
QDialog,
|
|
QFrame,
|
|
QGridLayout,
|
|
QLabel,
|
|
QSizePolicy,
|
|
)
|
|
|
|
import qt_ui.uiconstants as CONST
|
|
from game.game import Game
|
|
from game.income import BuildingIncome, Income
|
|
from game.theater import ControlPoint
|
|
|
|
|
|
class QHorizontalSeparationLine(QFrame):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setMinimumWidth(1)
|
|
self.setFixedHeight(20)
|
|
self.setFrameShape(QFrame.HLine)
|
|
self.setFrameShadow(QFrame.Sunken)
|
|
self.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Minimum)
|
|
|
|
|
|
class FinancesLayout(QGridLayout):
|
|
def __init__(self, game: Game, player: bool, total_at_top: bool = False) -> None:
|
|
super().__init__()
|
|
self.row = itertools.count(0)
|
|
|
|
income = Income(game, player)
|
|
|
|
if total_at_top:
|
|
self.add_total(game, income, player)
|
|
self.add_line()
|
|
|
|
control_points = reversed(
|
|
sorted(income.control_points, key=lambda c: c.income_per_turn)
|
|
)
|
|
for control_point in control_points:
|
|
self.add_control_point(control_point)
|
|
|
|
self.add_line()
|
|
|
|
buildings = reversed(sorted(income.buildings, key=lambda b: b.income))
|
|
for building in buildings:
|
|
self.add_building(building)
|
|
|
|
if not total_at_top:
|
|
self.add_line()
|
|
self.add_total(game, income, player)
|
|
|
|
def add_total(self, game, income, player):
|
|
self.add_row(
|
|
middle=f"Income multiplier: {income.multiplier:.1f}",
|
|
right=f"<b>{income.total}M</b>",
|
|
)
|
|
budget = game.coalition_for(player).budget
|
|
self.add_row(middle="Balance", right=f"<b>{budget}M</b>")
|
|
self.setRowStretch(next(self.row), 1)
|
|
|
|
def add_row(
|
|
self,
|
|
left: Optional[str] = None,
|
|
middle: Optional[str] = None,
|
|
right: Optional[str] = None,
|
|
) -> None:
|
|
if not any([left, middle, right]):
|
|
raise ValueError
|
|
|
|
row = next(self.row)
|
|
if left is not None:
|
|
self.addWidget(QLabel(left), row, 0)
|
|
if middle is not None:
|
|
self.addWidget(QLabel(middle), row, 1)
|
|
if right is not None:
|
|
self.addWidget(QLabel(right), row, 2)
|
|
|
|
def add_control_point(self, control_point: ControlPoint) -> None:
|
|
self.add_row(
|
|
left=f"<b>{control_point.name}</b>",
|
|
right=f"{control_point.income_per_turn}M",
|
|
)
|
|
|
|
def add_building(self, building: BuildingIncome) -> None:
|
|
row = next(self.row)
|
|
self.addWidget(
|
|
QLabel(f"<b>{building.category.upper()} [{building.name}]</b>"), row, 0
|
|
)
|
|
self.addWidget(
|
|
QLabel(f"{building.number} buildings x {building.income_per_building}M"),
|
|
row,
|
|
1,
|
|
)
|
|
rlabel = QLabel(f"{building.income}M")
|
|
rlabel.setProperty("style", "green")
|
|
self.addWidget(rlabel, row, 2)
|
|
|
|
def add_line(self) -> None:
|
|
self.addWidget(QHorizontalSeparationLine(), next(self.row), 0, 1, 3)
|
|
|
|
|
|
class QFinancesMenu(QDialog):
|
|
def __init__(self, game: Game):
|
|
super(QFinancesMenu, self).__init__()
|
|
|
|
self.game = game
|
|
self.setModal(True)
|
|
self.setWindowTitle("Finances")
|
|
self.setWindowIcon(CONST.ICONS["Money"])
|
|
self.setMinimumSize(450, 200)
|
|
|
|
self.setLayout(FinancesLayout(game, player=True))
|