mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
140 lines
4.9 KiB
Python
140 lines
4.9 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from PySide2.QtCore import Qt
|
|
from PySide2.QtGui import QIcon, QPixmap
|
|
from PySide2.QtWidgets import (
|
|
QDialog,
|
|
QGridLayout,
|
|
QLabel,
|
|
QTextBrowser,
|
|
QFrame,
|
|
)
|
|
|
|
from game.dcs.aircrafttype import AircraftType
|
|
from game.dcs.groundunittype import GroundUnitType
|
|
from game.dcs.unittype import UnitType
|
|
from game.game import Game
|
|
|
|
AIRCRAFT_BANNERS_BASE = Path("resources/ui/units/aircrafts/banners")
|
|
VEHICLE_BANNERS_BASE = Path("resources/ui/units/vehicles/banners")
|
|
MISSING_BANNER_PATH = AIRCRAFT_BANNERS_BASE / "Missing.jpg"
|
|
|
|
|
|
def aircraft_banner_for(unit_type: AircraftType) -> Path:
|
|
if unit_type.dcs_id in {
|
|
"Mirage-F1CT",
|
|
"Mirage-F1EE",
|
|
"Mirage-F1M-EE",
|
|
"Mirage-F1EQ",
|
|
}:
|
|
name = "Mirage-F1C-200"
|
|
elif unit_type.dcs_id in {"Mirage-F1CE", "Mirage-F1M-CE"}:
|
|
name = "Mirage-F1C"
|
|
elif unit_type.dcs_id in {"Su-30MKA", "Su-30MKI", "Su-30MKM"}:
|
|
name = "Su-30SM"
|
|
elif unit_type.dcs_id == "F-15ESE":
|
|
name = "F-15E"
|
|
else:
|
|
name = unit_type.dcs_id
|
|
return AIRCRAFT_BANNERS_BASE / f"{name}.jpg"
|
|
|
|
|
|
def vehicle_banner_for(unit_type: GroundUnitType) -> Path:
|
|
if unit_type.dcs_id == "(IDF Mods Project) BM-21 Grad 122mm":
|
|
return VEHICLE_BANNERS_BASE / "Grad-URAL.jpg"
|
|
elif unit_type.dcs_id == "(IDF Mods Project) Urgan BM-27 220mm":
|
|
return VEHICLE_BANNERS_BASE / "Uragan_BM-27.jpg"
|
|
elif unit_type.dcs_id == "(IDF Mods Project) 9A52 Smerch CM 300mm":
|
|
return VEHICLE_BANNERS_BASE / "Smerch_HE.jpg"
|
|
else:
|
|
return VEHICLE_BANNERS_BASE / f"{unit_type.dcs_id}.jpg"
|
|
|
|
|
|
def banner_path_for(unit_type: UnitType) -> Path:
|
|
if isinstance(unit_type, AircraftType):
|
|
return aircraft_banner_for(unit_type)
|
|
if isinstance(unit_type, GroundUnitType):
|
|
return vehicle_banner_for(unit_type)
|
|
raise NotImplementedError(f"Unhandled UnitType subclass: {unit_type.__class__}")
|
|
|
|
|
|
class QUnitInfoWindow(QDialog):
|
|
def __init__(self, game: Game, unit_type: UnitType) -> None:
|
|
super().__init__()
|
|
self.setModal(True)
|
|
self.game = game
|
|
self.unit_type = unit_type
|
|
self.name = unit_type.name
|
|
self.setWindowTitle(f"Unit Info: {self.name}")
|
|
self.setWindowIcon(QIcon("./resources/icon.png"))
|
|
self.setMinimumHeight(570)
|
|
self.setMaximumWidth(640)
|
|
self.setWindowFlags(Qt.WindowStaysOnTopHint)
|
|
|
|
self.layout = QGridLayout()
|
|
|
|
header = QLabel(self)
|
|
header.setGeometry(0, 0, 720, 360)
|
|
|
|
banner_path = banner_path_for(unit_type)
|
|
if not banner_path.exists():
|
|
banner_path = MISSING_BANNER_PATH
|
|
pixmap = QPixmap(banner_path.as_posix())
|
|
header.setPixmap(pixmap.scaled(header.width(), header.height()))
|
|
self.layout.addWidget(header, 0, 0)
|
|
|
|
self.gridLayout = QGridLayout()
|
|
|
|
# Build the topmost details grid.
|
|
self.details_grid = QFrame()
|
|
self.details_grid_layout = QGridLayout()
|
|
self.details_grid_layout.setMargin(0)
|
|
|
|
self.name_box = QLabel(
|
|
f"<b>Name:</b> {unit_type.manufacturer} {unit_type.name}"
|
|
)
|
|
self.name_box.setProperty("style", "info-element")
|
|
|
|
self.country_box = QLabel(
|
|
f"<b>Country of Origin:</b> {unit_type.country_of_origin}"
|
|
)
|
|
self.country_box.setProperty("style", "info-element")
|
|
|
|
self.role_box = QLabel(f"<b>Role:</b> {unit_type.role}")
|
|
self.role_box.setProperty("style", "info-element")
|
|
|
|
self.year_box = QLabel(
|
|
f"<b>Variant Introduction:</b> {unit_type.year_introduced}"
|
|
)
|
|
self.year_box.setProperty("style", "info-element")
|
|
|
|
self.details_grid_layout.addWidget(self.name_box, 0, 0)
|
|
self.details_grid_layout.addWidget(self.country_box, 0, 1)
|
|
self.details_grid_layout.addWidget(self.role_box, 1, 0)
|
|
self.details_grid_layout.addWidget(self.year_box, 1, 1)
|
|
|
|
self.details_grid.setLayout(self.details_grid_layout)
|
|
|
|
self.gridLayout.addWidget(self.details_grid, 1, 0)
|
|
|
|
# If it's an aircraft, include the task list.
|
|
if isinstance(unit_type, AircraftType):
|
|
tasks = ", ".join(str(t) for t in unit_type.iter_task_capabilities())
|
|
self.tasks_box = QLabel(f"<b>In-Game Tasks:</b> {tasks}")
|
|
self.tasks_box.setProperty("style", "info-element")
|
|
self.gridLayout.addWidget(self.tasks_box, 2, 0)
|
|
|
|
# Finally, add the description box.
|
|
self.details_text = QTextBrowser()
|
|
self.details_text.setProperty("style", "info-desc")
|
|
self.details_text.setText(unit_type.description)
|
|
self.details_text.setOpenExternalLinks(
|
|
True
|
|
) # in aircrafttype.py and groundunittype.py, for the descriptions, if No Data. including a google search link
|
|
self.gridLayout.addWidget(self.details_text, 3, 0)
|
|
|
|
self.layout.addLayout(self.gridLayout, 1, 0)
|
|
self.setLayout(self.layout)
|