mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
from abc import ABC, abstractmethod
|
|
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 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 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 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
|