mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
e8df6a3d54
The weapon laser codes can be set more easily from the weapon laser code combo box. Setting the properties explicitly here will just cause conflicts and annoying UI bugs. Hide those properties from the UI.
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from abc import ABC, abstractmethod
|
|
from collections.abc import Iterator
|
|
from typing import Any
|
|
|
|
|
|
class LaserCodeConfig(ABC):
|
|
def __init__(self, pylon: int) -> None:
|
|
self.pylon = pylon
|
|
|
|
@staticmethod
|
|
def from_yaml(data: dict[str, Any]) -> LaserCodeConfig:
|
|
pylon = data["pylon"]
|
|
if (property_def := data.get("property")) is not None:
|
|
return SinglePropertyLaserCodeConfig(
|
|
pylon, property_def["id"], int(property_def["digits"])
|
|
)
|
|
return MultiplePropertyLaserCodeConfig(
|
|
pylon, [(d["id"], d["digit"]) for d in data["properties"]]
|
|
)
|
|
|
|
@abstractmethod
|
|
def iter_prop_ids(self) -> Iterator[str]:
|
|
...
|
|
|
|
@abstractmethod
|
|
def property_dict_for_code(self, code: int) -> dict[str, int]:
|
|
...
|
|
|
|
|
|
class SinglePropertyLaserCodeConfig(LaserCodeConfig):
|
|
def __init__(self, pylon: int, property_id: str, digits: int) -> None:
|
|
super().__init__(pylon)
|
|
self.property_id = property_id
|
|
self.digits = digits
|
|
|
|
def iter_prop_ids(self) -> Iterator[str]:
|
|
yield self.property_id
|
|
|
|
def property_dict_for_code(self, code: int) -> dict[str, int]:
|
|
return {self.property_id: code % 10**self.digits}
|
|
|
|
|
|
class MultiplePropertyLaserCodeConfig(LaserCodeConfig):
|
|
def __init__(
|
|
self, pylon: int, property_digit_mappings: list[tuple[str, int]]
|
|
) -> None:
|
|
super().__init__(pylon)
|
|
self.property_digit_mappings = property_digit_mappings
|
|
|
|
def iter_prop_ids(self) -> Iterator[str]:
|
|
yield from (i for i, p in self.property_digit_mappings)
|
|
|
|
def property_dict_for_code(self, code: int) -> dict[str, int]:
|
|
d = {}
|
|
for prop_id, idx in self.property_digit_mappings:
|
|
d[prop_id] = code // 10**idx % 10
|
|
return d
|