Helicopter flights can be planned from FOBs

game/helipad.py

@@ -0,0 +1,13 @@
+from typing import Optional
+
+from dcs.unitgroup import StaticGroup
+
+from game.point_with_heading import PointWithHeading
+
+
+class Helipad(PointWithHeading):
+    def __init__(self):
+        super(Helipad, self).__init__()
+        self.heading = 0
+        self.occupied = False
+        self.static_unit: Optional[StaticGroup] = None

game/theater/conflicttheater.py

@@ -41,6 +41,7 @@ from dcs.unitgroup import (
 )
 from dcs.vehicles import AirDefence, Armor, MissilesSS, Unarmed
 
+from ..helipad import Helipad
 from ..scenery_group import SceneryGroup
 from pyproj import CRS, Transformer
 from shapely import geometry, ops
@@ -549,7 +550,7 @@ class MizCampaignLoader:
         for group in self.helipads:
             closest, distance = self.objective_info(group)
             closest.helipads.append(
-                PointWithHeading.from_point(group.position, group.units[0].heading)
+                Helipad.from_point(group.position, group.units[0].heading)
             )
 
         for group in self.factories:

game/theater/controlpoint.py

@@ -10,6 +10,7 @@ from enum import Enum
 from functools import total_ordering
 from typing import Any, Dict, Iterator, List, Optional, Set, TYPE_CHECKING, Type, Union
 
+from dcs import helicopters
 from dcs.mapping import Point
 from dcs.ships import (
     CVN_74_John_C__Stennis,
@@ -39,6 +40,7 @@ from .theatergroundobject import (
     VehicleGroupGroundObject,
 )
 from ..db import PRICES
+from ..helipad import Helipad
 from ..utils import nautical_miles
 from ..weather import Conditions
 
@@ -296,7 +298,7 @@ class ControlPoint(MissionTarget, ABC):
         self.connected_objectives: List[TheaterGroundObject] = []
         self.base_defenses: List[BaseDefenseGroundObject] = []
         self.preset_locations = PresetLocations()
-        self.helipads: List[PointWithHeading] = []
+        self.helipads: List[Helipad] = []
 
         # TODO: Should be Airbase specific.
         self.size = size
@@ -378,6 +380,29 @@ class ControlPoint(MissionTarget, ABC):
                 return True
         return False
 
+    @property
+    def has_helipads(self) -> bool:
+        """
+        Returns true if cp has helipads
+        """
+        return len(self.helipads) > 0
+
+    @property
+    def has_free_helipad(self) -> bool:
+        """
+        Returns true if cp has a free helipad
+        """
+        return False in [h.occupied for h in self.helipads]
+
+    def get_free_helipad(self) -> Optional[Helipad]:
+        """
+        Returns the first free additional helipad
+        """
+        for h in self.helipads:
+            if not h.occupied:
+                return h
+        return None
+
     def can_recruit_ground_units(self, game: Game) -> bool:
         """Returns True if this control point is capable of recruiting ground units."""
         if not self.can_deploy_ground_units:
@@ -1084,10 +1109,13 @@ class Fob(ControlPoint):
 
     @property
     def total_aircraft_parking(self) -> int:
-        return 0
+        return len(self.helipads)
 
     def can_operate(self, aircraft: FlyingType) -> bool:
-        return False
+        if aircraft in helicopters.helicopter_map.values():
+            return True
+        else:
+            return False
 
     @property
     def heading(self) -> int:

game/theater/missiontarget.py

@@ -46,4 +46,4 @@ class MissionTarget:
 
     @property
     def strike_targets(self) -> List[Union[MissionTarget, Unit]]:
-        raise NotImplementedError
+        return []