Allow changing time, date & weather

Resolves #103

qt_ui/widgets/QConditionsDialog.py

@@ -0,0 +1,85 @@
+from copy import deepcopy
+from datetime import datetime, timedelta
+
+from PySide6.QtWidgets import QDialog, QVBoxLayout, QHBoxLayout, QPushButton
+
+from game.sim import GameUpdateEvents
+from game.weather.clouds import Clouds
+from qt_ui.widgets.conditions.QTimeAdjustmentWidget import QTimeAdjustmentWidget
+from qt_ui.widgets.conditions.QTimeTurnWidget import QTimeTurnWidget
+from qt_ui.widgets.conditions.QWeatherAdjustmentWidget import QWeatherAdjustmentWidget
+from qt_ui.widgets.conditions.QWeatherWidget import QWeatherWidget
+
+
+class QConditionsDialog(QDialog):
+    def __init__(self, time_turn: QTimeTurnWidget, weather: QWeatherWidget):
+        super().__init__()
+        self.time_turn = time_turn
+        self.weather = weather
+        self.init_ui()
+
+    def init_ui(self):
+        self.setWindowTitle("Time & Weather Conditions")
+        self.setMinimumSize(360, 380)
+
+        vbox = QVBoxLayout()
+
+        self.time_adjuster = QTimeAdjustmentWidget(self.time_turn)
+        vbox.addWidget(self.time_adjuster, 1)
+        self.weather_adjuster = QWeatherAdjustmentWidget(self.weather)
+        vbox.addWidget(self.weather_adjuster, 8)
+
+        hbox = QHBoxLayout()
+        reject_btn = QPushButton("REJECT")
+        reject_btn.setProperty("style", "btn-danger")
+        reject_btn.clicked.connect(self.close)
+        hbox.addWidget(reject_btn)
+        accept_btn = QPushButton("ACCEPT")
+        accept_btn.setProperty("style", "btn-success")
+        accept_btn.clicked.connect(self.apply_conditions)
+        hbox.addWidget(accept_btn)
+        vbox.addLayout(hbox, 1)
+
+        self.setLayout(vbox)
+
+    def apply_conditions(self) -> None:
+        qdt: datetime = self.time_adjuster.datetime_edit.dateTime().toPython()
+
+        sim = self.time_turn.sim_controller
+        current_time = sim.current_time_in_sim_if_game_loaded
+        if current_time:
+            current_time = deepcopy(current_time)
+        sim.game_loop.sim.time = qdt
+
+        game = sim.game_loop.game
+        game.date = qdt.date() - timedelta(days=game.turn // 4)
+        game.conditions.start_time = qdt
+        self.time_turn.set_current_turn(game.turn, game.conditions)
+
+        # TODO: create new weather object
+
+        new_weather_type = self.weather_adjuster.type_selector.currentData()
+        new_weather = new_weather_type(
+            seasonal_conditions=game.theater.seasonal_conditions,
+            day=qdt.date(),
+            time_of_day=game.current_turn_time_of_day,
+        )
+
+        # self.weather.conditions.weather = WeatherType()
+        preset = self.weather_adjuster.preset_selector.currentData()
+        new_weather.clouds = Clouds(
+            base=self.weather_adjuster.cloud_base.base.value(),
+            density=self.weather_adjuster.cloud_density.density.value(),
+            thickness=self.weather_adjuster.cloud_thickness.thickness.value(),
+            precipitation=self.weather_adjuster.precipitation.selector.currentData(),
+            preset=preset,
+        )
+
+        self.weather.conditions.weather = new_weather
+
+        self.weather.update_forecast()
+        if game.turn > 0 and current_time != qdt:
+            events = GameUpdateEvents()
+            game.initialize_turn(events, for_blue=True, for_red=True)
+            sim.sim_update.emit(events)
+        self.accept()