Added support for FollowBigFormations task

pydcs · Dec 21, 2023 · 6c2794e · 6c2794e
1 parent 2003c99
commit 6c2794e
Show file tree

Hide file tree

Showing 4 changed files with 174 additions and 2 deletions.
diff --git a/dcs/mapping.py b/dcs/mapping.py
@@ -98,6 +98,18 @@ class Vector2:
     y: float
 
 
+@dataclass
+class WWIIFollowBigFormationPosition:
+    x: int
+    y: int
+    z: int
+
+    def __eq__(self, other):
+        if isinstance(other, dict):
+            return self.__dict__ == other
+        return self.__dict__ == other.__dict__
+
+
 class Point(Vector2):
     def __init__(self, x: float, y: float, terrain: Terrain) -> None:
         super().__init__(x, y)

diff --git a/dcs/task.py b/dcs/task.py
@@ -9,7 +9,7 @@
 """
 from typing import List, Dict, Optional, Type, Any, Union
 from enum import Enum, IntEnum
-from dcs.mapping import Vector2
+from dcs.mapping import Vector2, WWIIFollowBigFormationPosition
 from dcs.unit import Unit
 
 
@@ -1027,6 +1027,47 @@ def __init__(self, from_index=None, to_index=None):
             self.params["nWaypointIndx"] = to_index
 
 
+class WWIIFollowBigFormation(Task):
+    Id = "FollowBigFormation"
+
+    class FormationType(IntEnum):
+        COMBAT_BOX_FOR_OPEN_FORMATION = 18
+        JAVELIN_DOWN = 16
+        COMBAT_BOX = 15
+
+    def __init__(self,
+                 group_id: Optional[int] = None,
+                 position: WWIIFollowBigFormationPosition = WWIIFollowBigFormationPosition(0, 0, 0),
+                 last_wpt_index_flag: bool = False,
+                 last_wpt_index_flag_changed_manually: bool = False,
+                 formation_type: FormationType = FormationType.COMBAT_BOX_FOR_OPEN_FORMATION,
+                 pos_in_box: int = 0,
+                 pos_in_group: int = 0,
+                 pos_in_wing: int = 0,
+                 last_wpt_index: Optional[int] = None) -> None:
+        super().__init__(self.Id)
+
+        self.params = {
+            "pos": position,
+            "lastWptIndexFlag": last_wpt_index_flag,
+            "lastWptIndexFlagChangedManually": last_wpt_index_flag_changed_manually,
+            "lastWptIndex": last_wpt_index,
+            "formationType": formation_type,
+            "posInBox": pos_in_box,
+            "posInGroup": pos_in_group,
+            "posInWing": pos_in_wing
+        }
+        if group_id:
+            self.params["groupId"] = group_id
+        if last_wpt_index:
+            self.params["lastWptIndex"] = last_wpt_index
+
+    def __eq__(self, other) -> bool:
+        if isinstance(other, self.__class__):
+            return self.__dict__ == other.__dict__
+        return False
+
+
 tasks_map: Dict[str, Type[Task]] = {
     ControlledTask.Id: ControlledTask,
     EscortTaskAction.Id: EscortTaskAction,
@@ -1058,7 +1099,8 @@ def __init__(self, from_index=None, to_index=None):
     FireAtPoint.Id: FireAtPoint,
     AttackUnit.Id: AttackUnit,
     AttackMapObject.Id: AttackMapObject,
-    EngageTargets.Id: EngageTargets
+    EngageTargets.Id: EngageTargets,
+    WWIIFollowBigFormation.Id: WWIIFollowBigFormation
 }
 
 

diff --git a/tests/missions/big-formation.miz b/tests/missions/big-formation.miz
diff --git a/tests/test_mission.py b/tests/test_mission.py
@@ -11,6 +11,8 @@
 from dcs.status_message import MessageSeverity, MessageType
 from dcs.flyingunit import FlyingUnit
 from dcs.unit import Ship
+from dcs.task import WWIIFollowBigFormation
+from dcs.mapping import WWIIFollowBigFormationPosition
 
 
 class BasicTests(unittest.TestCase):
@@ -884,3 +886,119 @@ def test_smoke_required_modules(self) -> None:
         m2 = dcs.mission.Mission()
         m2.load_file(saved_miz)
         self.assertEqual(m.required_modules, m2.required_modules)
+
+    def test_big_formation_action_leader(self) -> None:
+        m_name = "tests/missions/big-formation.miz"
+        m = dcs.mission.Mission()
+        m.load_file(m_name)
+
+        assert isinstance(m.coalition['blue'].country("Combined Joint Task Forces Blue")
+                          .plane_group[0].points[0].tasks[5], WWIIFollowBigFormation)
+
+        task = m.coalition['blue'].country("Combined Joint Task Forces Blue").plane_group[0].points[0].tasks[5]
+
+        self.assertNotIn("groupId", task.params)
+        self.assertNotIn("lastWptIndex", task.params)
+        self.assertEqual(task.params["formationType"], WWIIFollowBigFormation.FormationType.COMBAT_BOX_FOR_OPEN_FORMATION)
+        self.assertEqual(task.params["pos"], WWIIFollowBigFormationPosition(0, 0, 0))
+        self.assertTrue(task.params["lastWptIndexFlagChangedManually"])
+        self.assertEqual(len(task.params), 7)
+
+        m2_name = "missions/saved_big-formation.miz"
+        m.save(m2_name)
+
+        m2 = dcs.mission.Mission()
+        m2.load_file(m2_name)
+
+        assert isinstance(m2.coalition['blue'].country("Combined Joint Task Forces Blue")
+                          .plane_group[0].points[0].tasks[5], WWIIFollowBigFormation)
+        m2_task = m.coalition['blue'].country("Combined Joint Task Forces Blue").plane_group[0].points[0].tasks[5]
+
+        self.assertEqual(task, m2_task)
+
+    def test_big_formation_action_left(self) -> None:
+        m_name = "tests/missions/big-formation.miz"
+        m = dcs.mission.Mission()
+        m.load_file(m_name)
+
+        assert isinstance(m.coalition['blue'].country("Combined Joint Task Forces Blue")
+                          .plane_group[1].points[0].tasks[5], WWIIFollowBigFormation)
+        task = m.coalition['blue'].country("Combined Joint Task Forces Blue").plane_group[1].points[0].tasks[5]
+
+        self.assertEqual(task.params["formationType"], WWIIFollowBigFormation.FormationType.JAVELIN_DOWN)
+        self.assertEqual(task.params["pos"], WWIIFollowBigFormationPosition(-480, -70, -240))
+        self.assertEqual(task.params["groupId"], 2)
+        self.assertEqual(task.params["posInGroup"], 2)
+        self.assertEqual(task.params["lastWptIndex"], 3)
+        self.assertTrue(task.params["lastWptIndexFlag"])
+        self.assertEqual(len(task.params), 9)
+
+        m2_name = "missions/saved_big-formation.miz"
+        m.save(m2_name)
+
+        m2 = dcs.mission.Mission()
+        m2.load_file(m2_name)
+
+        assert isinstance(m2.coalition['blue'].country("Combined Joint Task Forces Blue")
+                          .plane_group[1].points[0].tasks[5], WWIIFollowBigFormation)
+        m2_task = m.coalition['blue'].country("Combined Joint Task Forces Blue").plane_group[1].points[0].tasks[5]
+
+        self.assertEqual(task, m2_task)
+
+    def test_big_formation_action_back(self) -> None:
+        m_name = "tests/missions/big-formation.miz"
+        m = dcs.mission.Mission()
+        m.load_file(m_name)
+
+        assert isinstance(m.coalition['blue'].country("Combined Joint Task Forces Blue")
+                          .plane_group[2].points[0].tasks[5], WWIIFollowBigFormation)
+        task = m.coalition['blue'].country("Combined Joint Task Forces Blue").plane_group[2].points[0].tasks[5]
+
+        self.assertEqual(task.params["formationType"], WWIIFollowBigFormation.FormationType.COMBAT_BOX)
+        self.assertEqual(task.params["pos"], WWIIFollowBigFormationPosition(-320, -50, -0))
+        self.assertEqual(task.params["groupId"], 2)
+        self.assertEqual(task.params["posInBox"], 3)
+        self.assertEqual(task.params["lastWptIndex"], 3)
+        self.assertFalse(task.params["lastWptIndexFlag"])
+        self.assertEqual(len(task.params), 9)
+
+        m2_name = "missions/saved_big-formation.miz"
+        m.save(m2_name)
+
+        m2 = dcs.mission.Mission()
+        m2.load_file(m2_name)
+
+        assert isinstance(m2.coalition['blue'].country("Combined Joint Task Forces Blue")
+                          .plane_group[2].points[0].tasks[5], WWIIFollowBigFormation)
+        m2_task = m.coalition['blue'].country("Combined Joint Task Forces Blue").plane_group[2].points[0].tasks[5]
+
+        self.assertEqual(task, m2_task)
+
+    def test_big_formation_action_right(self) -> None:
+        m_name = "tests/missions/big-formation.miz"
+        m = dcs.mission.Mission()
+        m.load_file(m_name)
+
+        assert isinstance(m.coalition['blue'].country("Combined Joint Task Forces Blue")
+                          .plane_group[3].points[0].tasks[5], WWIIFollowBigFormation)
+        task = m.coalition['blue'].country("Combined Joint Task Forces Blue").plane_group[3].points[0].tasks[5]
+
+        self.assertEqual(task.params["formationType"], WWIIFollowBigFormation.FormationType.COMBAT_BOX_FOR_OPEN_FORMATION)
+        self.assertEqual(task.params["pos"], WWIIFollowBigFormationPosition(-160, 50, 240))
+        self.assertEqual(task.params["groupId"], 2)
+        self.assertEqual(task.params["posInBox"], 1)
+        self.assertEqual(task.params["lastWptIndex"], 3)
+        self.assertTrue(task.params["lastWptIndexFlag"])
+        self.assertEqual(len(task.params), 9)
+
+        m2_name = "missions/saved_big-formation.miz"
+        m.save(m2_name)
+
+        m2 = dcs.mission.Mission()
+        m2.load_file(m2_name)
+
+        assert isinstance(m2.coalition['blue'].country("Combined Joint Task Forces Blue")
+                          .plane_group[3].points[0].tasks[5], WWIIFollowBigFormation)
+        m2_task = m.coalition['blue'].country("Combined Joint Task Forces Blue").plane_group[3].points[0].tasks[5]
+
+        self.assertEqual(task, m2_task)