mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
af596c58c3
Add central registries for allocating TACAN/radio channels to the Operation. These ensure that each channel is allocated uniquely, and removes the caller's need to think about which frequency to use. The registry allocates frequencies based on the radio it is given, which ensures that the allocated frequency will be compatible with the radio that needs it. A mapping from aircraft to the radio used by that aircraft for intra-flight comms (i.e. the F-16 uses the AN/ARC-222) exists for creating infra-flight channels appropriate for the aircraft. Inter-flight channels are allocated by a generic UHF radio. I've moved the inter-flight radio channels from the VHF to UHF range, since that's the most easily allocated band, and inter-flight will be in the highest demand. Intra-flight radios are now generally not shared. For aircraft where the radio type is not known we will still fall back to the shared channel, but that will stop being the case as we gain more data. Tankers have been moved to the Y TACAN band. Not completely needed, but seems typical for most missions and deconflicts the tankers from any unknown airfields (which always use the X band in DCS).
84 lines
2.4 KiB
Python
84 lines
2.4 KiB
Python
"""TACAN channel handling."""
|
|
from dataclasses import dataclass
|
|
from enum import Enum
|
|
from typing import Dict, Iterator, Set
|
|
|
|
|
|
class TacanBand(Enum):
|
|
X = "X"
|
|
Y = "Y"
|
|
|
|
def range(self) -> Iterator["TacanChannel"]:
|
|
"""Returns an iterator over the channels in this band."""
|
|
return (TacanChannel(x, self) for x in range(1, 100))
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class TacanChannel:
|
|
number: int
|
|
band: TacanBand
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.number}{self.band.value}"
|
|
|
|
|
|
class OutOfTacanChannelsError(RuntimeError):
|
|
"""Raised when all channels in this band have been allocated."""
|
|
|
|
def __init__(self, band: TacanBand) -> None:
|
|
super().__init__(f"No available channels in TACAN {band.value} band")
|
|
|
|
|
|
class TacanChannelInUseError(RuntimeError):
|
|
"""Raised when attempting to reserve an in-use channel."""
|
|
|
|
def __init__(self, channel: TacanChannel) -> None:
|
|
super().__init__(f"{channel} is already in use")
|
|
|
|
|
|
class TacanRegistry:
|
|
"""Manages allocation of TACAN channels."""
|
|
|
|
def __init__(self) -> None:
|
|
self.allocated_channels: Set[TacanChannel] = set()
|
|
self.band_allocators: Dict[TacanBand, Iterator[TacanChannel]] = {}
|
|
|
|
for band in TacanBand:
|
|
self.band_allocators[band] = band.range()
|
|
|
|
def alloc_for_band(self, band: TacanBand) -> TacanChannel:
|
|
"""Allocates a TACAN channel in the given band.
|
|
|
|
Args:
|
|
band: The TACAN band to allocate a channel for.
|
|
|
|
Returns:
|
|
A TACAN channel in the given band.
|
|
|
|
Raises:
|
|
OutOfChannelsError: All channels compatible with the given radio are
|
|
already allocated.
|
|
"""
|
|
allocator = self.band_allocators[band]
|
|
try:
|
|
while (channel := next(allocator)) in self.allocated_channels:
|
|
pass
|
|
return channel
|
|
except StopIteration:
|
|
raise OutOfTacanChannelsError(band)
|
|
|
|
def reserve(self, channel: TacanChannel) -> None:
|
|
"""Reserves the given channel.
|
|
|
|
Reserving a channel ensures that it will not be allocated in the future.
|
|
|
|
Args:
|
|
channel: The channel to reserve.
|
|
|
|
Raises:
|
|
ChannelInUseError: The given frequency is already in use.
|
|
"""
|
|
if channel in self.allocated_channels:
|
|
raise TacanChannelInUseError(channel)
|
|
self.allocated_channels.add(channel)
|