From b5ff32c5b6252146ac4fbb6d4579e8187ce6026f Mon Sep 17 00:00:00 2001
From: Dan Albert <dan@gingerhq.net>
Date: Sun, 6 Dec 2020 15:01:21 -0800
Subject: [PATCH] Show patrol end time on the kneeboard.

Fixed https://github.com/Khopa/dcs_liberation/issues/421
---
 gen/aircraft.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/gen/aircraft.py b/gen/aircraft.py
index 5ac8ba25..d3c25323 100644
--- a/gen/aircraft.py
+++ b/gen/aircraft.py
@@ -1404,6 +1404,7 @@ class PydcsWaypointBuilder:
             FlightWaypointType.JOIN: JoinPointBuilder,
             FlightWaypointType.LANDING_POINT: LandingPointBuilder,
             FlightWaypointType.LOITER: HoldPointBuilder,
+            FlightWaypointType.PATROL: RaceTrackEndBuilder,
             FlightWaypointType.PATROL_TRACK: RaceTrackBuilder,
         }
         builder = builders.get(waypoint.waypoint_type, DefaultWaypointBuilder)
@@ -1761,3 +1762,18 @@ class RaceTrackBuilder(PydcsWaypointBuilder):
                                                 targets=[Targets.All.Air]))
 
         return waypoint
+
+
+class RaceTrackEndBuilder(PydcsWaypointBuilder):
+    def build(self) -> MovingPoint:
+        waypoint = super().build()
+
+        if not isinstance(self.flight.flight_plan, PatrollingFlightPlan):
+            flight_plan_type = self.flight.flight_plan.__class__.__name__
+            logging.error(
+                f"Cannot create race track for {self.flight} because "
+                f"{flight_plan_type} does not define a patrol.")
+            return waypoint
+
+        self.waypoint.departure_time = self.flight.flight_plan.patrol_end_time
+        return waypoint