diff --git a/game/ato/flightplans/seadsweep.py b/game/ato/flightplans/seadsweep.py index cdd09270..d9f1c1d3 100644 --- a/game/ato/flightplans/seadsweep.py +++ b/game/ato/flightplans/seadsweep.py @@ -8,10 +8,12 @@ from .formationattack import ( FormationAttackFlightPlan, FormationAttackLayout, ) +from .uizonedisplay import UiZoneDisplay, UiZone from ..flightwaypointtype import FlightWaypointType +from ...utils import nautical_miles -class SeadSweepFlightPlan(FormationAttackFlightPlan): +class SeadSweepFlightPlan(FormationAttackFlightPlan, UiZoneDisplay): @staticmethod def builder_type() -> Type[Builder]: return Builder @@ -19,6 +21,14 @@ class SeadSweepFlightPlan(FormationAttackFlightPlan): def default_tot_offset(self) -> timedelta: return -timedelta(minutes=2) + def ui_zone(self) -> UiZone: + return UiZone( + [self.tot_waypoint.position], + nautical_miles( + self.flight.coalition.game.settings.sead_sweep_engagement_range_distance + ), + ) + class Builder(FormationAttackBuilder[SeadSweepFlightPlan, FormationAttackLayout]): def layout(self) -> FormationAttackLayout: