mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
705 lines
29 KiB
Python
705 lines
29 KiB
Python
import math
|
|
import operator
|
|
import random
|
|
|
|
from game import db
|
|
from game.data.doctrine import MODERN_DOCTRINE
|
|
from game.data.radar_db import UNITS_WITH_RADAR
|
|
from game.utils import meter_to_feet, nm_to_meter
|
|
from gen import Conflict
|
|
from gen.flights.ai_flight_planner_db import INTERCEPT_CAPABLE, CAP_CAPABLE, CAS_CAPABLE, SEAD_CAPABLE, STRIKE_CAPABLE
|
|
from gen.flights.flight import Flight, FlightType, FlightWaypoint, FlightWaypointType
|
|
|
|
|
|
MISSION_DURATION = 80
|
|
|
|
|
|
class FlightPlanner:
|
|
|
|
def __init__(self, from_cp, game):
|
|
# TODO : have the flight planner depend on a 'stance' setting : [Defensive, Aggresive... etc] and faction doctrine
|
|
# TODO : the flight planner should plan package and operations
|
|
self.from_cp = from_cp
|
|
self.game = game
|
|
self.aircraft_inventory = {} # local copy of the airbase inventory
|
|
|
|
if from_cp.captured:
|
|
self.faction = self.game.player_faction
|
|
else:
|
|
self.faction = self.game.enemy_faction
|
|
|
|
if "doctrine" in self.faction.keys():
|
|
self.doctrine = self.faction["doctrine"]
|
|
else:
|
|
self.doctrine = MODERN_DOCTRINE
|
|
|
|
|
|
def reset(self):
|
|
"""
|
|
Reset the planned flights and available units
|
|
"""
|
|
self.aircraft_inventory = dict({k: v for k, v in self.from_cp.base.aircraft.items()})
|
|
self.interceptor_flights = []
|
|
self.cap_flights = []
|
|
self.cas_flights = []
|
|
self.strike_flights = []
|
|
self.sead_flights = []
|
|
self.custom_flights = []
|
|
self.flights = []
|
|
self.potential_sead_targets = []
|
|
self.potential_strike_targets = []
|
|
|
|
def plan_flights(self):
|
|
|
|
self.reset()
|
|
self.compute_sead_targets()
|
|
self.compute_strike_targets()
|
|
|
|
# The priority is to assign air-superiority fighter or interceptor to interception roles, so they can scramble if there is an attacker
|
|
# self.commision_interceptors()
|
|
|
|
# Then some CAP patrol for the next 2 hours
|
|
self.commision_cap()
|
|
|
|
# Then setup cas
|
|
self.commision_cas()
|
|
|
|
# Then prepare some sead flights if required
|
|
self.commision_sead()
|
|
|
|
self.commision_strike()
|
|
|
|
# TODO : commision ANTISHIP
|
|
|
|
def remove_flight(self, index):
|
|
try:
|
|
flight = self.flights[index]
|
|
if flight in self.interceptor_flights: self.interceptor_flights.remove(flight)
|
|
if flight in self.cap_flights: self.cap_flights.remove(flight)
|
|
if flight in self.cas_flights: self.cas_flights.remove(flight)
|
|
if flight in self.strike_flights: self.strike_flights.remove(flight)
|
|
if flight in self.sead_flights: self.sead_flights.remove(flight)
|
|
if flight in self.custom_flights: self.custom_flights.remove(flight)
|
|
self.flights.remove(flight)
|
|
except IndexError:
|
|
return
|
|
|
|
|
|
def commision_interceptors(self):
|
|
"""
|
|
Pick some aircraft to assign them to interception roles
|
|
"""
|
|
|
|
# At least try to generate one interceptor group
|
|
number_of_interceptor_groups = min(max(sum([v for k, v in self.aircraft_inventory.items()]) / 4, self.doctrine["MAX_NUMBER_OF_INTERCEPTION_GROUP"]), 1)
|
|
possible_interceptors = [k for k in self.aircraft_inventory.keys() if k in INTERCEPT_CAPABLE]
|
|
|
|
if len(possible_interceptors) <= 0:
|
|
possible_interceptors = [k for k,v in self.aircraft_inventory.items() if k in CAP_CAPABLE and v >= 2]
|
|
|
|
if number_of_interceptor_groups > 0:
|
|
inventory = dict({k: v for k, v in self.aircraft_inventory.items() if k in possible_interceptors})
|
|
for i in range(number_of_interceptor_groups):
|
|
try:
|
|
unit = random.choice([k for k,v in inventory.items() if v >= 2])
|
|
except IndexError:
|
|
break
|
|
inventory[unit] = inventory[unit] - 2
|
|
flight = Flight(unit, 2, self.from_cp, FlightType.INTERCEPTION)
|
|
flight.scheduled_in = 1
|
|
flight.points = []
|
|
|
|
self.interceptor_flights.append(flight)
|
|
self.flights.append(flight)
|
|
|
|
# Update inventory
|
|
for k, v in inventory.items():
|
|
self.aircraft_inventory[k] = v
|
|
|
|
def commision_cap(self):
|
|
"""
|
|
Pick some aircraft to assign them to defensive CAP roles (BARCAP)
|
|
"""
|
|
|
|
possible_aircraft = [k for k, v in self.aircraft_inventory.items() if k in CAP_CAPABLE and v >= 2]
|
|
inventory = dict({k: v for k, v in self.aircraft_inventory.items() if k in possible_aircraft})
|
|
|
|
offset = random.randint(0,5)
|
|
for i in range(int(MISSION_DURATION/self.doctrine["CAP_EVERY_X_MINUTES"])):
|
|
|
|
try:
|
|
unit = random.choice([k for k, v in inventory.items() if v >= 2])
|
|
except IndexError:
|
|
break
|
|
|
|
inventory[unit] = inventory[unit] - 2
|
|
flight = Flight(unit, 2, self.from_cp, FlightType.CAP)
|
|
|
|
flight.points = []
|
|
flight.scheduled_in = offset + i*random.randint(self.doctrine["CAP_EVERY_X_MINUTES"] - 5, self.doctrine["CAP_EVERY_X_MINUTES"] + 5)
|
|
|
|
if len(self._get_cas_locations()) > 0:
|
|
enemy_cp = random.choice(self._get_cas_locations())
|
|
self.generate_frontline_cap(flight, flight.from_cp, enemy_cp)
|
|
else:
|
|
self.generate_barcap(flight, flight.from_cp)
|
|
|
|
self.cap_flights.append(flight)
|
|
self.flights.append(flight)
|
|
|
|
# Update inventory
|
|
for k, v in inventory.items():
|
|
self.aircraft_inventory[k] = v
|
|
|
|
def commision_cas(self):
|
|
"""
|
|
Pick some aircraft to assign them to CAS
|
|
"""
|
|
|
|
possible_aircraft = [k for k, v in self.aircraft_inventory.items() if k in CAS_CAPABLE and v >= 2]
|
|
inventory = dict({k: v for k, v in self.aircraft_inventory.items() if k in possible_aircraft})
|
|
cas_location = self._get_cas_locations()
|
|
|
|
if len(cas_location) > 0:
|
|
|
|
offset = random.randint(0,5)
|
|
for i in range(int(MISSION_DURATION/self.doctrine["CAS_EVERY_X_MINUTES"])):
|
|
|
|
try:
|
|
unit = random.choice([k for k, v in inventory.items() if v >= 2])
|
|
except IndexError:
|
|
break
|
|
|
|
inventory[unit] = inventory[unit] - 2
|
|
flight = Flight(unit, 2, self.from_cp, FlightType.CAS)
|
|
flight.points = []
|
|
flight.scheduled_in = offset + i * random.randint(self.doctrine["CAS_EVERY_X_MINUTES"] - 5, self.doctrine["CAS_EVERY_X_MINUTES"] + 5)
|
|
location = random.choice(cas_location)
|
|
|
|
self.generate_cas(flight, flight.from_cp, location)
|
|
|
|
self.cas_flights.append(flight)
|
|
self.flights.append(flight)
|
|
|
|
# Update inventory
|
|
for k, v in inventory.items():
|
|
self.aircraft_inventory[k] = v
|
|
|
|
def commision_sead(self):
|
|
"""
|
|
Pick some aircraft to assign them to SEAD tasks
|
|
"""
|
|
|
|
possible_aircraft = [k for k, v in self.aircraft_inventory.items() if k in SEAD_CAPABLE and v >= 2]
|
|
inventory = dict({k: v for k, v in self.aircraft_inventory.items() if k in possible_aircraft})
|
|
|
|
if len(self.potential_sead_targets) > 0:
|
|
|
|
offset = random.randint(0,5)
|
|
for i in range(int(MISSION_DURATION/self.doctrine["SEAD_EVERY_X_MINUTES"])):
|
|
|
|
if len(self.potential_sead_targets) <= 0:
|
|
break
|
|
|
|
try:
|
|
unit = random.choice([k for k, v in inventory.items() if v >= 2])
|
|
except IndexError:
|
|
break
|
|
|
|
inventory[unit] = inventory[unit] - 2
|
|
flight = Flight(unit, 2, self.from_cp, random.choice([FlightType.SEAD, FlightType.DEAD]))
|
|
|
|
flight.points = []
|
|
flight.scheduled_in = offset + i*random.randint(self.doctrine["SEAD_EVERY_X_MINUTES"] - 5, self.doctrine["SEAD_EVERY_X_MINUTES"] + 5)
|
|
|
|
location = self.potential_sead_targets[0][0]
|
|
self.potential_sead_targets.pop(0)
|
|
|
|
self.generate_sead(flight, location, [])
|
|
|
|
self.sead_flights.append(flight)
|
|
self.flights.append(flight)
|
|
|
|
# Update inventory
|
|
for k, v in inventory.items():
|
|
self.aircraft_inventory[k] = v
|
|
|
|
|
|
def commision_strike(self):
|
|
"""
|
|
Pick some aircraft to assign them to STRIKE tasks
|
|
"""
|
|
possible_aircraft = [k for k, v in self.aircraft_inventory.items() if k in STRIKE_CAPABLE and v >= 2]
|
|
inventory = dict({k: v for k, v in self.aircraft_inventory.items() if k in possible_aircraft})
|
|
|
|
if len(self.potential_strike_targets) > 0:
|
|
|
|
offset = random.randint(0,5)
|
|
for i in range(int(MISSION_DURATION/self.doctrine["STRIKE_EVERY_X_MINUTES"])):
|
|
|
|
if len(self.potential_strike_targets) <= 0:
|
|
break
|
|
|
|
try:
|
|
unit = random.choice([k for k, v in inventory.items() if v >= 2])
|
|
except IndexError:
|
|
break
|
|
|
|
inventory[unit] = inventory[unit] - 2
|
|
flight = Flight(unit, 2, self.from_cp, FlightType.STRIKE)
|
|
|
|
flight.points = []
|
|
flight.scheduled_in = offset + i*random.randint(self.doctrine["STRIKE_EVERY_X_MINUTES"] - 5, self.doctrine["STRIKE_EVERY_X_MINUTES"] + 5)
|
|
|
|
location = self.potential_strike_targets[0][0]
|
|
self.potential_strike_targets.pop(0)
|
|
|
|
self.generate_strike(flight, location)
|
|
|
|
self.strike_flights.append(flight)
|
|
self.flights.append(flight)
|
|
|
|
# Update inventory
|
|
for k, v in inventory.items():
|
|
self.aircraft_inventory[k] = v
|
|
|
|
def _get_cas_locations(self):
|
|
return self._get_cas_locations_for_cp(self.from_cp)
|
|
|
|
def _get_cas_locations_for_cp(self, for_cp):
|
|
cas_locations = []
|
|
for cp in for_cp.connected_points:
|
|
if cp.captured != for_cp.captured:
|
|
cas_locations.append(cp)
|
|
return cas_locations
|
|
|
|
def compute_strike_targets(self):
|
|
"""
|
|
@return a list of potential strike targets in range
|
|
"""
|
|
|
|
# target, distance
|
|
self.potential_strike_targets = []
|
|
|
|
for cp in [c for c in self.game.theater.controlpoints if c.captured != self.from_cp.captured]:
|
|
|
|
# Compute distance to current cp
|
|
distance = math.hypot(cp.position.x - self.from_cp.position.x,
|
|
cp.position.y - self.from_cp.position.y)
|
|
|
|
if distance > 2*self.doctrine["STRIKE_MAX_RANGE"]:
|
|
# Then it's unlikely any child ground object is in range
|
|
return
|
|
|
|
added_group = []
|
|
for g in cp.ground_objects:
|
|
if g.group_id in added_group or g.is_dead: continue
|
|
|
|
# Compute distance to current cp
|
|
distance = math.hypot(cp.position.x - self.from_cp.position.x,
|
|
cp.position.y - self.from_cp.position.y)
|
|
|
|
if distance < self.doctrine["SEAD_MAX_RANGE"]:
|
|
self.potential_strike_targets.append((g, distance))
|
|
added_group.append(g)
|
|
|
|
self.potential_strike_targets.sort(key=operator.itemgetter(1))
|
|
|
|
def compute_sead_targets(self):
|
|
"""
|
|
@return a list of potential sead targets in range
|
|
"""
|
|
|
|
# target, distance
|
|
self.potential_sead_targets = []
|
|
|
|
for cp in [c for c in self.game.theater.controlpoints if c.captured != self.from_cp.captured]:
|
|
|
|
# Compute distance to current cp
|
|
distance = math.hypot(cp.position.x - self.from_cp.position.x,
|
|
cp.position.y - self.from_cp.position.y)
|
|
|
|
# Then it's unlikely any ground object is range
|
|
if distance > 2*self.doctrine["SEAD_MAX_RANGE"]:
|
|
return
|
|
|
|
for g in cp.ground_objects:
|
|
|
|
if g.dcs_identifier == "AA":
|
|
|
|
# Check that there is at least one unit with a radar in the ground objects unit groups
|
|
number_of_units = sum([len([r for r in group.units if db.unit_type_from_name(r.type) in UNITS_WITH_RADAR]) for group in g.groups])
|
|
if number_of_units <= 0:
|
|
continue
|
|
|
|
# Compute distance to current cp
|
|
distance = math.hypot(cp.position.x - self.from_cp.position.x,
|
|
cp.position.y - self.from_cp.position.y)
|
|
|
|
if distance < self.doctrine["SEAD_MAX_RANGE"]:
|
|
self.potential_sead_targets.append((g, distance))
|
|
|
|
self.potential_sead_targets.sort(key=operator.itemgetter(1))
|
|
|
|
def __repr__(self):
|
|
return "-"*40 + "\n" + self.from_cp.name + " planned flights :\n"\
|
|
+ "-"*40 + "\n" + "\n".join([repr(f) for f in self.flights]) + "\n" + "-"*40
|
|
|
|
def get_available_aircraft(self):
|
|
base_aircraft_inventory = dict({k: v for k, v in self.from_cp.base.aircraft.items()})
|
|
for f in self.flights:
|
|
if f.unit_type in base_aircraft_inventory.keys():
|
|
base_aircraft_inventory[f.unit_type] = base_aircraft_inventory[f.unit_type] - f.count
|
|
if base_aircraft_inventory[f.unit_type] <= 0:
|
|
del base_aircraft_inventory[f.unit_type]
|
|
return base_aircraft_inventory
|
|
|
|
|
|
def generate_strike(self, flight, location):
|
|
|
|
flight.flight_type = FlightType.STRIKE
|
|
ascend = self.generate_ascend_point(flight.from_cp)
|
|
flight.points.append(ascend)
|
|
|
|
heading = flight.from_cp.position.heading_between_point(location.position)
|
|
ingress_heading = heading - 180 + 25
|
|
egress_heading = heading - 180 - 25
|
|
|
|
ingress_pos = location.position.point_from_heading(ingress_heading, self.doctrine["INGRESS_EGRESS_DISTANCE"])
|
|
ingress_point = FlightWaypoint(ingress_pos.x, ingress_pos.y, self.doctrine["INGRESS_ALT"])
|
|
ingress_point.pretty_name = "INGRESS on " + location.obj_name
|
|
ingress_point.description = "INGRESS on " + location.obj_name
|
|
ingress_point.name = "INGRESS"
|
|
ingress_point.waypoint_type = FlightWaypointType.INGRESS_STRIKE
|
|
flight.points.append(ingress_point)
|
|
|
|
if len(location.groups) > 0 and location.dcs_identifier == "AA":
|
|
for g in location.groups:
|
|
for j, u in enumerate(g.units):
|
|
point = FlightWaypoint(u.position.x, u.position.y, 0)
|
|
point.description = "STRIKE " + "[" + str(location.obj_name) + "] : " + u.type + " #" + str(j)
|
|
point.pretty_name = "STRIKE " + "[" + str(location.obj_name) + "] : " + u.type + " #" + str(j)
|
|
point.name = location.obj_name + "#" + str(j)
|
|
point.only_for_player = True
|
|
ingress_point.targets.append(location)
|
|
flight.points.append(point)
|
|
else:
|
|
if hasattr(location, "obj_name"):
|
|
buildings = self.game.theater.find_ground_objects_by_obj_name(location.obj_name)
|
|
print(buildings)
|
|
for building in buildings:
|
|
print("BUILDING " + str(building.is_dead) + " " + str(building.dcs_identifier))
|
|
if building.is_dead:
|
|
continue
|
|
|
|
point = FlightWaypoint(building.position.x, building.position.y, 0)
|
|
point.description = "STRIKE on " + building.obj_name + " " + building.category + " [" + str(building.dcs_identifier) + " ]"
|
|
point.pretty_name = "STRIKE on " + building.obj_name + " " + building.category + " [" + str(building.dcs_identifier) + " ]"
|
|
point.name = building.obj_name
|
|
point.only_for_player = True
|
|
ingress_point.targets.append(building)
|
|
flight.points.append(point)
|
|
else:
|
|
point = FlightWaypoint(location.position.x, location.position.y, 0)
|
|
point.description = "STRIKE on " + location.obj_name
|
|
point.pretty_name = "STRIKE on " + location.obj_name
|
|
point.name = location.obj_name
|
|
point.only_for_player = True
|
|
ingress_point.targets.append(location)
|
|
flight.points.append(point)
|
|
|
|
egress_pos = location.position.point_from_heading(egress_heading, self.doctrine["INGRESS_EGRESS_DISTANCE"])
|
|
egress_point = FlightWaypoint(egress_pos.x, egress_pos.y, self.doctrine["EGRESS_ALT"])
|
|
egress_point.name = "EGRESS"
|
|
egress_point.pretty_name = "EGRESS from " + location.obj_name
|
|
egress_point.description = "EGRESS from " + location.obj_name
|
|
egress_point.waypoint_type = FlightWaypointType.EGRESS
|
|
flight.points.append(egress_point)
|
|
|
|
descend = self.generate_descend_point(flight.from_cp)
|
|
flight.points.append(descend)
|
|
|
|
rtb = self.generate_rtb_waypoint(flight.from_cp)
|
|
flight.points.append(rtb)
|
|
|
|
def generate_barcap(self, flight, for_cp):
|
|
"""
|
|
Generate a barcap flight at a given location
|
|
:param flight: Flight to setup
|
|
:param for_cp: CP to protect
|
|
"""
|
|
flight.flight_type = FlightType.BARCAP if for_cp.is_carrier else FlightType.CAP
|
|
patrol_alt = random.randint(self.doctrine["PATROL_ALT_RANGE"][0], self.doctrine["PATROL_ALT_RANGE"][1])
|
|
|
|
if len(for_cp.ground_objects) > 0:
|
|
loc = random.choice(for_cp.ground_objects)
|
|
hdg = for_cp.position.heading_between_point(loc.position)
|
|
radius = random.randint(self.doctrine["CAP_PATTERN_LENGTH"][0], self.doctrine["CAP_PATTERN_LENGTH"][1])
|
|
orbit0p = loc.position.point_from_heading(hdg - 90, radius)
|
|
orbit1p = loc.position.point_from_heading(hdg + 90, radius)
|
|
else:
|
|
loc = for_cp.position.point_from_heading(random.randint(0, 360), random.randint(self.doctrine["CAP_DISTANCE_FROM_CP"][0], self.doctrine["CAP_DISTANCE_FROM_CP"][1]))
|
|
hdg = for_cp.position.heading_between_point(loc)
|
|
radius = random.randint(self.doctrine["CAP_PATTERN_LENGTH"][0], self.doctrine["CAP_PATTERN_LENGTH"][1])
|
|
orbit0p = loc.point_from_heading(hdg - 90, radius)
|
|
orbit1p = loc.point_from_heading(hdg + 90, radius)
|
|
|
|
# Create points
|
|
ascend = self.generate_ascend_point(flight.from_cp)
|
|
flight.points.append(ascend)
|
|
|
|
orbit0 = FlightWaypoint(orbit0p.x, orbit0p.y, patrol_alt)
|
|
orbit0.name = "ORBIT 0"
|
|
orbit0.description = "Standby between this point and the next one"
|
|
orbit0.pretty_name = "Race-track start"
|
|
orbit0.waypoint_type = FlightWaypointType.PATROL_TRACK
|
|
flight.points.append(orbit0)
|
|
|
|
orbit1 = FlightWaypoint(orbit1p.x, orbit1p.y, patrol_alt)
|
|
orbit1.name = "ORBIT 1"
|
|
orbit1.description = "Standby between this point and the previous one"
|
|
orbit1.pretty_name = "Race-track end"
|
|
orbit1.waypoint_type = FlightWaypointType.PATROL
|
|
flight.points.append(orbit1)
|
|
|
|
orbit0.targets.append(for_cp)
|
|
obj_added = []
|
|
for ground_object in for_cp.ground_objects:
|
|
if ground_object.obj_name not in obj_added and not ground_object.airbase_group:
|
|
orbit0.targets.append(ground_object)
|
|
obj_added.append(ground_object.obj_name)
|
|
|
|
descend = self.generate_descend_point(flight.from_cp)
|
|
flight.points.append(descend)
|
|
|
|
rtb = self.generate_rtb_waypoint(flight.from_cp)
|
|
flight.points.append(rtb)
|
|
|
|
|
|
def generate_frontline_cap(self, flight, ally_cp, enemy_cp):
|
|
"""
|
|
Generate a cap flight for the frontline between ally_cp and enemy cp in order to ensure air superiority and
|
|
protect friendly CAP airbase
|
|
:param flight: Flight to setup
|
|
:param ally_cp: CP to protect
|
|
:param enemy_cp: Enemy connected cp
|
|
"""
|
|
flight.flight_type = FlightType.CAP
|
|
patrol_alt = random.randint(self.doctrine["PATROL_ALT_RANGE"][0], self.doctrine["PATROL_ALT_RANGE"][1])
|
|
|
|
# Find targets waypoints
|
|
ingress, heading, distance = Conflict.frontline_vector(ally_cp, enemy_cp, self.game.theater)
|
|
center = ingress.point_from_heading(heading, distance / 2)
|
|
orbit_center = center.point_from_heading(heading - 90, random.randint(nm_to_meter(6), nm_to_meter(15)))
|
|
|
|
combat_width = distance / 2
|
|
if combat_width > 500000:
|
|
combat_width = 500000
|
|
if combat_width < 35000:
|
|
combat_width = 35000
|
|
|
|
radius = combat_width*1.25
|
|
orbit0p = orbit_center.point_from_heading(heading, radius)
|
|
orbit1p = orbit_center.point_from_heading(heading + 180, radius)
|
|
|
|
# Create points
|
|
ascend = self.generate_ascend_point(flight.from_cp)
|
|
flight.points.append(ascend)
|
|
|
|
orbit0 = FlightWaypoint(orbit0p.x, orbit0p.y, patrol_alt)
|
|
orbit0.name = "ORBIT 0"
|
|
orbit0.description = "Standby between this point and the next one"
|
|
orbit0.pretty_name = "Race-track start"
|
|
orbit0.waypoint_type = FlightWaypointType.PATROL_TRACK
|
|
flight.points.append(orbit0)
|
|
|
|
orbit1 = FlightWaypoint(orbit1p.x, orbit1p.y, patrol_alt)
|
|
orbit1.name = "ORBIT 1"
|
|
orbit1.description = "Standby between this point and the previous one"
|
|
orbit1.pretty_name = "Race-track end"
|
|
orbit1.waypoint_type = FlightWaypointType.PATROL
|
|
flight.points.append(orbit1)
|
|
|
|
# Note : Targets of a PATROL TRACK waypoints are the points to be defended
|
|
orbit0.targets.append(flight.from_cp)
|
|
orbit0.targets.append(center)
|
|
|
|
descend = self.generate_descend_point(flight.from_cp)
|
|
flight.points.append(descend)
|
|
|
|
rtb = self.generate_rtb_waypoint(flight.from_cp)
|
|
flight.points.append(rtb)
|
|
|
|
|
|
def generate_sead(self, flight, location, custom_targets = []):
|
|
"""
|
|
Generate a sead flight at a given location
|
|
:param flight: Flight to setup
|
|
:param location: Location of the SEAD target
|
|
:param custom_targets: Custom targets if any
|
|
"""
|
|
flight.points = []
|
|
flight.flight_type = random.choice([FlightType.SEAD, FlightType.DEAD])
|
|
|
|
ascend = self.generate_ascend_point(flight.from_cp)
|
|
flight.points.append(ascend)
|
|
|
|
heading = flight.from_cp.position.heading_between_point(location.position)
|
|
ingress_heading = heading - 180 + 25
|
|
egress_heading = heading - 180 - 25
|
|
|
|
ingress_pos = location.position.point_from_heading(ingress_heading, self.doctrine["INGRESS_EGRESS_DISTANCE"])
|
|
ingress_point = FlightWaypoint(ingress_pos.x, ingress_pos.y, self.doctrine["INGRESS_ALT"])
|
|
ingress_point.name = "INGRESS"
|
|
ingress_point.pretty_name = "INGRESS on " + location.obj_name
|
|
ingress_point.description = "INGRESS on " + location.obj_name
|
|
ingress_point.waypoint_type = FlightWaypointType.INGRESS_SEAD
|
|
flight.points.append(ingress_point)
|
|
|
|
if len(custom_targets) > 0:
|
|
for target in custom_targets:
|
|
point = FlightWaypoint(target.position.x, target.position.y, 0)
|
|
point.alt_type = "RADIO"
|
|
if flight.flight_type == FlightType.DEAD:
|
|
point.description = "SEAD on " + target.type
|
|
point.pretty_name = "SEAD on " + location.obj_name
|
|
point.only_for_player = True
|
|
else:
|
|
point.description = "DEAD on " + location.obj_name
|
|
point.pretty_name = "DEAD on " + location.obj_name
|
|
point.only_for_player = True
|
|
ingress_point.targets.append(location)
|
|
ingress_point.targetGroup = location
|
|
flight.points.append(point)
|
|
else:
|
|
point = FlightWaypoint(location.position.x, location.position.y, 0)
|
|
point.alt_type = "RADIO"
|
|
if flight.flight_type == FlightType.DEAD:
|
|
point.description = "SEAD on " + location.obj_name
|
|
point.pretty_name = "SEAD on " + location.obj_name
|
|
point.only_for_player = True
|
|
else:
|
|
point.description = "DEAD on " + location.obj_name
|
|
point.pretty_name = "DEAD on " + location.obj_name
|
|
point.only_for_player = True
|
|
ingress_point.targets.append(location)
|
|
ingress_point.targetGroup = location
|
|
flight.points.append(point)
|
|
|
|
egress_pos = location.position.point_from_heading(egress_heading, self.doctrine["INGRESS_EGRESS_DISTANCE"])
|
|
egress_point = FlightWaypoint(egress_pos.x, egress_pos.y, self.doctrine["EGRESS_ALT"])
|
|
egress_point.name = "EGRESS"
|
|
egress_point.pretty_name = "EGRESS from " + location.obj_name
|
|
egress_point.description = "EGRESS from " + location.obj_name
|
|
egress_point.waypoint_type = FlightWaypointType.EGRESS
|
|
flight.points.append(egress_point)
|
|
|
|
descend = self.generate_descend_point(flight.from_cp)
|
|
flight.points.append(descend)
|
|
|
|
rtb = self.generate_rtb_waypoint(flight.from_cp)
|
|
flight.points.append(rtb)
|
|
|
|
|
|
def generate_cas(self, flight, from_cp, location):
|
|
"""
|
|
Generate a CAS flight at a given location
|
|
:param flight: Flight to setup
|
|
:param location: Location of the CAS targets
|
|
"""
|
|
is_helo = hasattr(flight.unit_type, "helicopter") and flight.unit_type.helicopter
|
|
cap_alt = 1000
|
|
flight.points = []
|
|
flight.flight_type = FlightType.CAS
|
|
|
|
ingress, heading, distance = Conflict.frontline_vector(from_cp, location, self.game.theater)
|
|
center = ingress.point_from_heading(heading, distance / 2)
|
|
egress = ingress.point_from_heading(heading, distance)
|
|
|
|
ascend = self.generate_ascend_point(flight.from_cp)
|
|
if is_helo:
|
|
cap_alt = 500
|
|
ascend.alt = 500
|
|
flight.points.append(ascend)
|
|
|
|
ingress_point = FlightWaypoint(ingress.x, ingress.y, cap_alt)
|
|
ingress_point.alt_type = "RADIO"
|
|
ingress_point.name = "INGRESS"
|
|
ingress_point.pretty_name = "INGRESS"
|
|
ingress_point.description = "Ingress into CAS area"
|
|
ingress_point.waypoint_type = FlightWaypointType.INGRESS_CAS
|
|
flight.points.append(ingress_point)
|
|
|
|
center_point = FlightWaypoint(center.x, center.y, cap_alt)
|
|
center_point.alt_type = "RADIO"
|
|
center_point.description = "Provide CAS"
|
|
center_point.name = "CAS"
|
|
center_point.pretty_name = "CAS"
|
|
center_point.waypoint_type = FlightWaypointType.CAS
|
|
flight.points.append(center_point)
|
|
|
|
egress_point = FlightWaypoint(egress.x, egress.y, cap_alt)
|
|
egress_point.alt_type = "RADIO"
|
|
egress_point.description = "Egress from CAS area"
|
|
egress_point.name = "EGRESS"
|
|
egress_point.pretty_name = "EGRESS"
|
|
egress_point.waypoint_type = FlightWaypointType.EGRESS
|
|
flight.points.append(egress_point)
|
|
|
|
descend = self.generate_descend_point(flight.from_cp)
|
|
if is_helo:
|
|
descend.alt = 300
|
|
flight.points.append(descend)
|
|
|
|
rtb = self.generate_rtb_waypoint(flight.from_cp)
|
|
flight.points.append(rtb)
|
|
|
|
|
|
def generate_ascend_point(self, from_cp):
|
|
"""
|
|
Generate ascend point
|
|
:param from_cp: Airport you're taking off from
|
|
:return:
|
|
"""
|
|
ascend_heading = from_cp.heading
|
|
pos_ascend = from_cp.position.point_from_heading(ascend_heading, 10000)
|
|
ascend = FlightWaypoint(pos_ascend.x, pos_ascend.y, self.doctrine["PATTERN_ALTITUDE"])
|
|
ascend.name = "ASCEND"
|
|
ascend.alt_type = "RADIO"
|
|
ascend.description = "Ascend"
|
|
ascend.pretty_name = "Ascend"
|
|
ascend.waypoint_type = FlightWaypointType.ASCEND_POINT
|
|
return ascend
|
|
|
|
|
|
def generate_descend_point(self, from_cp):
|
|
"""
|
|
Generate approach/descend point
|
|
:param from_cp: Airport you're landing at
|
|
:return:
|
|
"""
|
|
ascend_heading = from_cp.heading
|
|
descend = from_cp.position.point_from_heading(ascend_heading - 180, 10000)
|
|
descend = FlightWaypoint(descend.x, descend.y, self.doctrine["PATTERN_ALTITUDE"])
|
|
descend.name = "DESCEND"
|
|
descend.alt_type = "RADIO"
|
|
descend.description = "Descend to pattern alt"
|
|
descend.pretty_name = "Descend to pattern alt"
|
|
descend.waypoint_type = FlightWaypointType.DESCENT_POINT
|
|
return descend
|
|
|
|
|
|
def generate_rtb_waypoint(self, from_cp):
|
|
"""
|
|
Generate RTB landing point
|
|
:param from_cp: Airport you're landing at
|
|
:return:
|
|
"""
|
|
rtb = from_cp.position
|
|
rtb = FlightWaypoint(rtb.x, rtb.y, 0)
|
|
rtb.name = "LANDING"
|
|
rtb.alt_type = "RADIO"
|
|
rtb.description = "RTB"
|
|
rtb.pretty_name = "RTB"
|
|
rtb.waypoint_type = FlightWaypointType.LANDING_POINT
|
|
return rtb |