mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
from PySide2.QtCore import Qt
|
|
from PySide2.QtWidgets import QGridLayout, QLabel, QGroupBox, QPushButton, QVBoxLayout
|
|
|
|
from qt_ui.uiconstants import VEHICLES_ICONS
|
|
from qt_ui.windows.groundobject.QGroundObjectMenu import QGroundObjectMenu
|
|
from theater import ControlPoint, TheaterGroundObject
|
|
|
|
|
|
class QBaseDefenseGroupInfo(QGroupBox):
|
|
|
|
def __init__(self, cp: ControlPoint, ground_object: TheaterGroundObject, game):
|
|
super(QBaseDefenseGroupInfo, self).__init__("Group : " + ground_object.obj_name)
|
|
self.ground_object = ground_object
|
|
self.cp = cp
|
|
self.game = game
|
|
self.buildings = game.theater.find_ground_objects_by_obj_name(self.ground_object.obj_name)
|
|
|
|
self.main_layout = QVBoxLayout()
|
|
self.unit_layout = QGridLayout()
|
|
|
|
self.init_ui()
|
|
|
|
def init_ui(self):
|
|
|
|
self.buildLayout()
|
|
manage_button = QPushButton("Manage")
|
|
manage_button.setProperty("style", "btn-success")
|
|
manage_button.setMaximumWidth(180)
|
|
manage_button.clicked.connect(self.onManage)
|
|
|
|
self.main_layout.addLayout(self.unit_layout)
|
|
self.main_layout.addWidget(manage_button, 0, Qt.AlignLeft)
|
|
|
|
self.setLayout(self.main_layout)
|
|
|
|
def buildLayout(self):
|
|
unit_dict = {}
|
|
for i in range(self.unit_layout.count()):
|
|
item = self.unit_layout.itemAt(i)
|
|
if item is not None and item.widget() is not None:
|
|
self.unit_layout.removeItem(item)
|
|
item.widget().setParent(None)
|
|
item.widget().deleteLater()
|
|
|
|
for g in self.ground_object.groups:
|
|
for u in g.units:
|
|
if u.type in unit_dict.keys():
|
|
unit_dict[u.type] = unit_dict[u.type] + 1
|
|
else:
|
|
unit_dict[u.type] = 1
|
|
i = 0
|
|
for k, v in unit_dict.items():
|
|
icon = QLabel()
|
|
if k in VEHICLES_ICONS.keys():
|
|
icon.setPixmap(VEHICLES_ICONS[k])
|
|
else:
|
|
icon.setText("<b>" + k[:8] + "</b>")
|
|
icon.setProperty("style", "icon-armor")
|
|
self.unit_layout.addWidget(icon, i, 0)
|
|
self.unit_layout.addWidget(QLabel(str(v) + " x " + "<strong>" + k + "</strong>"), i, 1)
|
|
i = i + 1
|
|
|
|
self.setLayout(self.main_layout)
|
|
|
|
def onManage(self):
|
|
self.edition_menu = QGroundObjectMenu(self.window(), self.ground_object, self.buildings, self.cp, self.game)
|
|
self.edition_menu.show()
|
|
self.edition_menu.changed.connect(self.onEdition)
|
|
|
|
def onEdition(self):
|
|
self.buildLayout() |