from __future__ import annotations import logging from typing import TYPE_CHECKING, List from dcs import Point from dcs.action import ( ClearFlag, DoScript, MarkToAll, SetFlag, RemoveSceneObjects, RemoveSceneObjectsMask, SceneryDestructionZone, Smoke, ) from dcs.condition import ( AllOfCoalitionOutsideZone, FlagIsFalse, FlagIsTrue, PartOfCoalitionInZone, TimeAfter, TimeSinceFlag, ) from dcs.mission import Mission from dcs.task import Option from dcs.translation import String from dcs.triggers import Event, TriggerCondition, TriggerOnce from dcs.unit import Skill from game.theater import Airfield from game.theater.controlpoint import Fob, TRIGGER_RADIUS_CAPTURE if TYPE_CHECKING: from game.game import Game PUSH_TRIGGER_SIZE = 3000 PUSH_TRIGGER_ACTIVATION_AGL = 25 REGROUP_ZONE_DISTANCE = 12000 REGROUP_ALT = 5000 TRIGGER_WAYPOINT_OFFSET = 2 TRIGGER_MIN_DISTANCE_FROM_START = 10000 # modified since we now have advanced SAM units TRIGGER_RADIUS_MINIMUM = 3000000 TRIGGER_RADIUS_SMALL = 50000 TRIGGER_RADIUS_MEDIUM = 100000 TRIGGER_RADIUS_LARGE = 150000 TRIGGER_RADIUS_ALL_MAP = 3000000 TRIGGER_RADIUS_CLEAR_SCENERY = 1000 TRIGGER_RADIUS_PRETENSE_TGO = 500 TRIGGER_RADIUS_PRETENSE_SUPPLY = 500 TRIGGER_RADIUS_PRETENSE_HELI = 1000 class Silence(Option): Key = 7 class PretenseTriggerGenerator: capture_zone_types = (Fob, Airfield) capture_zone_flag = 600 def __init__(self, mission: Mission, game: Game) -> None: self.mission = mission self.game = game def _set_allegiances(self, player_coalition: str, enemy_coalition: str) -> None: """ Set airbase initial coalition """ # Empty neutrals airports airfields = [ cp for cp in self.game.theater.controlpoints if isinstance(cp, Airfield) ] airport_ids = {cp.airport.id for cp in airfields} for airport in self.mission.terrain.airport_list(): if airport.id not in airport_ids: airport.unlimited_fuel = False airport.unlimited_munitions = False airport.unlimited_aircrafts = False airport.gasoline_init = 0 airport.methanol_mixture_init = 0 airport.diesel_init = 0 airport.jet_init = 0 airport.operating_level_air = 0 airport.operating_level_equipment = 0 airport.operating_level_fuel = 0 for airport in self.mission.terrain.airport_list(): if airport.id not in airport_ids: airport.unlimited_fuel = True airport.unlimited_munitions = True airport.unlimited_aircrafts = True for airfield in airfields: cp_airport = self.mission.terrain.airport_by_id(airfield.airport.id) if cp_airport is None: raise RuntimeError( f"Could not find {airfield.airport.name} in the mission" ) cp_airport.set_coalition( airfield.captured and player_coalition or enemy_coalition ) def _set_skill(self, player_coalition: str, enemy_coalition: str) -> None: """ Set skill level for all aircraft in the mission """ for coalition_name, coalition in self.mission.coalition.items(): if coalition_name == player_coalition: skill_level = Skill(self.game.settings.player_skill) elif coalition_name == enemy_coalition: skill_level = Skill(self.game.settings.enemy_vehicle_skill) else: continue for country in coalition.countries.values(): for vehicle_group in country.vehicle_group: vehicle_group.set_skill(skill_level) def _gen_markers(self) -> None: """ Generate markers on F10 map for each existing objective """ if self.game.settings.generate_marks: mark_trigger = TriggerOnce(Event.NoEvent, "Marks generator") mark_trigger.add_condition(TimeAfter(1)) v = 10 for cp in self.game.theater.controlpoints: seen = set() for ground_object in cp.ground_objects: if ground_object.obj_name in seen: continue seen.add(ground_object.obj_name) for location in ground_object.mark_locations: zone = self.mission.triggers.add_triggerzone( location, radius=10, hidden=True, name="MARK" ) if cp.captured: name = ground_object.obj_name + " [ALLY]" else: name = ground_object.obj_name + " [ENEMY]" mark_trigger.add_action(MarkToAll(v, zone.id, String(name))) v += 1 self.mission.triggerrules.triggers.append(mark_trigger) def _generate_pretense_zone_triggers( self, player_coalition: str, enemy_coalition: str ) -> None: """Creates a pair of triggers for each control point of `cls.capture_zone_types`. One for the initial capture of a control point, and one if it is recaptured. Directly appends to the global `base_capture_events` var declared by `dcs_libaration.lua` """ for cp in self.game.theater.controlpoints: if isinstance(cp, self.capture_zone_types) and not cp.is_fleet: zone_color = {1: 0.0, 2: 0.0, 3: 0.0, 4: 0.15} trigger_zone = self.mission.triggers.add_triggerzone( cp.position, radius=TRIGGER_RADIUS_CAPTURE, hidden=False, name=cp.name, color=zone_color, ) cp_name_trimmed = "".join([i for i in cp.name.lower() if i.isalnum()]) tgo_num = 0 for tgo in cp.ground_objects: if cp.is_fleet or tgo.sea_object: continue tgo_num += 1 zone_color = {1: 1.0, 2: 1.0, 3: 1.0, 4: 0.15} trigger_zone = self.mission.triggers.add_triggerzone( tgo.position, radius=TRIGGER_RADIUS_PRETENSE_TGO, hidden=False, name=f"{cp_name_trimmed}-{tgo_num}", color=zone_color, ) for helipad in cp.helipads + cp.helipads_invisible + cp.helipads_quad: zone_color = {1: 1.0, 2: 1.0, 3: 1.0, 4: 0.15} trigger_zone = self.mission.triggers.add_triggerzone( position=helipad, radius=TRIGGER_RADIUS_PRETENSE_HELI, hidden=False, name=f"{cp_name_trimmed}-hsp", color=zone_color, ) break for supply_route in cp.convoy_routes.values(): tgo_num += 1 zone_color = {1: 1.0, 2: 1.0, 3: 1.0, 4: 0.15} origin_position = supply_route[0] next_position = supply_route[1] convoy_heading = origin_position.heading_between_point(next_position) supply_position = origin_position.point_from_heading( convoy_heading, 300 ) trigger_zone = self.mission.triggers.add_triggerzone( supply_position, radius=TRIGGER_RADIUS_PRETENSE_TGO, hidden=False, name=f"{cp_name_trimmed}-sp", color=zone_color, ) break def generate(self) -> None: player_coalition = "blue" enemy_coalition = "red" self._set_skill(player_coalition, enemy_coalition) self._set_allegiances(player_coalition, enemy_coalition) self._gen_markers() self._generate_pretense_zone_triggers(player_coalition, enemy_coalition) @classmethod def get_capture_zone_flag(cls) -> int: flag = cls.capture_zone_flag cls.capture_zone_flag += 1 return flag