mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
31c80dfd02
refactor to enum typing and many other fixes fix tests attempt to fix some typescript more typescript fixes more typescript test fixes revert all API changes update to pydcs mypy fixes Use properties to check if player is blue/red/neutral update requirements.txt black -_- bump pydcs and fix mypy add opponent property bump pydcs
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Iterator
|
|
from typing import TYPE_CHECKING
|
|
|
|
from dcs import Point
|
|
from shapely.ops import unary_union
|
|
|
|
from game.utils import dcs_to_shapely_point
|
|
|
|
if TYPE_CHECKING:
|
|
from game.theater import ConflictTheater, TheaterGroundObject, Player
|
|
from game.threatzones import ThreatPoly
|
|
|
|
|
|
class SamEngagementZones:
|
|
def __init__(
|
|
self,
|
|
threat_zones: ThreatPoly,
|
|
individual_zones: list[tuple[TheaterGroundObject, ThreatPoly]],
|
|
) -> None:
|
|
self.threat_zones = threat_zones
|
|
self.individual_zones = individual_zones
|
|
|
|
def covers(self, position: Point) -> bool:
|
|
return self.threat_zones.intersects(dcs_to_shapely_point(position))
|
|
|
|
def iter_threatening_sams(self, position: Point) -> Iterator[TheaterGroundObject]:
|
|
for tgo, zone in self.individual_zones:
|
|
if zone.intersects(dcs_to_shapely_point(position)):
|
|
yield tgo
|
|
|
|
@classmethod
|
|
def from_theater(
|
|
cls, theater: ConflictTheater, player: Player
|
|
) -> SamEngagementZones:
|
|
commit_regions = []
|
|
individual_zones = []
|
|
for cp in theater.control_points_for(player):
|
|
for tgo in cp.connected_objectives:
|
|
if (region := tgo.threat_poly()) is not None:
|
|
commit_regions.append(region)
|
|
individual_zones.append((tgo, region))
|
|
return SamEngagementZones(unary_union(commit_regions), individual_zones)
|