Add HAL and Marvin

Files specifically configured for my personal use cases of a HAL-9000 and Marvin voice assistants.

linux_voice_assistant/led_controller_marvin.py

@@ -0,0 +1,135 @@
+import asyncio
+import logging
+from typing import Any, Tuple
+
+import board
+import adafruit_dotstar
+
+from .event_bus import EventHandler, subscribe
+
+_LOGGER = logging.getLogger(__name__)
+
+# --- Marvin Theme Constants ---
+NUM_LEDS = 2
+_OFF = (0, 0, 0)
+_GREEN = (0, 255, 0)
+
+
+class LedController(EventHandler):
+    def __init__(
+        self,
+        state,
+        interface: str = "spi",
+        clock_pin: int = 13,
+        data_pin: int = 12,
+    ):
+        super().__init__(state)
+        self.loop = state.loop
+        self.current_task = None
+        self._is_ready = False
+
+        try:
+            if interface == "gpio":
+                _LOGGER.debug(f"Initializing LEDs on GPIO data={data_pin}, clock={clock_pin}")
+                data_pin_obj = getattr(board, f"D{data_pin}")
+                clock_pin_obj = getattr(board, f"D{clock_pin}")
+                self.leds = adafruit_dotstar.DotStar(
+                    clock_pin_obj, data_pin_obj, NUM_LEDS, brightness=0.5, auto_write=False
+                )
+            else:  # Default to SPI
+                _LOGGER.debug("Initializing LEDs on hardware SPI")
+                self.leds = adafruit_dotstar.DotStar(
+                    board.SCLK, board.MOSI, NUM_LEDS, brightness=0.5, auto_write=False
+                )
+            
+            self._is_ready = True
+            _LOGGER.info(f"LED Controller initialized using {interface} interface.")
+            # Signal that the system is ready
+            self.run_action("blink", _GREEN, 2)
+
+        except Exception:
+            _LOGGER.exception("Failed to initialize LED controller. LEDs will be disabled.")
+
+    def run_action(self, action_method_name: str, *args: Any) -> None:
+        if not self._is_ready:
+            return
+        if self.current_task and not self.current_task.done():
+            self.current_task.cancel()
+        
+        coro = getattr(self, action_method_name)(*args)
+        self.current_task = asyncio.run_coroutine_threadsafe(coro, self.loop)
+
+    async def color(self, rgb: Tuple[int, int, int]):
+        if not self._is_ready:
+            return
+        self.leds.fill(rgb)
+        self.leds.show()
+
+    async def blink(self, color, count=3):
+        for _ in range(count):
+            await self.color(color)
+            await asyncio.sleep(0.2)
+            await self.color(_OFF)
+            await asyncio.sleep(0.2)
+
+    async def pulse(self, color: Tuple[int, int, int], speed: float = 0.015):
+        try:
+            r, g, b = color
+            while True:
+                for i in range(0, 101, 5):
+                    brightness = i / 100.0
+                    await self.color((int(r * brightness), int(g * brightness), int(b * brightness)))
+                    await asyncio.sleep(speed)
+                for i in range(100, -1, -5):
+                    brightness = i / 100.0
+                    await self.color((int(r * brightness), int(g * brightness), int(b * brightness)))
+                    await asyncio.sleep(speed)
+        except asyncio.CancelledError:
+            await self.color(_OFF)
+
+    async def fast_pulse(self, color: Tuple[int, int, int], speed: float = 0.008):
+        """Faster version of the pulse animation."""
+        try:
+            r, g, b = color
+            while True:
+                for i in range(0, 101, 10):
+                    brightness = i / 100.0
+                    await self.color((int(r * brightness), int(g * brightness), int(b * brightness)))
+                    await asyncio.sleep(speed)
+                for i in range(100, -1, -10):
+                    brightness = i / 100.0
+                    await self.color((int(r * brightness), int(g * brightness), int(b * brightness)))
+                    await asyncio.sleep(speed)
+        except asyncio.CancelledError:
+            await self.color(_OFF)
+
+
+    # --- EVENT HANDLERS FOR MARVIN ---
+    @subscribe
+    def ha_connected(self, data: dict):
+        _LOGGER.debug("HA Connected, setting idle LED state.")
+        self.voice_run_end(data)
+
+    @subscribe
+    def voice_wakeword(self, data: dict):
+        self.run_action("pulse", _GREEN)
+    
+    @subscribe
+    def voice_stt_start(self, data: dict):
+        self.run_action("pulse", _GREEN)
+
+    @subscribe
+    def voice_vad_start(self, data: dict):
+        pass # Continues pulsing green
+
+    @subscribe
+    def voice_stt_end(self, data: dict):
+        self.run_action("fast_pulse", _GREEN)
+
+    @subscribe
+    def voice_tts_start(self, data: dict):
+        self.run_action("pulse", _GREEN)
+
+    @subscribe
+    def voice_run_end(self, data: dict):
+        self.run_action("color", _GREEN) # Idle state is solid green