mirror of
https://github.com/spencershepard/RotorOps.git
synced 2025-11-10 15:45:30 +00:00
bdf5d95f48
-UH-60L troop capacity set to 11 -bugfix: AI enemy planes/helicopters attacked invisible FARPS -generator now produces an error log -Syria scenario FARP support units invulnerable -all red/blue coalition objects now swap sides for defense mode, including carriers and FARPs -forces templates can include air units with customization for loadout, livery, and skill -carrier and FARP parking for enemy helicopters -parking now supports multiple airports per side -improved AI flight orbits (now onside and perpendicular to closest enemy airport) -enemy transport helicopters! Random troop drops in enemy-occupied zones -'APCs spawn infantry' now disables conflict zones as infinite troop pickup zones -bug fix: add zone triggers can be added in wrong order -Combined Joint Task Forces now supported -disable pickup zone smoke option added -Scenario briefing now viewable in generator -defending vehicles disperse on attack (script option) -AWACs escorts now weapons-free and engage enemy air threats -briefing images added -new Persian Gulf scenario
255 lines
10 KiB
Python
255 lines
10 KiB
Python
import math
|
|
import sys
|
|
import os
|
|
import dcs
|
|
import RotorOpsMission as ROps
|
|
import RotorOpsUtils
|
|
import RotorOpsUnits
|
|
import logging
|
|
|
|
from PyQt5.QtWidgets import (
|
|
QApplication, QDialog, QMainWindow, QMessageBox
|
|
)
|
|
from PyQt5 import QtGui
|
|
from MissionGeneratorUI import Ui_MainWindow
|
|
|
|
|
|
#Setup logfile and exception handler
|
|
logger = logging.getLogger(__name__)
|
|
logging.basicConfig(filename='generator.log', encoding='utf-8', level=logging.DEBUG, format='%(asctime)s %(levelname)-8s %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
|
|
handler = logging.StreamHandler(stream=sys.stdout)
|
|
logger.addHandler(handler)
|
|
|
|
def handle_exception(exc_type, exc_value, exc_traceback):
|
|
if issubclass(exc_type, KeyboardInterrupt): #example of handling error subclasses
|
|
sys.__excepthook__(exc_type, exc_value, exc_traceback)
|
|
return
|
|
|
|
logger.error("Uncaught exception", exc_info=(exc_type, exc_value, exc_traceback))
|
|
msg = QMessageBox()
|
|
msg.setWindowTitle("Uncaught exception")
|
|
msg.setText("Oops, there was a problem. Please check the log file or post it in the RotorOps discord where some helpful people will have a look.")
|
|
x = msg.exec_()
|
|
|
|
|
|
sys.excepthook = handle_exception
|
|
|
|
|
|
maj_version = 0
|
|
minor_version = 4
|
|
version_string = str(maj_version) + "." + str(minor_version)
|
|
scenarios = []
|
|
red_forces_files = []
|
|
blue_forces_files = []
|
|
defenders_text = "Defending Forces:"
|
|
attackers_text = "Attacking Forces:"
|
|
|
|
logger.info("RotorOps v" + version_string)
|
|
|
|
class Window(QMainWindow, Ui_MainWindow):
|
|
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
|
|
if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
|
|
logger.info('running in a PyInstaller bundle')
|
|
home_dir = os.getcwd()
|
|
os.chdir(home_dir + "/Generator")
|
|
else:
|
|
logger.info('running in a normal Python process')
|
|
|
|
|
|
self.m = ROps.RotorOpsMission()
|
|
self.setupUi(self)
|
|
self.connectSignalsSlots()
|
|
self.populateScenarios()
|
|
self.populateForces("red", self.redforces_comboBox, red_forces_files)
|
|
self.populateForces("blue", self.blueforces_comboBox, blue_forces_files)
|
|
self.populateSlotSelection()
|
|
|
|
self.blue_forces_label.setText(attackers_text)
|
|
self.red_forces_label.setText(defenders_text)
|
|
self.background_label.setPixmap(QtGui.QPixmap(self.m.assets_dir + "/background.PNG"))
|
|
self.statusbar.setStyleSheet(
|
|
"QStatusBar{padding-left:5px;color:black;font-weight:bold;}")
|
|
|
|
self.version_label.setText("Version " + version_string)
|
|
|
|
|
|
def connectSignalsSlots(self):
|
|
self.action_generateMission.triggered.connect(self.generateMissionAction)
|
|
self.action_scenarioSelected.triggered.connect(self.scenarioChanged)
|
|
self.action_defensiveModeChanged.triggered.connect(self.defensiveModeChanged)
|
|
|
|
def populateScenarios(self):
|
|
os.chdir(self.m.scenarios_dir)
|
|
path = os.getcwd()
|
|
dir_list = os.listdir(path)
|
|
logger.info("Looking for mission files in " + path)
|
|
|
|
for filename in dir_list:
|
|
if filename.endswith(".miz"):
|
|
scenarios.append(filename)
|
|
self.scenario_comboBox.addItem(filename.removesuffix('.miz'))
|
|
|
|
def populateForces(self, side, combobox, files_list):
|
|
os.chdir(self.m.home_dir)
|
|
os.chdir(self.m.forces_dir + "/" + side)
|
|
path = os.getcwd()
|
|
dir_list = os.listdir(path)
|
|
logger.info("Looking for " + side + " Forces files in '" + path)
|
|
|
|
for filename in dir_list:
|
|
if filename.endswith(".miz"):
|
|
files_list.append(filename)
|
|
combobox.addItem(filename.removesuffix('.miz'))
|
|
|
|
def populateSlotSelection(self):
|
|
self.slot_template_comboBox.addItem("Multiple Slots")
|
|
for type in RotorOpsUnits.client_helos:
|
|
self.slot_template_comboBox.addItem(type.id)
|
|
|
|
def defensiveModeChanged(self):
|
|
if self.defense_checkBox.isChecked():
|
|
self.red_forces_label.setText(attackers_text)
|
|
self.blue_forces_label.setText(defenders_text)
|
|
else:
|
|
self.red_forces_label.setText(defenders_text)
|
|
self.blue_forces_label.setText(attackers_text)
|
|
|
|
|
|
def scenarioChanged(self):
|
|
os.chdir(self.m.scenarios_dir)
|
|
filename = scenarios[self.scenario_comboBox.currentIndex()]
|
|
source_mission = dcs.mission.Mission()
|
|
source_mission.load_file(filename)
|
|
zones = source_mission.triggers.zones()
|
|
conflict_zones = 0
|
|
staging_zones = 0
|
|
conflict_zone_size_sum = 0
|
|
conflict_zone_distance_sum = 0
|
|
spawn_zones = 0
|
|
conflict_zone_positions = []
|
|
#friendly_airports = source_mission.getCoalitionAirports("blue")
|
|
#enemy_airports = source_mission.getCoalitionAirports("red")
|
|
friendly_airports = True
|
|
enemy_airports = True
|
|
|
|
for zone in zones:
|
|
if zone.name == "STAGING":
|
|
staging_zones += 1
|
|
if zone.name == "ALPHA" or zone.name == "BRAVO" or zone.name == "CHARLIE" or zone.name == "DELTA":
|
|
conflict_zones += 1
|
|
conflict_zone_size_sum += zone.radius
|
|
conflict_zone_positions.append(zone.position)
|
|
if zone.name.rfind("_SPAWN") > 0:
|
|
spawn_zones += 1
|
|
if conflict_zones > 1:
|
|
for index, position in enumerate(conflict_zone_positions):
|
|
if index > 0:
|
|
conflict_zone_distance_sum += RotorOpsUtils.getDistance(conflict_zone_positions[index], conflict_zone_positions[index - 1])
|
|
|
|
def validateTemplate():
|
|
valid = True
|
|
if len(staging_zones) < 1:
|
|
valid = False
|
|
if len(conflict_zones) < 1:
|
|
valid = False
|
|
if not friendly_airports:
|
|
valid = False
|
|
if not enemy_airports:
|
|
valid = False
|
|
return valid
|
|
|
|
if conflict_zones and staging_zones :
|
|
average_zone_size = conflict_zone_size_sum / conflict_zones
|
|
self.description_textBrowser.setText(
|
|
"Map: " + source_mission.terrain.name + "\n" +
|
|
"Conflict Zones: " + str(conflict_zones) + "\n" +
|
|
"Average Zone Size " + str(math.floor(average_zone_size)) + "m \n" +
|
|
"Infantry Spawn Zones: " + str(spawn_zones) + "\n" +
|
|
"Approx Distance: " + str(math.floor(RotorOpsUtils.convertMeterToNM(conflict_zone_distance_sum))) + "nm \n"
|
|
#"Validity Check:" + str(validateTemplate())
|
|
+ "\n== BRIEFING ==\n\n"
|
|
+ source_mission.description_text()
|
|
)
|
|
#self.description_textBrowser.setText("File error occured.")
|
|
|
|
|
|
def generateMissionAction(self):
|
|
red_forces_filename = red_forces_files[self.redforces_comboBox.currentIndex()]
|
|
blue_forces_filename = blue_forces_files[self.blueforces_comboBox.currentIndex()]
|
|
scenario_filename = scenarios[self.scenario_comboBox.currentIndex()]
|
|
data = {
|
|
"scenario_filename": scenario_filename,
|
|
"red_forces_filename": red_forces_filename,
|
|
"blue_forces_filename": blue_forces_filename,
|
|
"red_quantity": self.redqty_spinBox.value(),
|
|
"blue_quantity": self.blueqty_spinBox.value(),
|
|
"inf_spawn_qty": self.inf_spawn_spinBox.value(),
|
|
"apc_spawns_inf": self.apcs_spawn_checkBox.isChecked(),
|
|
"e_attack_helos": self.e_attack_helos_spinBox.value(),
|
|
"e_attack_planes": self.e_attack_planes_spinBox.value(),
|
|
"crates": self.logistics_crates_checkBox.isChecked(),
|
|
"f_awacs": self.awacs_checkBox.isChecked(),
|
|
"f_tankers": self.tankers_checkBox.isChecked(),
|
|
"voiceovers": self.voiceovers_checkBox.isChecked(),
|
|
"force_offroad": self.force_offroad_checkBox.isChecked(),
|
|
"game_display": self.game_status_checkBox.isChecked(),
|
|
"defending": self.defense_checkBox.isChecked(),
|
|
"slots": self.slot_template_comboBox.currentText(),
|
|
"zone_protect_sams": self.zone_sams_checkBox.isChecked(),
|
|
"zone_farps": self.farp_buttonGroup.checkedButton().objectName(),
|
|
"inf_spawn_msgs": self.inf_spawn_voiceovers_checkBox.isChecked(),
|
|
"e_transport_helos": self.e_transport_helos_spinBox.value(),
|
|
"transport_drop_qty": self.troop_drop_spinBox.value(),
|
|
"smoke_pickup_zones": self.smoke_pickup_zone_checkBox.isChecked(),
|
|
}
|
|
os.chdir(self.m.home_dir + '/Generator')
|
|
n = ROps.RotorOpsMission()
|
|
result = n.generateMission(data)
|
|
logger.info("Generating mission with options:")
|
|
logger.info(str(data))
|
|
|
|
# generate the mission
|
|
#result = self.m.generateMission(data)
|
|
|
|
#display results
|
|
if result["success"]:
|
|
logger.info(result["filename"] + "' successfully generated in " + result["directory"])
|
|
self.statusbar.showMessage(result["filename"] + "' successfully generated in " + result["directory"], 10000)
|
|
msg = QMessageBox()
|
|
msg.setWindowTitle("Mission Generated")
|
|
msg.setText("Awesome, your mission is ready! It's located in this directory: \n" +
|
|
self.m.output_dir + "\n" +
|
|
"\n" +
|
|
"Next, you should use the DCS Mission Editor to fine tune unit placements. Don't be afraid to edit the missions that this generator produces. \n" +
|
|
"\n" +
|
|
"There are no hidden script changes, everything is visible in the ME. Triggers have been created to help you to add your own actions based on active zone and game status. \n" +
|
|
"\n" +
|
|
"Units can be changed or moved without issue. Player slots can be changed or moved without issue. \n" +
|
|
"\n" +
|
|
"Don't forget, you can also create your own templates that can include any mission options, objects, or even scripts. \n" +
|
|
"\n" +
|
|
"Have fun! \n"
|
|
)
|
|
x = msg.exec_()
|
|
elif not result["success"]:
|
|
logger.warning(result["failure_msg"])
|
|
msg = QMessageBox()
|
|
msg.setWindowTitle("Error")
|
|
msg.setText(result["failure_msg"])
|
|
x = msg.exec_()
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app = QApplication(sys.argv)
|
|
win = Window()
|
|
win.show()
|
|
sys.exit(app.exec())
|
|
|
|
|
|
|