From 465399f8038ed24560c016fef29952b89efbb274 Mon Sep 17 00:00:00 2001
From: Dan Albert <dan@gingerhq.net>
Date: Sun, 1 Nov 2020 17:12:42 -0800
Subject: [PATCH] Add radio configuration for the UH-1H and Ka-50.

---
 gen/aircraft.py | 30 ++++++++++++++++++++++++++++++
 gen/radios.py   | 10 ++++++++++
 2 files changed, 40 insertions(+)

diff --git a/gen/aircraft.py b/gen/aircraft.py
index 102d9a49..25c4f65e 100644
--- a/gen/aircraft.py
+++ b/gen/aircraft.py
@@ -165,6 +165,14 @@ class FarmerChannelNamer(ChannelNamer):
         return f"Ch {channel_id}"
 
 
+class HueyChannelNamer(ChannelNamer):
+    """Channel namer for the UH-1H."""
+
+    @staticmethod
+    def channel_name(radio_id: int, channel_id: int) -> str:
+        return f"COM3 Ch {channel_id}"
+
+
 class MirageChannelNamer(ChannelNamer):
     """Channel namer for the M-2000."""
 
@@ -548,6 +556,15 @@ AIRCRAFT_DATA: Dict[str, AircraftData] = {
         channel_namer=ViperChannelNamer
     ),
 
+    "Ka-50": AircraftData(
+        inter_flight_radio=get_radio("R-800L1"),
+        intra_flight_radio=get_radio("R-800L1"),
+        # The R-800L1 doesn't have preset channels, and the other radio is for
+        # communications with FAC and ground units, which don't currently have
+        # radios assigned, so no channels to configure.
+        channel_allocator=NoOpChannelAllocator(),
+    ),
+
     "M-2000C": AircraftData(
         inter_flight_radio=get_radio("TRT ERA 7000 V/UHF"),
         intra_flight_radio=get_radio("TRT ERA 7200 UHF"),
@@ -580,6 +597,19 @@ AIRCRAFT_DATA: Dict[str, AircraftData] = {
         ),
         channel_namer=SCR522ChannelNamer
     ),
+
+    "UH-1H": AircraftData(
+        inter_flight_radio=get_radio("AN/ARC-51BX"),
+        # Ideally this would use the AN/ARC-131 because that radio is supposed
+        # to be used for flight comms, but DCS won't allow it as the flight's
+        # frequency, nor will it allow the AN/ARC-134.
+        intra_flight_radio=get_radio("AN/ARC-51BX"),
+        channel_allocator=CommonRadioChannelAllocator(
+            inter_flight_radio_index=1,
+            intra_flight_radio_index=1
+        ),
+        channel_namer=HueyChannelNamer
+    )
 }
 AIRCRAFT_DATA["A-10C_2"] = AIRCRAFT_DATA["A-10C"]
 AIRCRAFT_DATA["P-51D-30-NA"] = AIRCRAFT_DATA["P-51D"]
diff --git a/gen/radios.py b/gen/radios.py
index 5a4976c8..b80683a8 100644
--- a/gen/radios.py
+++ b/gen/radios.py
@@ -132,6 +132,16 @@ RADIOS: List[Radio] = [
 
     # MiG-19P
     Radio("RSIU-4V", MHz(100), MHz(150), step=MHz(1)),
+
+    # Ka-50
+    # Note: Also capable of 100MHz-150MHz, but we can't model gaps.
+    Radio("R-800L1", MHz(220), MHz(400), step=kHz(25)),
+    Radio("R-828", MHz(20), MHz(60), step=kHz(25)),
+
+    # UH-1H
+    Radio("AN/ARC-51BX", MHz(225), MHz(400), step=kHz(50)),
+    Radio("AN/ARC-131", MHz(30), MHz(76), step=kHz(50)),
+    Radio("AN/ARC-134", MHz(116), MHz(150), step=kHz(25)),
 ]