mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
from collections import defaultdict
|
|
|
|
from PySide2.QtCore import Qt
|
|
from PySide2.QtWidgets import (
|
|
QFrame,
|
|
QGridLayout,
|
|
QGroupBox,
|
|
QLabel,
|
|
QVBoxLayout,
|
|
QScrollArea,
|
|
QWidget,
|
|
)
|
|
|
|
from game import Game, db
|
|
from game.theater import ControlPoint
|
|
|
|
|
|
class QIntelInfo(QFrame):
|
|
def __init__(self, cp: ControlPoint, game: Game):
|
|
super(QIntelInfo, self).__init__()
|
|
self.cp = cp
|
|
self.game = game
|
|
|
|
layout = QVBoxLayout()
|
|
scroll_content = QWidget()
|
|
intel_layout = QVBoxLayout()
|
|
|
|
units_by_task: dict[str, dict[str, int]] = defaultdict(lambda: defaultdict(int))
|
|
for unit_type, count in self.cp.base.aircraft.items():
|
|
if count:
|
|
name = db.unit_get_expanded_info(
|
|
self.game.enemy_country, unit_type, "name"
|
|
)
|
|
units_by_task[unit_type.task_default.name][name] += count
|
|
|
|
units_by_task = {
|
|
task: units_by_task[task] for task in sorted(units_by_task.keys())
|
|
}
|
|
|
|
front_line_units = defaultdict(int)
|
|
for unit_type, count in self.cp.base.armor.items():
|
|
if count:
|
|
name = db.unit_get_expanded_info(
|
|
self.game.enemy_country, unit_type, "name"
|
|
)
|
|
front_line_units[name] += count
|
|
|
|
units_by_task["Front line units"] = front_line_units
|
|
for task, unit_types in units_by_task.items():
|
|
task_group = QGroupBox(task)
|
|
task_layout = QGridLayout()
|
|
task_group.setLayout(task_layout)
|
|
|
|
for row, (name, count) in enumerate(unit_types.items()):
|
|
task_layout.addWidget(QLabel(f"<b>{name}</b>"), row, 0)
|
|
task_layout.addWidget(QLabel(str(count)), row, 1)
|
|
|
|
intel_layout.addWidget(task_group)
|
|
|
|
scroll_content.setLayout(intel_layout)
|
|
scroll = QScrollArea()
|
|
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
|
|
scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
|
|
scroll.setWidgetResizable(True)
|
|
scroll.setWidget(scroll_content)
|
|
|
|
layout.addWidget(scroll)
|
|
|
|
self.setLayout(layout)
|