mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
31289adb50
The release behavior isn't used yet, but I'm working on pre-allocating laser codes for front lines and flights to make it easier for players to pick the laser codes for their weapons.
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
import logging
|
|
from collections import deque
|
|
|
|
from .ilasercoderegistry import ILaserCodeRegistry
|
|
from .lasercode import LaserCode
|
|
|
|
|
|
class LaserCodeRegistry(ILaserCodeRegistry):
|
|
def __init__(self) -> None:
|
|
self.allocated_codes: set[int] = set()
|
|
self.available_codes = LaserCodeRegistry._all_valid_laser_codes()
|
|
self.fc3_code = LaserCode(1113, self)
|
|
|
|
def alloc_laser_code(self) -> LaserCode:
|
|
try:
|
|
code = self.available_codes.popleft()
|
|
self.allocated_codes.add(code)
|
|
return LaserCode(code, self)
|
|
except IndexError:
|
|
raise RuntimeError("All laser codes have been allocated")
|
|
|
|
def release_code(self, code: LaserCode) -> None:
|
|
if code.code in self.allocated_codes:
|
|
self.allocated_codes.remove(code.code)
|
|
self.available_codes.appendleft(code.code)
|
|
else:
|
|
logging.error(
|
|
"attempted to release laser code %d which was not allocated", code.code
|
|
)
|
|
|
|
@staticmethod
|
|
def _all_valid_laser_codes() -> deque[int]:
|
|
# Valid laser codes are as follows
|
|
# First digit is always 1
|
|
# Second digit is 5-7
|
|
# Third and fourth digits are 1 - 8
|
|
# We iterate backward (reversed()) so that 1687 follows 1688
|
|
q = deque(int(oct(code)[2:]) + 11 for code in reversed(range(0o1500, 0o2000)))
|
|
|
|
# We start with the default of 1688 and wrap around when we reach the end
|
|
q.rotate(-q.index(1688))
|
|
return q
|