mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
105 lines
3.6 KiB
Python
105 lines
3.6 KiB
Python
import typing
|
|
import pdb
|
|
import dcs
|
|
|
|
from random import randint
|
|
|
|
import globals
|
|
|
|
from .conflictgen import *
|
|
from .naming import *
|
|
|
|
from dcs.mission import *
|
|
from dcs.vehicles import *
|
|
from dcs.unitgroup import *
|
|
from dcs.unittype import *
|
|
from dcs.mapping import *
|
|
from dcs.point import *
|
|
from dcs.task import *
|
|
|
|
SPREAD_DISTANCE_FACTOR = 0.1, 0.25
|
|
SPREAD_DISTANCE_SIZE_FACTOR = 0.5
|
|
ESCORT_MAX_DIST = 30000
|
|
|
|
WARM_START_ALTITUDE = 6000
|
|
WARM_START_AIRSPEED = 300
|
|
CAS_ALTITUDE = 3000
|
|
|
|
class AircraftConflictGenerator:
|
|
escort_targets = [] # type: typing.List[PlaneGroup]
|
|
|
|
def __init__(self, mission: Mission, conflict: Conflict):
|
|
self.m = mission
|
|
self.conflict = conflict
|
|
|
|
def _group_point(self, point) -> Point:
|
|
distance = randint(
|
|
int(self.conflict.size * SPREAD_DISTANCE_FACTOR[0]),
|
|
int(self.conflict.size * SPREAD_DISTANCE_FACTOR[1]),
|
|
)
|
|
|
|
return point.random_point_within(distance, self.conflict.size * SPREAD_DISTANCE_SIZE_FACTOR)
|
|
|
|
def _generate_group(
|
|
self,
|
|
name: str,
|
|
side: Country,
|
|
unit: UnitType,
|
|
count: int,
|
|
at: Point = None,
|
|
airport: Airport = None) -> PlaneGroup:
|
|
starttype = airport == None and StartType.Warm or StartType.Runway
|
|
return self.m.flight_group(
|
|
country=side,
|
|
name=name,
|
|
aircraft_type=unit,
|
|
airport=airport,
|
|
position=at,
|
|
altitude=WARM_START_ALTITUDE,
|
|
speed=WARM_START_AIRSPEED,
|
|
maintask=None,
|
|
start_type=starttype,
|
|
group_size=count)
|
|
|
|
def generate_cas(self, attackers: typing.Dict[PlaneType, int], airport: Airport = None):
|
|
for type, count in attackers.items():
|
|
group = self._generate_group(
|
|
name=namegen.next_cas_group_name(),
|
|
side=self.conflict.attackers_side,
|
|
unit=type,
|
|
count=count,
|
|
at=airport == None and self._group_point(self.conflict.air_attackers_location) or None,
|
|
airport=airport)
|
|
self.escort_targets.append(group)
|
|
|
|
group.add_waypoint(self.conflict.point, CAS_ALTITUDE)
|
|
group.task = CAS.name
|
|
|
|
def generate_escort(self, attackers: typing.Dict[PlaneType, int], airport: Airport = None):
|
|
for type, count in attackers.items():
|
|
group = self._generate_group(
|
|
name=namegen.next_escort_group_name(),
|
|
side=self.conflict.attackers_side,
|
|
unit=type,
|
|
count=count,
|
|
at=airport == None and self._group_point(self.conflict.air_attackers_location) or None,
|
|
airport=airport)
|
|
|
|
group.task = Escort.name
|
|
for group in self.escort_targets:
|
|
group.tasks.append(EscortTaskAction(group.id, engagement_max_dist=ESCORT_MAX_DIST))
|
|
|
|
|
|
def generate_interceptors(self, defenders: typing.Dict[PlaneType, int], airport: Airport = None):
|
|
for type, count in defenders.items():
|
|
group = self._generate_group(
|
|
name=namegen.next_intercept_group_name(),
|
|
side=self.conflict.defenders_side,
|
|
unit=type,
|
|
count=count,
|
|
at=airport == None and self._group_point(self.conflict.air_defenders_location) or None,
|
|
airport=airport)
|
|
|
|
group.add_waypoint(self.conflict.point, CAS_ALTITUDE)
|
|
group.task = FighterSweep()
|