mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
558dc591a3
A user can now modify the existing default loadout, rather than starting from scratch. Also adds WIP handling for removed pylons, which looks like it may need some PyDCS work. Also fixes the F-14B default loadouts for everything OTHER than fighter sweep again.
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
import logging
|
|
import operator
|
|
from typing import Optional
|
|
|
|
from PySide2.QtWidgets import QComboBox
|
|
|
|
from game import Game, db
|
|
from game.data.weapons import Pylon, Weapon
|
|
from gen.flights.flight import Flight
|
|
from dcs import weapons_data
|
|
|
|
|
|
class QPylonEditor(QComboBox):
|
|
|
|
def __init__(self, game: Game, flight: Flight, pylon: Pylon) -> None:
|
|
super().__init__()
|
|
self.flight = flight
|
|
self.pylon = pylon
|
|
|
|
current = self.flight.loadout.get(self.pylon.number)
|
|
|
|
self.addItem("None", None)
|
|
if game.settings.restrict_weapons_by_date:
|
|
weapons = pylon.available_on(game.date)
|
|
else:
|
|
weapons = pylon.allowed
|
|
allowed = sorted(weapons, key=operator.attrgetter("name"))
|
|
for i, weapon in enumerate(allowed):
|
|
self.addItem(weapon.name, weapon)
|
|
if current == weapon:
|
|
self.setCurrentIndex(i + 1)
|
|
|
|
self.default_loadout(self.pylon)
|
|
self.currentIndexChanged.connect(self.on_pylon_change)
|
|
|
|
def on_pylon_change(self):
|
|
selected: Optional[Weapon] = self.currentData()
|
|
self.flight.loadout[self.pylon.number] = selected
|
|
|
|
if selected is None:
|
|
logging.debug(f"Pylon {self.pylon.number} emptied")
|
|
else:
|
|
logging.debug(
|
|
f"Pylon {self.pylon.number} changed to {selected.name}")
|
|
|
|
def default_loadout(self, pylon: Pylon) -> None:
|
|
self.flight.unit_type.load_payloads()
|
|
self.setCurrentText("None")
|
|
pylon_default_weapon = None
|
|
loadout = None
|
|
# Iterate through each possible payload type for a given aircraft.
|
|
# Some aircraft have custom loadouts that in aren't the standard set.
|
|
for payload_override in db.EXPANDED_TASK_PAYLOAD_OVERRIDE.get(self.flight.flight_type.name):
|
|
if loadout is None:
|
|
loadout = self.flight.unit_type.loadout_by_name(payload_override)
|
|
if loadout is not None:
|
|
for i in loadout:
|
|
if i[0] == self.pylon.number:
|
|
pylon_default_weapon = i[1]["clsid"]
|
|
# TODO: Handle removed pylons better.
|
|
if pylon_default_weapon == "<CLEAN>":
|
|
pylon_default_weapon = None
|
|
if pylon_default_weapon is not None:
|
|
#self.setCurrentIndex(self.findText(weapons_data.weapon_ids.get(pylon_default_weapon).get("name")))
|
|
self.setCurrentText(weapons_data.weapon_ids.get(pylon_default_weapon).get("name")) |