Add configuration for semaphore signals and implement railway signal control

test_lighter_main.py

@@ -0,0 +1,132 @@
+import sys
+import time
+
+try:
+	import ujson as json  # MicroPython
+except ImportError:
+	import json
+
+from railway_signal import RailwaySignal
+
+
+def _sleep_ms(ms: int):
+	sleep_ms = getattr(time, "sleep_ms", None)
+	if sleep_ms is not None:
+		sleep_ms(ms)
+	else:
+		time.sleep(ms / 1000)
+
+
+def load_config():
+	# Поддержка обоих имён (в проекте встречаются оба варианта)
+	for name in ("semaphore-config.json", "semaphor-config.json", "semaphor-config.json"):
+		try:
+			with open(name, "r") as f:
+				return json.load(f)
+		except OSError:
+			continue
+	raise OSError("Config not found: semaphore-config.json")
+
+
+def init_semaphores(cfg: dict):
+	semaphores = {}
+	lamps_by_id = {}
+
+	for sem in cfg.get("semaphores", []):
+		sem_id = sem.get("id")
+		pin = sem.get("pin")
+		bpp = sem.get("bpp", 4)
+		lamps = sem.get("lamps", [])
+
+		if sem_id is None or pin is None:
+			continue
+
+		signal = RailwaySignal(pin=int(pin), num_leds=len(lamps), bpp=int(bpp), timing=1)
+		semaphores[int(sem_id)] = signal
+
+		# Маппинг: порядок ламп в конфиге -> индекс пикселя
+		for pixel_index, lamp_cfg in enumerate(lamps):
+			lamp_id = lamp_cfg.get("id")
+			color = lamp_cfg.get("color", "OFF")
+			if lamp_id is None:
+				continue
+
+			lamp = signal.add_lamp(int(lamp_id), pixel_index, str(color))
+			lamps_by_id[int(lamp_id)] = lamp
+
+	return semaphores, lamps_by_id
+
+
+def print_help():
+	print("Commands:")
+	print("  <lamp_id> off")
+	print("  <lamp_id> on")
+	print("  <lamp_id> blink [period_ms]")
+	print("Examples:")
+	print("  101 on")
+	print("  102 blink 500")
+
+
+def main():
+	cfg = load_config()
+	semaphores, lamps_by_id = init_semaphores(cfg)
+
+	print("READY")
+	print(f"Semaphores: {sorted(semaphores.keys())}")
+	print(f"Lamps: {sorted(lamps_by_id.keys())}")
+	print_help()
+
+	while True:
+		line = sys.stdin.readline()
+		if not line:
+			_sleep_ms(20)
+			continue
+
+		line = line.strip()
+		if not line:
+			continue
+
+		if line.lower() in ("help", "h", "?"):
+			print_help()
+			continue
+
+		parts = line.split()
+		if len(parts) < 2:
+			print("ERR ожидаю: <lamp_id> <state> [period_ms]")
+			continue
+
+		try:
+			lamp_id = int(parts[0])
+		except ValueError:
+			print("ERR lamp_id должен быть числом")
+			continue
+
+		state = parts[1].lower()
+		lamp = lamps_by_id.get(lamp_id)
+		if lamp is None:
+			print(f"ERR lamp_id {lamp_id} не найден")
+			continue
+
+		if state == "off":
+			lamp.off()
+			print(f"OK {lamp_id} off")
+		elif state == "on":
+			lamp.on()
+			print(f"OK {lamp_id} on")
+		elif state == "blink":
+			period_ms = 500
+			if len(parts) >= 3:
+				try:
+					period_ms = int(parts[2])
+				except ValueError:
+					print("ERR period_ms должен быть числом")
+					continue
+			lamp.blink(period_ms)
+			print(f"OK {lamp_id} blink {period_ms}")
+		else:
+			print("ERR state должен быть off/on/blink")
+
+
+if __name__ == "__main__":
+	main()
+