mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
64d60e5ccf
Support for 2.9.10.4160, though without Iraq, but finally there's some basic support for datalink...
110 lines
4.2 KiB
Python
110 lines
4.2 KiB
Python
import itertools
|
|
import logging
|
|
from typing import Callable, Optional
|
|
|
|
from PySide6.QtCore import QRect
|
|
from PySide6.QtWidgets import QGridLayout, QLabel, QWidget
|
|
from dcs.unitpropertydescription import UnitPropertyDescription
|
|
|
|
from game.ato import Flight
|
|
from game.ato.flightmember import FlightMember
|
|
from .missingpropertydataerror import MissingPropertyDataError
|
|
from .propertycheckbox import PropertyCheckBox
|
|
from .propertycombobox import PropertyComboBox
|
|
from .propertyeditbox import PropertyEditBox
|
|
from .propertyspinbox import PropertySpinBox
|
|
|
|
|
|
class UnhandledControlTypeError(RuntimeError):
|
|
def __init__(self, control: str) -> None:
|
|
super().__init__(f"Unhandled control type {control}")
|
|
|
|
|
|
class PropertyEditor(QGridLayout):
|
|
def __init__(self, flight: Flight, flight_member: FlightMember) -> None:
|
|
super().__init__()
|
|
self.flight = flight
|
|
self.flight_member = flight_member
|
|
self.flight_member_update_listeners: list[Callable[[FlightMember], None]] = []
|
|
|
|
self.build_props(flight)
|
|
|
|
def build_props(self, flight):
|
|
self.setGeometry(QRect())
|
|
last_label: QWidget | None = None
|
|
row = itertools.count()
|
|
for prop in flight.unit_type.iter_props():
|
|
if prop.label is None:
|
|
if prop.control != "label":
|
|
logging.error(
|
|
"Found non-label aircraft property with no display name."
|
|
)
|
|
continue
|
|
|
|
if prop.player_only and not flight.client_count:
|
|
continue
|
|
|
|
if not flight.unit_type.should_show_prop(prop.identifier):
|
|
continue
|
|
|
|
try:
|
|
widget = self.control_for_property(prop)
|
|
except (MissingPropertyDataError, UnhandledControlTypeError):
|
|
logging.exception(
|
|
f"Cannot create property control for property %s of %s",
|
|
prop.identifier,
|
|
flight.unit_type,
|
|
)
|
|
continue
|
|
|
|
# Draw any deferred label if this is a real control.
|
|
if last_label is not None and widget is not None:
|
|
self.addWidget(last_label, next(row), 0)
|
|
last_label = None
|
|
|
|
label = prop.label
|
|
if widget is None:
|
|
label = f"<strong>{label}</label>"
|
|
|
|
label = QLabel(label)
|
|
if widget is None:
|
|
# The "control" is only a section label. Defer adding it to the layout
|
|
# so that we can skip empty sections.
|
|
last_label = label
|
|
else:
|
|
# Else the label is for the control itself and should be drawn
|
|
# immediately.
|
|
this_row = next(row)
|
|
self.addWidget(label, this_row, 0)
|
|
self.addWidget(widget, this_row, 1)
|
|
|
|
def control_for_property(self, prop: UnitPropertyDescription) -> Optional[QWidget]:
|
|
# Valid values are:
|
|
# "checkbox", "comboList", "groupbox", "label", "slider", "spinbox", "editbox"
|
|
match prop.control:
|
|
case "checkbox":
|
|
widget = PropertyCheckBox(self.flight_member, prop)
|
|
self.flight_member_update_listeners.append(widget.set_flight_member)
|
|
return widget
|
|
case "comboList":
|
|
widget = PropertyComboBox(self.flight_member, prop)
|
|
self.flight_member_update_listeners.append(widget.set_flight_member)
|
|
return widget
|
|
case "groupbox" | "label":
|
|
return None
|
|
case "slider" | "spinbox":
|
|
widget = PropertySpinBox(self.flight_member, prop)
|
|
self.flight_member_update_listeners.append(widget.set_flight_member)
|
|
return widget
|
|
case "editbox":
|
|
widget = PropertyEditBox(self.flight_member, prop, self.flight)
|
|
self.flight_member_update_listeners.append(widget.set_flight_member)
|
|
return widget
|
|
case _:
|
|
raise UnhandledControlTypeError(prop.control)
|
|
|
|
def set_flight_member(self, flight_member: FlightMember) -> None:
|
|
self.flight_member = flight_member
|
|
for listener in self.flight_member_update_listeners:
|
|
listener(self.flight_member)
|