mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
8bc77bbf18
Make the type of the waypoint a non-optional part of the constructor. Every waypoint needs a type, and there's no good default (the previous default, `TAKEOFF`, is actually unused). All of the target waypoints were mistakenly being set as `TAKEOFF`, so I've fixed that in the process. Also, fix the bug where only the last custom target of a SEAD objective was being added to the waypoint list because the append was scoped incorrectly.
155 lines
7.6 KiB
Python
155 lines
7.6 KiB
Python
from PySide2.QtCore import QSortFilterProxyModel, Qt, QModelIndex
|
|
from PySide2.QtGui import QStandardItem, QStandardItemModel
|
|
from PySide2.QtWidgets import QComboBox, QCompleter
|
|
from game import Game
|
|
from gen import Conflict, FlightWaypointType
|
|
from gen.flights.flight import FlightWaypoint, PredefinedWaypointCategory
|
|
from qt_ui.widgets.combos.QFilteredComboBox import QFilteredComboBox
|
|
from theater import ControlPointType
|
|
|
|
|
|
class QPredefinedWaypointSelectionComboBox(QFilteredComboBox):
|
|
|
|
def __init__(self, game: Game, parent=None, include_targets=True, include_airbases=True,
|
|
include_frontlines=True, include_units=True, include_enemy=True, include_friendly=True):
|
|
super(QPredefinedWaypointSelectionComboBox, self).__init__(parent)
|
|
self.game = game
|
|
self.include_targets = include_targets
|
|
self.include_airbases = include_airbases
|
|
self.include_frontlines = include_frontlines
|
|
self.include_units = include_units
|
|
self.include_enemy = include_enemy
|
|
self.include_friendly = include_friendly
|
|
self.find_possible_waypoints()
|
|
|
|
def get_selected_waypoints(self, include_all_from_same_location=False):
|
|
n = self.currentText()
|
|
first_waypoint = None
|
|
for w in self.wpts:
|
|
if w.pretty_name == n:
|
|
first_waypoint = w
|
|
break
|
|
if first_waypoint is None:
|
|
return []
|
|
waypoints = [first_waypoint]
|
|
if include_all_from_same_location:
|
|
for w in self.wpts:
|
|
if w is not first_waypoint and w.obj_name and w.obj_name == first_waypoint.obj_name:
|
|
waypoints.append(w)
|
|
return waypoints
|
|
|
|
def find_possible_waypoints(self):
|
|
|
|
self.wpts = []
|
|
model = QStandardItemModel()
|
|
i = 0
|
|
|
|
def add_model_item(i, model, name, wpt):
|
|
print(name)
|
|
item = QStandardItem(name)
|
|
model.setItem(i, 0, item)
|
|
self.wpts.append(wpt)
|
|
return i + 1
|
|
|
|
if self.include_frontlines:
|
|
for cp in self.game.theater.controlpoints:
|
|
if cp.captured:
|
|
enemy_cp = [ecp for ecp in cp.connected_points if ecp.captured != cp.captured]
|
|
for ecp in enemy_cp:
|
|
pos = Conflict.frontline_position(self.game.theater, cp, ecp)[0]
|
|
wpt = FlightWaypoint(
|
|
FlightWaypointType.CUSTOM,
|
|
pos.x,
|
|
pos.y,
|
|
800)
|
|
wpt.name = "Frontline " + cp.name + "/" + ecp.name + " [CAS]"
|
|
wpt.alt_type = "RADIO"
|
|
wpt.pretty_name = wpt.name
|
|
wpt.description = "Frontline"
|
|
wpt.data = [cp, ecp]
|
|
wpt.category = PredefinedWaypointCategory.FRONTLINE
|
|
i = add_model_item(i, model, wpt.pretty_name, wpt)
|
|
|
|
if self.include_targets:
|
|
for cp in self.game.theater.controlpoints:
|
|
if (self.include_enemy and not cp.captured) or (self.include_friendly and cp.captured):
|
|
for ground_object in cp.ground_objects:
|
|
if not ground_object.is_dead and not ground_object.dcs_identifier == "AA":
|
|
wpt = FlightWaypoint(
|
|
FlightWaypointType.CUSTOM,
|
|
ground_object.position.x,
|
|
ground_object.position.y,
|
|
0
|
|
)
|
|
wpt.alt_type = "RADIO"
|
|
wpt.name = wpt.name = "[" + str(ground_object.obj_name) + "] : " + ground_object.category + " #" + str(ground_object.object_id)
|
|
wpt.pretty_name = wpt.name
|
|
wpt.obj_name = ground_object.obj_name
|
|
wpt.targets.append(ground_object)
|
|
wpt.data = ground_object
|
|
if cp.captured:
|
|
wpt.description = "Friendly Building"
|
|
wpt.category = PredefinedWaypointCategory.ALLY_BUILDING
|
|
else:
|
|
wpt.description = "Enemy Building"
|
|
wpt.category = PredefinedWaypointCategory.ENEMY_BUILDING
|
|
i = add_model_item(i, model, wpt.pretty_name, wpt)
|
|
|
|
if self.include_units:
|
|
for cp in self.game.theater.controlpoints:
|
|
if (self.include_enemy and not cp.captured) or (self.include_friendly and cp.captured):
|
|
for ground_object in cp.ground_objects:
|
|
if not ground_object.is_dead and ground_object.dcs_identifier == "AA":
|
|
for g in ground_object.groups:
|
|
for j, u in enumerate(g.units):
|
|
wpt = FlightWaypoint(
|
|
FlightWaypointType.CUSTOM,
|
|
u.position.x,
|
|
u.position.y,
|
|
0
|
|
)
|
|
wpt.alt_type = "RADIO"
|
|
wpt.name = wpt.name = "[" + str(ground_object.obj_name) + "] : " + u.type + " #" + str(j)
|
|
wpt.pretty_name = wpt.name
|
|
wpt.targets.append(u)
|
|
wpt.data = u
|
|
wpt.obj_name = ground_object.obj_name
|
|
wpt.waypoint_type = FlightWaypointType.CUSTOM
|
|
if cp.captured:
|
|
wpt.description = "Friendly unit : " + u.type
|
|
wpt.category = PredefinedWaypointCategory.ALLY_UNIT
|
|
else:
|
|
wpt.description = "Enemy unit : " + u.type
|
|
wpt.category = PredefinedWaypointCategory.ENEMY_UNIT
|
|
i = add_model_item(i, model, wpt.pretty_name, wpt)
|
|
|
|
if self.include_airbases:
|
|
for cp in self.game.theater.controlpoints:
|
|
if (self.include_enemy and not cp.captured) or (self.include_friendly and cp.captured):
|
|
wpt = FlightWaypoint(
|
|
FlightWaypointType.CUSTOM,
|
|
cp.position.x,
|
|
cp.position.y,
|
|
0
|
|
)
|
|
wpt.alt_type = "RADIO"
|
|
wpt.name = cp.name
|
|
wpt.data = cp
|
|
if cp.captured:
|
|
wpt.description = "Position of " + cp.name + " [Friendly Airbase]"
|
|
wpt.category = PredefinedWaypointCategory.ALLY_CP
|
|
else:
|
|
wpt.description = "Position of " + cp.name + " [Enemy Airbase]"
|
|
wpt.category = PredefinedWaypointCategory.ENEMY_CP
|
|
|
|
if cp.cptype == ControlPointType.AIRCRAFT_CARRIER_GROUP:
|
|
wpt.pretty_name = cp.name + " (Aircraft Carrier Group)"
|
|
elif cp.cptype == ControlPointType.LHA_GROUP:
|
|
wpt.pretty_name = cp.name + " (LHA Group)"
|
|
else:
|
|
wpt.pretty_name = cp.name + " (Airbase)"
|
|
|
|
i = add_model_item(i, model, wpt.pretty_name, wpt)
|
|
|
|
self.setModel(model)
|