Add configuration for semaphore signals and implement railway signal control

main.py

@@ -2,35 +2,40 @@ import sys
 from switch import Switch
 import select
 from machine import Pin
-from time import sleep_ms
+from time import sleep_ms, ticks_diff, ticks_ms
 from ir_pair import IRRxTxPollPair
+from double_ir import DoubleIr
 
 
-SEMINSU = dict()
+SEMINSU_V1 = dict()
+SEMINSU_V2 = dict()
 SWITCHES = dict()
 LED = Pin("LED", Pin.OUT)  # "LED" — специальное имя для встроенного индикатора
 
 
-
 def load_switches():
     with open("config.json", "r") as file:
         import json
+
         config = json.load(file)
         for sw_cfg in config["switches"]:
             sw = Switch(
                 id=sw_cfg["id"],
                 pin=sw_cfg["pin"],
                 angle_minus=sw_cfg["angle_minus"],
-                angle_plus=sw_cfg["angle_plus"]
+                angle_plus=sw_cfg["angle_plus"],
             )
             SWITCHES[sw.id] = sw
 
 
-def load_seminsus():
+def load_seminsus_v1():
     with open("config.json", "r") as file:
         import json
+
         config = json.load(file)
-        for sem_cfg in config["seminsus"]:
+        if "seminsus_v1" not in config:
+            return
+        for sem_cfg in config["seminsus_v1"]:
             seminsu = IRRxTxPollPair(
                 rx_pin=sem_cfg["pin_rx"],
                 tx_pin=sem_cfg["pin_tx"],
@@ -44,8 +49,20 @@ def load_seminsus():
                 count_rising=False,
                 count_falling=True,
             )
-            SEMINSU[sem_cfg["id"]] = seminsu
-        
+            SEMINSU_V1[sem_cfg["id"]] = seminsu
+
+
+def load_seminsus_v2():
+    with open("config.json", "r") as file:
+        import json
+
+        config = json.load(file)
+        if "irs" not in config:
+            return
+        for sem_cfg in config["irs"]:
+            pin = Pin(sem_cfg["pin"], Pin.IN)
+            seminsu = DoubleIr(sem_cfg["id"], pin)
+            SEMINSU_V2[sem_cfg["id"]] = seminsu
 
 
 def resolve_command(command: str):
@@ -75,7 +92,9 @@ def resolve_command(command: str):
         evts = []
         for id, sw in SWITCHES.items():
             evts.append(f"EVENT SWITCH {id} {sw.pos}")
-        for id, seminsu in SEMINSU.items():
+        for id, seminsu in SEMINSU_V1.items():
+            evts.append(f"EVENT IK_MODULE {id} {seminsu.last_state}")
+        for id, seminsu in SEMINSU_V2.items():
             evts.append(f"EVENT IK_MODULE {id} {seminsu.last_state}")
         return "\n".join(evts)
 
@@ -95,31 +114,65 @@ def _next_seminsu_id():
 
 
 def poll_seminsus_step():
-    """Неблокирующий шаг опроса seminsu.
+    """Неблокирующий шаг опроса IK_MODULE (seminsus_v1 + seminsu_v2).
 
-    В каждый момент времени опрашивается только ОДНА пара.
+    Идём round-robin по объединению ID из SEMINSU_V1 и SEMINSU_V2.
+    ID не пересекаются, поэтому тип выбирается по наличию в словаре.
+
+    Для v1 сохраняется текущая неблокирующая логика: start_poll()/update()
+    и печать события только при изменении состояния.
+
+    Для v2 вызывается check() (печать события внутри check() при изменении).
     """
     global _active_seminsu_id
 
-    if not SEMINSU:
+    if not SEMINSU_V1 and not SEMINSU_V2:
         return
 
-    if _active_seminsu_id is None:
-        _active_seminsu_id = _next_seminsu_id()
-        if _active_seminsu_id is None:
+    # Если есть активный v1-цикл — продолжаем его до завершения.
+    if _active_seminsu_id is not None:
+        seminsu = SEMINSU_V1.get(_active_seminsu_id)
+        if seminsu is None:
+            _active_seminsu_id = None
             return
-        SEMINSU[_active_seminsu_id].start_poll()
 
-    seminsu = SEMINSU[_active_seminsu_id]
+        done = seminsu.update()
+        if not done:
+            return
+
+        # Цикл завершён — печатаем при изменении состояния.
+        if seminsu.prev_state is not None and seminsu.last_state is not None:
+            if seminsu.prev_state != seminsu.last_state:
+                # state: 1 = перекрыт, 0 = не перекрыт
+                print(f"EVENT IK_MODULE {_active_seminsu_id} {seminsu.last_state}")
+
+        _active_seminsu_id = None
+        return
+
+    # Берём следующий ID из объединённого round-robin.
+    sid = _next_seminsu_id()
+    if sid is None:
+        return
+
+    if sid in SEMINSU_V2:
+        SEMINSU_V2[sid].check()
+        return
+
+    # Иначе — v1.
+    seminsu = SEMINSU_V1.get(sid)
+    if seminsu is None:
+        return
+    _active_seminsu_id = sid
+    seminsu.start_poll()
+
     done = seminsu.update()
     if not done:
         return
 
-    # Цикл завершён — печатаем при изменении состояния.
+    # Быстрый случай: цикл успел завершиться в этом же шаге.
     if seminsu.prev_state is not None and seminsu.last_state is not None:
         if seminsu.prev_state != seminsu.last_state:
-            # state: 1 = перекрыт, 0 = не перекрыт
-            print(f"EVENT IK_MODULE {_active_seminsu_id} {seminsu.last_state}")
+            print(f"EVENT IK_MODULE {sid} {seminsu.last_state}")
 
     _active_seminsu_id = None
 
@@ -128,14 +181,22 @@ def work():
     poll = select.poll()
     poll.register(sys.stdin, select.POLLIN)
 
-    # Готовим список id seminsu для round-robin.
+    # Мигание встроенного светодиода: полный период 1 секунда.
+    # toggle() каждые 500 мс => 0.5с ON + 0.5с OFF.
+    LED.value(0)
+    last_led_toggle_ms = ticks_ms()
+
+    # Готовим список id IK_MODULE для round-robin (v1 + v2).
     global _seminsu_ids, _seminsu_idx, _active_seminsu_id
-    _seminsu_ids = list(SEMINSU.keys())
+    _seminsu_ids = sorted(list(SEMINSU_V1.keys()) + list(SEMINSU_V2.keys()))
     _seminsu_idx = 0
     _active_seminsu_id = None
 
     while True:
-        LED.toggle()
+        now_ms = ticks_ms()
+        if ticks_diff(now_ms, last_led_toggle_ms) >= 500:
+            LED.toggle()
+            last_led_toggle_ms = now_ms
 
         # 1) Обработка stdin НЕ блокирует цикл seminsu.
         events = poll.poll(0)
@@ -159,7 +220,9 @@ def work():
         # Небольшая пауза, чтобы не крутить CPU на 100%.
         sleep_ms(1)
 
+
 if __name__ == "__main__":
     load_switches()
-    load_seminsus()
-    work()
+    load_seminsus_v1()
+    load_seminsus_v2()
+    work()