diff --git a/qt_ui/windows/QBriefingWindow.py b/qt_ui/windows/QBriefingWindow.py
index 8ff20338..d8c9d9ca 100644
--- a/qt_ui/windows/QBriefingWindow.py
+++ b/qt_ui/windows/QBriefingWindow.py
@@ -2,9 +2,11 @@ import os
from PySide2.QtGui import QWindow
from PySide2.QtWidgets import QHBoxLayout, QLabel, QWidget, QDialog, QVBoxLayout, QGridLayout, QGroupBox, QCheckBox, \
- QSpinBox, QPushButton
+ QSpinBox, QPushButton, QMessageBox, QComboBox
+from pip._internal.utils import typing
-from game.game import AWACS_BUDGET_COST, PinpointStrike, db, Event
+from game.game import AWACS_BUDGET_COST, PinpointStrike, db, Event, FrontlineAttackEvent, FrontlinePatrolEvent, Task, \
+ UnitType
from userdata.persistency import base_path
import qt_ui.uiconstants as CONST
@@ -19,6 +21,7 @@ class QBriefingWindow(QDialog):
self.setWindowIcon(CONST.EVENT_ICONS[self.gameEvent.__class__])
self.setModal(True)
self.base = self.gameEvent.from_cp.base
+ self.game = self.gameEvent.game
self.scramble_entries = {k: {} for k in self.gameEvent.tasks}
self.initUi()
@@ -29,6 +32,19 @@ class QBriefingWindow(QDialog):
self.scramble_box = QGroupBox("Units")
self.gridLayout = QGridLayout()
self.scramble_box.setLayout(self.gridLayout)
+
+ self.depart_box = QGroupBox("Departure")
+ self.depart_layout = QHBoxLayout()
+ self.depart_box.setLayout(self.depart_layout)
+ self.depart_from_label = QLabel("Depart from : ")
+ self.depart_from = QComboBox()
+
+ for cp in [b for b in self.game.theater.controlpoints if b.captured]:
+ self.depart_from.addItem(str(cp.name))
+
+ self.depart_layout.addWidget(self.depart_from_label)
+ self.depart_layout.addWidget(self.depart_from)
+
row = 0
def header(text, row):
@@ -92,11 +108,13 @@ class QBriefingWindow(QDialog):
self.action_layout = QHBoxLayout()
self.commit_button = QPushButton("Commit")
self.back_button = QPushButton("Commit")
+ self.commit_button.clicked.connect(self.start)
self.action_layout.addWidget(self.commit_button)
self.action_layout.addWidget(self.back_button)
self.support_box = self.initSupportBox()
self.layout.addWidget(QLabel("
{} on {}
".format(self.gameEvent, self.gameEvent.to_cp.name)))
+ self.layout.addWidget(self.depart_box)
self.layout.addWidget(self.scramble_box)
self.layout.addWidget(self.support_box)
self.layout.addWidget(QLabel("Ready?"))
@@ -149,4 +167,91 @@ class QBriefingWindow(QDialog):
pass
def debriefing_directory_location(self) -> str:
- return os.path.join(base_path(), "liberation_debriefings")
\ No newline at end of file
+ return os.path.join(base_path(), "liberation_debriefings")
+
+
+ def start(self):
+
+ if self.awacs_checkbox.isChecked() == 1:
+ self.gameEvent.is_awacs_enabled = True
+ self.game.awacs_expense_commit()
+ else:
+ self.gameEvent.is_awacs_enabled = False
+
+ ca_slot_entry_value = self.ca_slot_entry.value()
+ try:
+ ca_slots = int(ca_slot_entry_value and ca_slot_entry_value or "0")
+ except:
+ ca_slots = 0
+ self.gameEvent.ca_slots = ca_slots
+
+ flights = {k: {} for k in self.gameEvent.tasks} # type: db.TaskForceDict
+ units_scramble_counts = {} # type: typing.Dict[typing.Type[UnitType], int]
+ tasks_scramble_counts = {} # type: typing.Dict[typing.Type[Task], int]
+ tasks_clients_counts = {} # type: typing.Dict[typing.Type[Task], int]
+
+ def dampen_count(unit_type, count: int) -> int:
+ nonlocal units_scramble_counts
+ total_count = self.base.total_units_of_type(unit_type)
+
+ total_scrambled = units_scramble_counts.get(unit_type, 0)
+ dampened_value = count if count + total_scrambled < total_count else total_count - total_scrambled
+ units_scramble_counts[unit_type] = units_scramble_counts.get(unit_type, 0) + dampened_value
+
+ return dampened_value
+
+ for task_type, dict in self.scramble_entries.items():
+ for unit_type, (count_entry, clients_entry) in dict.items():
+ try:
+ count = int(count_entry.value())
+ except:
+ count = 0
+
+ try:
+ clients_count = int(clients_entry and clients_entry.value() or 0)
+ except:
+ clients_count = 0
+
+ dampened_count = dampen_count(unit_type, count)
+ tasks_clients_counts[task_type] = tasks_clients_counts.get(task_type, 0) + clients_count
+ tasks_scramble_counts[task_type] = tasks_scramble_counts.get(task_type, 0) + dampened_count
+
+ flights[task_type][unit_type] = dampened_count, clients_count
+
+ for task in self.gameEvent.ai_banned_tasks:
+ if tasks_clients_counts.get(task, 0) == 0 and tasks_scramble_counts.get(task, 0) > 0:
+ self.showErrorMessage("Need at least one player in flight {}".format(self.gameEvent.flight_name(task)))
+ return
+
+ for task in self.gameEvent.player_banned_tasks:
+ if tasks_clients_counts.get(task, 0) != 0:
+ self.showErrorMessage("Players are not allowed on flight {}".format(self.gameEvent.flight_name(task)))
+ return
+
+ if self.game.is_player_attack(self.gameEvent):
+ if isinstance(self.gameEvent, FrontlineAttackEvent) or isinstance(self.gameEvent, FrontlinePatrolEvent):
+ if self.gameEvent.from_cp.base.total_armor == 0:
+ self.showErrorMessage("No ground vehicles available to attack!")
+ return
+
+ self.gameEvent.player_attacking(flights)
+ else:
+ if isinstance(self.gameEvent, FrontlineAttackEvent) or isinstance(self.gameEvent, FrontlinePatrolEvent):
+ if self.gameEvent.to_cp.base.total_armor == 0:
+ self.showErrorMessage("No ground vehicles available to defend!")
+ return
+
+ self.gameEvent.player_defending(flights)
+
+ self.gameEvent.departure_cp = self.gameEvent.from_cp
+ self.game.initiate_event(self.gameEvent)
+
+ # EventResultsMenu(self.window, self.parent, self.game, self.gameEvent).display()
+
+
+ def showErrorMessage(self, text):
+ about = QMessageBox()
+ about.setWindowTitle("Error")
+ about.setIcon(QMessageBox.Icon.Critical)
+ about.setText(text)
+ about.exec_()
\ No newline at end of file