From 169ef26329fa52f8f382f0493e49472f15f69dd8 Mon Sep 17 00:00:00 2001
From: M Chimiste <37678080+M-Chimiste@users.noreply.github.com>
Date: Sat, 5 Oct 2024 19:12:02 -0400
Subject: [PATCH] Set Pretense Player Flight Waypoints to CPs (#385)

---
 .../pretenseflightgroupconfigurator.py        | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/game/pretense/pretenseflightgroupconfigurator.py b/game/pretense/pretenseflightgroupconfigurator.py
index 10bc45a4..9726c707 100644
--- a/game/pretense/pretenseflightgroupconfigurator.py
+++ b/game/pretense/pretenseflightgroupconfigurator.py
@@ -9,6 +9,8 @@ from dcs.unitgroup import FlyingGroup
 
 from game.ato import Flight, FlightType
 from game.ato.flightmember import FlightMember
+from game.ato.flightwaypoint import FlightWaypoint
+from game.ato.flightwaypointtype import FlightWaypointType
 from game.data.weapons import Pylon
 from game.lasercodes.lasercoderegistry import LaserCodeRegistry
 from game.missiongenerator.aircraft.aircraftbehavior import AircraftBehavior
@@ -19,6 +21,9 @@ from game.missiongenerator.aircraft.flightgroupconfigurator import (
     FlightGroupConfigurator,
 )
 from game.missiongenerator.aircraft.waypoints import WaypointGenerator
+from game.missiongenerator.aircraft.waypoints.pydcswaypointbuilder import (
+    PydcsWaypointBuilder,
+)
 from game.missiongenerator.missiondata import MissionData
 from game.radio.radios import RadioRegistry
 from game.radio.tacan import (
@@ -95,6 +100,26 @@ class PretenseFlightGroupConfigurator(FlightGroupConfigurator):
             self.mission_data,
         ).create_waypoints()
 
+        if self.flight.client_count >= 1:
+            waypoints = waypoints[:1]
+            self.group.points = self.group.points[:1]
+
+            for cp in self.game.theater.controlpoints:
+                PydcsWaypointBuilder(
+                    FlightWaypoint(
+                        name=cp.full_name,
+                        waypoint_type=FlightWaypointType.NAV,
+                        position=cp.position,
+                        alt_type="RADIO",
+                        control_point=cp,
+                    ),
+                    self.group,
+                    self.flight,
+                    self.mission,
+                    self.time,
+                    self.mission_data,
+                ).build()
+
         divert_position: Point | None = None
         if self.flight.divert is not None:
             divert_position = self.flight.divert.position