mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
import logging
|
|
from typing import Callable, Dict, TypeVar
|
|
|
|
from PySide2.QtGui import QIcon, QPixmap
|
|
from PySide2.QtWidgets import (
|
|
QDialog,
|
|
QGridLayout,
|
|
QGroupBox,
|
|
QLabel,
|
|
QPushButton,
|
|
QVBoxLayout,
|
|
)
|
|
|
|
from game import db
|
|
from game.debriefing import Debriefing
|
|
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
class LossGrid(QGridLayout):
|
|
def __init__(self, debriefing: Debriefing, player: bool) -> None:
|
|
super().__init__()
|
|
|
|
if player:
|
|
country = debriefing.player_country
|
|
else:
|
|
country = debriefing.enemy_country
|
|
|
|
self.add_loss_rows(
|
|
debriefing.air_losses.by_type(player),
|
|
lambda u: db.unit_get_expanded_info(country, u, "name"),
|
|
)
|
|
self.add_loss_rows(
|
|
debriefing.front_line_losses_by_type(player),
|
|
lambda u: db.unit_type_name(u),
|
|
)
|
|
self.add_loss_rows(
|
|
debriefing.convoy_losses_by_type(player),
|
|
lambda u: f"{db.unit_type_name(u)} from convoy",
|
|
)
|
|
self.add_loss_rows(
|
|
debriefing.cargo_ship_losses_by_type(player),
|
|
lambda u: f"{db.unit_type_name(u)} from cargo ship",
|
|
)
|
|
self.add_loss_rows(
|
|
debriefing.airlift_losses_by_type(player),
|
|
lambda u: f"{db.unit_type_name(u)} from airlift",
|
|
)
|
|
self.add_loss_rows(
|
|
debriefing.building_losses_by_type(player),
|
|
lambda u: u,
|
|
)
|
|
|
|
# TODO: Display dead ground object units and runways.
|
|
|
|
def add_loss_rows(self, losses: Dict[T, int], make_name: Callable[[T], str]):
|
|
for unit_type, count in losses.items():
|
|
row = self.rowCount()
|
|
try:
|
|
name = make_name(unit_type)
|
|
except AttributeError:
|
|
logging.exception(f"Could not make unit name for {unit_type}")
|
|
name = unit_type.id
|
|
self.addWidget(QLabel(name), row, 0)
|
|
self.addWidget(QLabel(str(count)), row, 1)
|
|
|
|
|
|
class QDebriefingWindow(QDialog):
|
|
def __init__(self, debriefing: Debriefing):
|
|
super(QDebriefingWindow, self).__init__()
|
|
self.debriefing = debriefing
|
|
|
|
self.setModal(True)
|
|
self.setWindowTitle("Debriefing")
|
|
self.setMinimumSize(300, 200)
|
|
self.setWindowIcon(QIcon("./resources/icon.png"))
|
|
|
|
layout = QVBoxLayout()
|
|
self.setLayout(layout)
|
|
|
|
header = QLabel(self)
|
|
header.setGeometry(0, 0, 655, 106)
|
|
pixmap = QPixmap("./resources/ui/debriefing.png")
|
|
header.setPixmap(pixmap)
|
|
layout.addWidget(header)
|
|
layout.addStretch()
|
|
|
|
title = QLabel("<b>Casualty report</b>")
|
|
layout.addWidget(title)
|
|
|
|
player_lost_units = QGroupBox(f"{self.debriefing.player_country}'s lost units:")
|
|
player_lost_units.setLayout(LossGrid(debriefing, player=True))
|
|
layout.addWidget(player_lost_units)
|
|
|
|
enemy_lost_units = QGroupBox(f"{self.debriefing.enemy_country}'s lost units:")
|
|
enemy_lost_units.setLayout(LossGrid(debriefing, player=False))
|
|
layout.addWidget(enemy_lost_units)
|
|
|
|
okay = QPushButton("Okay")
|
|
okay.clicked.connect(self.close)
|
|
layout.addWidget(okay)
|