Base menu layout improved.

qt_ui/widgets/base/QBaseInformation.py

@@ -0,0 +1,35 @@
+from PySide2.QtWidgets import QGridLayout, QLabel, QGroupBox, QVBoxLayout
+
+from theater import ControlPoint, Airport
+
+
+class QBaseInformation(QGroupBox):
+
+    def __init__(self, cp:ControlPoint, airport:Airport):
+        super(QBaseInformation, self).__init__("Base defenses")
+        self.cp = cp
+        self.airport = airport
+        self.init_ui()
+
+    def init_ui(self):
+        self.layout = QGridLayout()
+
+        unit_dict = {}
+        for g in self.cp.ground_objects:
+            if g.airbase_group:
+                for group in g.groups:
+                    for u in group.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():
+            self.layout.addWidget(QLabel(str(v) + " x " + k), i, 0)
+            i = i + 1
+
+        stretch = QVBoxLayout()
+        stretch.addStretch()
+        self.layout.addLayout(stretch, len(unit_dict) + 1, 0)
+        self.setLayout(self.layout)