mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
104 lines
3.9 KiB
Python
104 lines
3.9 KiB
Python
from typing import List
|
|
|
|
from PySide2.QtCore import Qt
|
|
from PySide2.QtWidgets import QDialog, QGridLayout, QLabel, QComboBox, QHBoxLayout, QVBoxLayout, QPushButton, QSpinBox
|
|
from dcs import Point
|
|
from dcs.unittype import UnitType
|
|
|
|
from game import Game
|
|
from gen.flights.ai_flight_planner import FlightPlanner
|
|
from gen.flights.flight import Flight, FlightWaypoint, FlightType
|
|
from qt_ui.uiconstants import EVENT_ICONS
|
|
from qt_ui.windows.mission.flight.waypoints.QFlightWaypointInfoBox import QFlightWaypointInfoBox
|
|
from theater import ControlPoint
|
|
|
|
PREDEFINED_WAYPOINT_CATEGORIES = [
|
|
"Frontline (CAS AREA)",
|
|
"Building",
|
|
"Units",
|
|
"Airbase"
|
|
]
|
|
|
|
|
|
class QFlightCreator(QDialog):
|
|
|
|
def __init__(self, game: Game, from_cp:ControlPoint, possible_aircraft_type:List[UnitType], flight_view=None):
|
|
super(QFlightCreator, self).__init__()
|
|
self.game = game
|
|
self.from_cp = from_cp
|
|
self.flight_view = flight_view
|
|
self.planner = self.game.planners[from_cp.id]
|
|
|
|
self.setWindowFlags(Qt.WindowStaysOnTopHint)
|
|
self.setModal(True)
|
|
self.setWindowTitle("Create flight")
|
|
self.setWindowIcon(EVENT_ICONS["strike"])
|
|
|
|
self.select_type_aircraft = QComboBox()
|
|
for aircraft_type in possible_aircraft_type:
|
|
print(aircraft_type)
|
|
print(aircraft_type.name)
|
|
self.select_type_aircraft.addItem(aircraft_type.id, userData=aircraft_type)
|
|
self.select_type_aircraft.setCurrentIndex(0)
|
|
|
|
self.select_flight_type = QComboBox()
|
|
self.select_flight_type.addItem("CAP", userData=FlightType.CAP)
|
|
self.select_flight_type.addItem("BARCAP", userData=FlightType.BARCAP)
|
|
self.select_flight_type.addItem("TARCAP", userData=FlightType.TARCAP)
|
|
self.select_flight_type.addItem("INTERCEPT", userData=FlightType.INTERCEPTION)
|
|
self.select_flight_type.addItem("CAS", userData=FlightType.CAS)
|
|
self.select_flight_type.addItem("SEAD", userData=FlightType.SEAD)
|
|
self.select_flight_type.addItem("DEAD", userData=FlightType.DEAD)
|
|
self.select_flight_type.addItem("STRIKE", userData=FlightType.STRIKE)
|
|
self.select_flight_type.setCurrentIndex(0)
|
|
|
|
self.select_count_of_aircraft = QSpinBox()
|
|
self.select_count_of_aircraft.setMinimum(1)
|
|
self.select_count_of_aircraft.setMaximum(4)
|
|
self.select_count_of_aircraft.setValue(2)
|
|
|
|
self.add_button = QPushButton("Add")
|
|
self.add_button.clicked.connect(self.create)
|
|
|
|
self.init_ui()
|
|
|
|
|
|
def init_ui(self):
|
|
layout = QVBoxLayout()
|
|
|
|
type_layout = QHBoxLayout()
|
|
type_layout.addWidget(QLabel("Type of Aircraft : "))
|
|
type_layout.addStretch()
|
|
type_layout.addWidget(self.select_type_aircraft, alignment=Qt.AlignRight)
|
|
|
|
|
|
count_layout = QHBoxLayout()
|
|
count_layout.addWidget(QLabel("Count : "))
|
|
count_layout.addStretch()
|
|
count_layout.addWidget(self.select_count_of_aircraft, alignment=Qt.AlignRight)
|
|
|
|
|
|
flight_type_layout = QHBoxLayout()
|
|
flight_type_layout.addWidget(QLabel("Task : "))
|
|
flight_type_layout.addStretch()
|
|
flight_type_layout.addWidget(self.select_flight_type, alignment=Qt.AlignRight)
|
|
|
|
layout.addLayout(type_layout)
|
|
layout.addLayout(count_layout)
|
|
layout.addLayout(flight_type_layout)
|
|
layout.addStretch()
|
|
layout.addWidget(self.add_button, alignment=Qt.AlignRight)
|
|
|
|
self.setLayout(layout)
|
|
|
|
def create(self):
|
|
aircraft_type = self.select_type_aircraft.currentData()
|
|
count = self.select_count_of_aircraft.value()
|
|
flight = Flight(aircraft_type, count, self.from_cp, self.select_flight_type.currentData())
|
|
self.planner.flights.append(flight)
|
|
self.planner.custom_flights.append(flight)
|
|
if self.flight_view is not None:
|
|
self.flight_view.set_flight_planner(self.planner)
|
|
self.close()
|
|
|