diff --git a/qt_ui/windows/basemenu/intel/QIntelInfo.py b/qt_ui/windows/basemenu/intel/QIntelInfo.py
index ea5dfa4c..cc9c3ca6 100644
--- a/qt_ui/windows/basemenu/intel/QIntelInfo.py
+++ b/qt_ui/windows/basemenu/intel/QIntelInfo.py
@@ -1,3 +1,6 @@
+from collections import defaultdict
+
+from PySide2.QtCore import Qt
from PySide2.QtWidgets import (
QFrame,
QGridLayout,
@@ -7,8 +10,6 @@ from PySide2.QtWidgets import (
QScrollArea,
QWidget,
)
-from PySide2.QtCore import Qt
-from dcs.task import CAP, CAS, Embarking, PinpointStrike
from game import Game, db
from game.theater import ControlPoint
@@ -19,51 +20,44 @@ class QIntelInfo(QFrame):
super(QIntelInfo, self).__init__()
self.cp = cp
self.game = game
- self.init_ui()
- def init_ui(self):
layout = QVBoxLayout()
scroll_content = QWidget()
- intelLayout = QVBoxLayout()
+ intel_layout = QVBoxLayout()
- units = {
- CAP: db.find_unittype(CAP, self.game.enemy_name),
- Embarking: db.find_unittype(Embarking, self.game.enemy_name),
- CAS: db.find_unittype(CAS, self.game.enemy_name),
- PinpointStrike: db.find_unittype(PinpointStrike, self.game.enemy_name),
+ 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())
}
- for task_type in units.keys():
- units_column = list(set(units[task_type]))
+ 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
- if sum([self.cp.base.total_units_of_type(u) for u in units_column]) > 0:
+ 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)
- group = QGroupBox(db.task_name(task_type))
- groupLayout = QGridLayout()
- group.setLayout(groupLayout)
+ for row, (name, count) in enumerate(unit_types.items()):
+ task_layout.addWidget(QLabel(f"{name}"), row, 0)
+ task_layout.addWidget(QLabel(str(count)), row, 1)
- row = 0
- for unit_type in units_column:
- existing_units = self.cp.base.total_units_of_type(unit_type)
- if existing_units == 0:
- continue
- groupLayout.addWidget(
- QLabel(
- ""
- + db.unit_get_expanded_info(
- self.game.enemy_country, unit_type, "name"
- )
- + ""
- ),
- row,
- 0,
- )
- groupLayout.addWidget(QLabel(str(existing_units)), row, 1)
- row += 1
+ intel_layout.addWidget(task_group)
- intelLayout.addWidget(group)
-
- scroll_content.setLayout(intelLayout)
+ scroll_content.setLayout(intel_layout)
scroll = QScrollArea()
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)