Further improvements to livery-selector

Unlocking the full potential, i.e. allowing users to switch back & forth between livery-sets, livery-overrides and normal liveries, including support for overriding flight/flight-member specific liveries.

qt_ui/widgets/combos/QSquadronLiverySelector.py

@@ -4,6 +4,8 @@ from PySide6.QtWidgets import QComboBox
 
 from game.squadrons import Squadron
 
+LIVERY_SET_TEXT = "Use livery-set from squadron's yaml"
+
 
 class SquadronLiverySelector(QComboBox):
     """
@@ -11,15 +13,17 @@ class SquadronLiverySelector(QComboBox):
     The combo box will automatically be populated with all available liveries.
     """
 
-    def __init__(
-        self, squadron: Squadron, full_list_view_override: bool = False
-    ) -> None:
+    def __init__(self, squadron: Squadron, update_squadron: bool = True) -> None:
         super().__init__()
         self.setSizeAdjustPolicy(QComboBox.SizeAdjustPolicy.AdjustToContents)
 
+        self.squadron = squadron
         self.aircraft_type = squadron.aircraft
         selected_livery = squadron.livery
 
+        if update_squadron:
+            self.currentTextChanged.connect(self.on_change)
+
         liveries = set()
         cc = squadron.coalition.faction.country.shortname
         aircraft_liveries = set(self.aircraft_type.dcs_unit_type.iter_liveries())
@@ -35,23 +39,32 @@ class SquadronLiverySelector(QComboBox):
             for x in faction.liveries_overrides.get(self.aircraft_type, [])
             if x in [y.id.lower() for y in liveries]
         ]
-        if (
-            selected_livery is None
-            and squadron.livery_set
-            and not full_list_view_override
-        ):
-            self.addItem("Using livery-set from squadron's yaml", userData=None)
-            self.setEnabled(False)
-            return
-        if selected_livery is None and squadron.aircraft.default_livery:
-            selected_livery = squadron.aircraft.default_livery
+        if squadron.livery_set:
+            self.addItem(LIVERY_SET_TEXT, userData=None)
         if len(overrides) > 0:
             self.addItem("Use livery overrides", userData=None)
+        if (
+            selected_livery is None
+            and not squadron.livery_set
+            and squadron.aircraft.default_livery
+        ):
+            selected_livery = squadron.aircraft.default_livery
         for livery in sorted(liveries):
             self.addItem(livery.name, userData=livery.id)
-            if selected_livery is not None:
+            if selected_livery is not None and not squadron.livery_set:
                 if selected_livery.lower() == livery.id:
                     self.setCurrentText(livery.name)
         if len(liveries) == 0:
             self.addItem("No available liveries (using DCS default)")
             self.setEnabled(False)
+
+    @property
+    def using_livery_set(self) -> bool:
+        return self.currentText() == LIVERY_SET_TEXT
+
+    def on_change(self, text: str) -> None:
+        self.squadron.livery = self.currentData()
+        if text == LIVERY_SET_TEXT:
+            self.squadron.use_livery_set = True
+        else:
+            self.squadron.use_livery_set = False

qt_ui/windows/AirWingConfigurationDialog.py

@@ -439,7 +439,6 @@ class SquadronConfigurationBox(QGroupBox):
         if base is None:
             raise RuntimeError("Base cannot be none")
         self.squadron.assign_to_base(base)
-        self.squadron.livery = self.livery_selector.currentData()
         self.return_players_to_squadron()
 
         # Also update the auto assignable mission types

qt_ui/windows/SquadronDialog.py

@@ -268,7 +268,6 @@ class SquadronDialog(QDialog):
 
         left_column.addWidget(QLabel("Livery"))
         self.livery_selector = SquadronLiverySelector(self.squadron_model.squadron)
-        self.livery_selector.currentIndexChanged.connect(self.on_livery_changed)
         left_column.addWidget(self.livery_selector)
 
         auto_assigned_tasks = AutoAssignedTaskControls(squadron_model)
@@ -422,6 +421,3 @@ class SquadronDialog(QDialog):
         if task is None:
             raise RuntimeError("Selected task cannot be None")
         self.squadron.primary_task = task
-
-    def on_livery_changed(self) -> None:
-        self.squadron.livery = self.livery_selector.currentData()

qt_ui/windows/mission/flight/payload/QFlightPayloadTab.py

@@ -152,11 +152,7 @@ class QFlightPayloadTab(QFrame):
         self.same_livery_for_all_checkbox.toggled.connect(self.on_same_livery_toggled)
         hbox.addWidget(self.same_livery_for_all_checkbox)
         self.livery_selector = SquadronLiverySelector(
-            self.flight.squadron, full_list_view_override=True
-        )
-        self.livery_selector.insertItem(0, "Default", None)
-        self.livery_selector.setCurrentIndex(
-            self.livery_selector.findData(self.member_selector.selected_member.livery)
+            self.flight.squadron, update_squadron=False
         )
         self.livery_selector.currentIndexChanged.connect(self.on_livery_change)
         hbox.addWidget(self.livery_selector)
@@ -313,8 +309,11 @@ class QFlightPayloadTab(QFrame):
 
     def on_livery_change(self) -> None:
         livery = self.livery_selector.currentData()
+        use_livery_set = self.livery_selector.using_livery_set
         if self.flight.use_same_livery_for_all_members:
             for m in self.flight.roster.members:
                 m.livery = livery
+                m.use_livery_set = use_livery_set
         else:
             self.member_selector.selected_member.livery = livery
+            self.member_selector.selected_member.use_livery_set = use_livery_set