mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
2686a1ea77
Moves the stretch to the bottom of the page to avoid awkward whitespace in the middle. Presumably the totals used to be at the bottom (since that's a normal place for a total), but it was moved to the top, probably since that was the most interesting data and we didn't want to scroll though all the details to find that one point. This also removes the unused code path where the total would be shown at the bottom. Fixes https://github.com/dcs-liberation/dcs_liberation/issues/1288.
112 lines
3.2 KiB
Python
112 lines
3.2 KiB
Python
import itertools
|
|
from typing import Optional
|
|
|
|
from PySide6.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) -> None:
|
|
super().__init__()
|
|
self.row = itertools.count(0)
|
|
|
|
income = Income(game, player)
|
|
|
|
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)
|
|
|
|
self.setRowStretch(next(self.row), 1)
|
|
|
|
def add_total(self, game, income, player):
|
|
self.add_row(
|
|
middle=f"Income multiplier: {income.multiplier:.1f}",
|
|
right=f"<b>{income.total:.1f}M</b>",
|
|
)
|
|
budget = game.coalition_for(player).budget
|
|
self.add_row(middle="Balance", right=f"<b>{budget:.1f}M</b>")
|
|
|
|
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))
|