infantry transport missions for helis

game/db.py

@@ -122,8 +122,6 @@ UNIT_BY_TASK = {
     AWACS: [E_3A, A_50, ],
 
     PinpointStrike: [Armor.MBT_T_90, Armor.MBT_T_80U, Armor.MBT_T_55, Armor.MBT_M1A2_Abrams, Armor.MBT_M60A3_Patton, Armor.ATGM_M1134_Stryker, Armor.APC_BTR_80, ],
-    Reconnaissance: [Unarmed.Transport_M818, Unarmed.Transport_Ural_375, Unarmed.Transport_UAZ_469],
-    Nothing: [Infantry.Infantry_M4, Infantry.Soldier_AK, ],
     AirDefence: [
         AirDefence.AAA_Vulcan_M163,
         AirDefence.AAA_Vulcan_M163,
@@ -142,6 +140,10 @@ UNIT_BY_TASK = {
         AirDefence.SAM_SA_8_Osa_9A33,
     ],
 
+    Reconnaissance: [Unarmed.Transport_M818, Unarmed.Transport_Ural_375, Unarmed.Transport_UAZ_469],
+    Nothing: [Infantry.Infantry_M4, Infantry.Soldier_AK, ],
+    Embarking: [UH_1H, Mi_8MT, ],
+
     Carriage: [CVN_74_John_C__Stennis, CV_1143_5_Admiral_Kuznetsov, ],
     CargoTransportation: [Dry_cargo_ship_Ivanov, Bulk_cargo_ship_Yakushev, Tanker_Elnya_160, LHA_1_Tarawa],
 }
@@ -188,6 +190,7 @@ UNIT_BY_COUNTRY = {
         A_50,
 
         Ka_50,
+        UH_1H,
         Mi_8MT,
 
         AirDefence.AAA_ZU_23_on_Ural_375,
@@ -227,6 +230,7 @@ UNIT_BY_COUNTRY = {
 
         Ka_50,
         UH_1H,
+        Mi_8MT,
 
         Armor.MBT_M1A2_Abrams,
         Armor.MBT_M60A3_Patton,
@@ -297,6 +301,8 @@ def unit_type_name(unit_type) -> str:
 def task_name(task) -> str:
     if task == AirDefence:
         return "AirDefence"
+    elif task == Embarking:
+        return "Transportation"
     else:
         return task.name
 

game/event/__init__.py

@@ -4,4 +4,5 @@ from .intercept import *
 from .capture import *
 from .navalintercept import *
 from .antiaastrike import *
-from .groundattack import *
+from .groundattack import *
+from .infantrytransport import *

game/event/groundattack.py

@@ -23,8 +23,8 @@ class GroundAttackEvent(GroundInterceptEvent):
         op = GroundAttackOperation(game=self.game,
                                    attacker_name=self.attacker_name,
                                    defender_name=self.defender_name,
-                                   attacker_clients=clients,
-                                   defender_clients={},
+                                   attacker_clients={},
+                                   defender_clients=clients,
                                    from_cp=self.from_cp,
                                    to_cp=self.to_cp)
         op.setup(target=self.targets,

game/event/infantrytransport.py

@@ -0,0 +1,45 @@
+import math
+import random
+
+from dcs.task import *
+from dcs.vehicles import *
+
+from game import db
+from game.operation.infantrytransport import InfantryTransportOperation
+from theater.conflicttheater import *
+from userdata.debriefing import Debriefing
+
+from .event import Event
+
+
+class InfantryTransportEvent(Event):
+    STRENGTH_INFLUENCE = 0.3
+
+    def __str__(self):
+        return "Frontline transport troops to {}".format(self.to_cp)
+
+    def is_successfull(self, debriefing: Debriefing):
+        return True
+
+    def commit(self, debriefing: Debriefing):
+        super(InfantryTransportEvent, self).commit(debriefing)
+
+        if self.is_successfull(debriefing):
+            self.to_cp.base.affect_strength(-self.STRENGTH_INFLUENCE)
+        else:
+            self.from_cp.base.affect_strength(-self.STRENGTH_INFLUENCE)
+
+    def player_attacking(self, transport: db.HeliDict, clients: db.HeliDict):
+        op = InfantryTransportOperation(
+            game=self.game,
+            attacker_name=self.attacker_name,
+            defender_name=self.defender_name,
+            attacker_clients=clients,
+            defender_clients={},
+            from_cp=self.from_cp,
+            to_cp=self.to_cp
+        )
+
+        op.setup(transport=transport)
+
+        self.operation = op

game/game.py

@@ -35,9 +35,10 @@ EVENT_PROBABILITIES = {
     CaptureEvent: [100, 4],
     InterceptEvent: [35, 5],
     GroundInterceptEvent: [35, 5],
-    GroundAttackEvent: [0, 35],
+    GroundAttackEvent: [0, 5],
     NavalInterceptEvent: [35, 5],
     AntiAAStrikeEvent: [35, 5],
+    InfantryTransportEvent: [100, 0],
 }
 
 PLAYER_BASE_STRENGTH_RECOVERY = 0.2

game/operation/infantrytransport.py

@@ -0,0 +1,52 @@
+from dcs.terrain import Terrain
+
+from game import db
+from gen.armor import *
+from gen.aircraft import *
+from gen.aaa import *
+from gen.shipgen import *
+from gen.settingsgen import *
+from gen.awacsgen import *
+from gen.visualgen import *
+from gen.conflictgen import Conflict
+
+from .operation import Operation
+
+
+class InfantryTransportOperation(Operation):
+    transport = None  # type: db.HeliDict
+
+    def setup(self, transport: db.HeliDict):
+        self.transport = transport
+
+    def prepare(self, terrain: Terrain, is_quick: bool):
+        super(InfantryTransportOperation, self).prepare(terrain, is_quick)
+
+        conflict = Conflict.transport_conflict(
+            attacker=self.mission.country(self.attacker_name),
+            defender=self.mission.country(self.defender_name),
+            from_cp=self.from_cp,
+            to_cp=self.to_cp,
+            theater=self.game.theater
+        )
+
+        self.initialize(mission=self.mission,
+                        conflict=conflict)
+
+    def generate(self):
+        self.airgen.generate_passenger_transport(
+            helis=self.transport,
+            clients=self.attacker_clients,
+            at=self.attackers_starting_position
+        )
+
+        self.armorgen.generate_passengers(count=8)
+
+        self.visualgen.generate_transportation_marker(self.conflict.ground_attackers_location)
+        self.visualgen.generate_transportation_destination(self.conflict.position)
+
+        # this will disable vehicle activation triggers,
+        # which aren't needed on this type of missions
+        self.is_quick = True
+        super(InfantryTransportOperation, self).generate()
+        self.is_quick = False