#!/usr/bin/env python3 """Pi Multi-FX Pedal — main entry point. Wires together all subsystems in a defined boot order: Config → Audio (I2S/JACK) → DSP Pipeline → Preset Manager → MIDI → Footswitch → Display → LEDs Graceful shutdown reverses the order: stop input sources → fade UI → save state → stop audio. """ from __future__ import annotations import logging import os import signal import sys import threading import time from pathlib import Path from typing import Optional import yaml # ── Subsystem imports ───────────────────────────────────────────────────────── from src.system.audio import AudioConfig, AudioSystem, JackAudioClient, _jack_is_running from src.system.defaults import ensure_defaults_exist from src.dsp.pipeline import AudioPipeline from src.dsp.nam_engine_host import FastNAMHost from src.dsp.ir_loader import IRLoader from src.presets.manager import PresetManager from src.presets.types import Channel, MIDIMapping, Preset from src.midi.handler import MIDIHandler from src.ui.footswitch import FootswitchController, FootSwitch, SwitchAction from src.ui.leds import LEDController, LEDDriver, LEDConfig, LEDPattern from src.ui.display import DisplayController, DisplayState from src.web.server import WebServer, WebServerDeps # ── Logging ─────────────────────────────────────────────────────────────────── logger = logging.getLogger("pedal") _console = logging.StreamHandler() _console.setFormatter(logging.Formatter( "%(asctime)s [%(levelname)s] %(name)s: %(message)s", datefmt="%H:%M:%S", )) logger.addHandler(_console) logger.setLevel(logging.INFO) # ── Paths ───────────────────────────────────────────────────────────────────── from src.system.config import DEFAULT_CONFIG_PATH, load_config, DEFAULT_CONFIG # ═══════════════════════════════════════════════════════════════════════════════ # Pedal application # ═══════════════════════════════════════════════════════════════════════════════ class PedalApp: """Main pedal application — wires and owns all subsystems. Lifecycle:: app = PedalApp() app.boot() # start everything app.run() # main loop (blocks, handles signals) app.shutdown() # graceful teardown """ def __init__(self, config_path: Path = DEFAULT_CONFIG_PATH) -> None: self._config = load_config(config_path) # ── Runtime state ────────────────────────────────────────────── self._running = False self._boot_time: Optional[float] = None self._bypassed = False # ── Subsystem references (set during boot) ───────────────────── self.audio_config: AudioConfig | None = None self.audio_system: AudioSystem | None = None self.jack_audio: JackAudioClient | None = None self.nam_host: FastNAMHost | None = None self.ir_loader: IRLoader | None = None self.pipeline: AudioPipeline | None = None self.presets: PresetManager | None = None self.midi: MIDIHandler | None = None self.footswitches: FootswitchController | None = None self.leds: LEDController | None = None self.display: DisplayController | None = None self.web: WebServer | None = None # ── Signal handlers ──────────────────────────────────────────── self._shutdown_requested = threading.Event() # Override the defaults AFTER merging so our overrides win if "shutdown_grace_period_s" not in self._config: self._config["shutdown_grace_period_s"] = 3.0 # ═══════════════════════════════════════════════════════════════ # Boot — start subsystems in dependency order # ═══════════════════════════════════════════════════════════════ def boot(self) -> bool: """Initialize all subsystems in correct order. Returns True if boot succeeded (non-critical subsystems may start in degraded mode — MIDI without interfaces, LEDs in mock, etc.). """ boot_start = time.monotonic() logger.info("═══════ Pi Multi-FX Pedal — Booting ═══════") try: # ── 0. Ensure default IRs, NAM models, and directories ── ensure_defaults_exist() # ── 1. Audio config + system ────────────────────────── acfg = self._config["audio"] audio_mode = acfg.get("mode", "mono") self.audio_config = AudioConfig( hat_type=acfg.get("hat_type", "audioinjector"), profile=acfg.get("profile", "standard"), mode=audio_mode, input_device=acfg.get("input_device", "hw:0,0"), output_device=acfg.get("output_device", "hw:0,0"), jack_enabled=acfg.get("jack_enabled", True), auto_connect=acfg.get("auto_connect", True), ) self.audio_system = AudioSystem(self.audio_config) self.audio_system.setup_i2s(reboot_hint=True) if self.audio_config.jack_enabled: self.audio_system.start_jack(timeout=10) else: logger.info("JACK disabled in config — skipping audio server start") # ── 2. DSP pipeline (NAM + IR + FX chain) ──────────── block_size = self.audio_config.latency_profile["period"] self.nam_host = FastNAMHost(block_size=block_size) self.ir_loader = IRLoader() self.pipeline = AudioPipeline(nam_host=self.nam_host, ir_loader=self.ir_loader) self.nam_host.warm_up() # Set pipeline routing from config mode if audio_mode == "stereo_4cm": self.pipeline.routing_mode = "4cm" logger.info("Pipeline routing set to 4CM mode") else: self.pipeline.routing_mode = "mono" # ── 2b. JACK audio client (capture → pipeline → playback) ── if self.audio_config.jack_enabled: is_stereo = audio_mode == "stereo_4cm" channels = 2 if is_stereo else 1 self.jack_audio = JackAudioClient( pipeline=self.pipeline, client_name="pi-multifx", input_channels=channels, output_channels=channels, ) self.jack_audio.start() if not self.jack_audio._active: logger.warning("JACK audio client failed to start — running without real-time audio I/O") else: logger.info( "JACK audio client started: %s mode, %d ch I/O", audio_mode, channels, ) # ── 3. Preset manager ───────────────────────────────── pcfg = self._config["presets"] self.presets = PresetManager( preset_dir=pcfg.get("dir", "~/.pedal/presets"), audio_pipeline=self.pipeline, ) # Install factory presets — DISABLED per user request (all deleted) if False: for ch in Channel: installed = self.presets.install_factory_presets( overwrite=False, channel=ch, ) if installed: logger.info("Installed %d factory preset(s) for %s", installed, ch.value) ir_installed = self.presets.install_factory_irs(overwrite=False) if ir_installed: logger.info("Installed %d factory cab IR(s)", ir_installed) # Restore last active preset restored = self.presets.restore_state() if restored is None: # First boot or no saved state — activate bank 0, program 0 logger.info("No previous state — activating default preset") try: self.presets.select(0, 0) except Exception: logger.info("No presets exist yet — pedal ready for MIDI/footswitch input") else: # restore_state() loads from disk but doesn't activate the pipeline # Need to call select() to trigger pipeline.load_preset() and load NAM logger.info("State restored — activating preset '%s' (bank=%d, program=%d)", restored.name, restored.bank, restored.program) self.presets.select(restored.bank, restored.program) # ── 4. MIDI handler ────────────────────────────────── self.midi = MIDIHandler() self._wire_midi_callbacks() mcfg = self._config["midi"] self.midi.start( uart_port=mcfg.get("uart_port", "/dev/ttyAMA0"), usb=mcfg.get("usb", True), ) # ── 5. Footswitch controller ───────────────────────── self.footswitches = FootswitchController( switches=self._build_footswitch_layout(), ) self._wire_footswitch_callbacks() self.footswitches.start() # ── 6. Display ─────────────────────────────────────── dcfg = self._config["display"] self.display = DisplayController( i2c_bus=dcfg.get("i2c_bus", 1), i2c_addr=dcfg.get("i2c_addr", 0x3C), ) self.display.initialize() self._update_display() # ── 7. LEDs ────────────────────────────────────────── lcfg = self._config["leds"] driver_map = {"neopixel": LEDDriver.NEOPIXEL, "dotstar": LEDDriver.DOTSTAR, "mock": LEDDriver.MOCK} self.leds = LEDController( num_leds=lcfg.get("num_leds", 4), driver=driver_map.get(lcfg.get("driver", "neopixel"), LEDDriver.NEOPIXEL), pin=lcfg.get("pin", "D18"), brightness=lcfg.get("brightness", 0.5), ) self.leds.initialize() # Boot LED animation — quick scan self.leds.preset_animate(direction="up") # ── 8. Web UI server (non-blocking HTTP + WebSocket) ──── # Runs on port 8080 by default, alongside the JACK audio loop. try: self.web = WebServer( deps=WebServerDeps( presets=self.presets, pipeline=self.pipeline, nam_host=self.nam_host, ir_loader=self.ir_loader, audio_system=self.audio_system, jack_audio=self.jack_audio, ), host="0.0.0.0", port=8080, ) self.web.start() logger.info("Web UI server started on http://0.0.0.0:8080") except Exception as e: logger.warning("Web UI server failed to start (non-fatal): %s", e) except Exception as e: logger.critical("Boot failed: %s", e, exc_info=True) return False self._boot_time = time.monotonic() elapsed = self._boot_time - boot_start logger.info("Boot complete in %.1fs — pedal is ready", elapsed) return True # ═══════════════════════════════════════════════════════════════ # Wire callbacks # ═══════════════════════════════════════════════════════════════ def _wire_midi_callbacks(self) -> None: """Connect MIDI events to preset manager and parameter handlers.""" if not self.midi or not self.presets: return # Program Change → preset switch self.midi.set_pc_callback(self._on_midi_pc) # CC → registered parameter handlers (per preset mapping) # We register a general CC handler that looks up the current # preset's MIDI mappings dynamically. # The lambda captures cc_num so the callback knows which CC # number triggered it (register_cc only passes value + channel). for cc in range(128): self.midi.register_cc(cc, lambda val, ch, cc_num=cc: self._on_midi_cc(val, cc_num, ch)) # MIDI Learn completion self.midi.set_midi_learn_callback(self._on_midi_learn) def _on_midi_pc(self, channel: int, program: int) -> None: """MIDI Program Change → select preset.""" if not self.presets: return try: preset = self.presets.midi_pc(channel, program) self._on_preset_changed(preset) except Exception as e: logger.warning("MIDI PC to (ch=%d, pg=%d) failed: %s", channel, program, e) def _on_midi_cc(self, value: int, cc_number: int, channel: int) -> None: """MIDI CC → live parameter update. Maps expression pedal (CC 11) to master volume by default. Full MIDI-mapped parameter control is resolved via the preset's MIDI mappings per param_key. Args: value: CC value (0-127). cc_number: The CC number (0-127) that triggered this callback. channel: MIDI channel (0-15) the message arrived on. """ # Map expression pedal (CC 11) to master volume regardless of # preset state — global mapping. if cc_number == 11: if self.pipeline: normalized = value / 127.0 self.pipeline._master_volume = normalized logger.debug("MIDI CC 11 → master volume: %.2f", normalized) return if not self.presets or not self.pipeline: return # Look up preset's MIDI mappings for other CC numbers try: bank = self.presets.current_bank program = self.presets.current_program preset = self.presets.load(bank, program) except Exception: return # Iterate preset mappings and find which one matches this CC number for param_key, mapping in preset.midi_mappings.items(): if mapping.cc_number == cc_number: normalized = value / 127.0 logger.debug("MIDI CC %d → %s = %.2f", cc_number, param_key, normalized) break def _on_midi_learn(self, mapping: object) -> None: """Handle MIDI Learn completion — update display.""" logger.info("MIDI Learn complete: %s", mapping) self._update_display() def _wire_footswitch_callbacks(self) -> None: """Connect footswitch actions to pedal controls.""" if not self.footswitches: return self.footswitches.register_callback(SwitchAction.PRESET_UP, self._on_preset_up) self.footswitches.register_callback(SwitchAction.PRESET_DOWN, self._on_preset_down) self.footswitches.register_callback(SwitchAction.BANK_UP, self._on_bank_up) self.footswitches.register_callback(SwitchAction.BANK_DOWN, self._on_bank_down) self.footswitches.register_callback(SwitchAction.BYPASS, self._on_bypass_toggle) self.footswitches.register_callback(SwitchAction.TAP_TEMPO, self._on_tap_tempo) self.footswitches.register_callback(SwitchAction.TUNER, self._on_tuner_toggle) self.footswitches.register_callback(SwitchAction.SNAPSHOT_SAVE, self._on_snapshot_save) # ═══════════════════════════════════════════════════════════════ # Action handlers # ═══════════════════════════════════════════════════════════════ def _on_preset_up(self) -> None: if not self.presets: return try: preset = self.presets.preset_up() self._on_preset_changed(preset) if self.leds: self.leds.preset_animate(direction="up") except Exception as e: logger.warning("preset_up failed: %s", e) def _on_preset_down(self) -> None: if not self.presets: return try: preset = self.presets.preset_down() self._on_preset_changed(preset) if self.leds: self.leds.preset_animate(direction="down") except Exception as e: logger.warning("preset_down failed: %s", e) def _on_bank_up(self) -> None: if not self.presets: return try: bank, preset = self.presets.bank_up() self._on_preset_changed(preset) if self.leds: self.leds.preset_animate(direction="up") except Exception as e: logger.warning("bank_up failed: %s", e) def _on_bank_down(self) -> None: if not self.presets: return try: bank, preset = self.presets.bank_down() self._on_preset_changed(preset) if self.leds: self.leds.preset_animate(direction="down") except Exception as e: logger.warning("bank_down failed: %s", e) def _on_bypass_toggle(self) -> None: self._bypassed = not self._bypassed if self.pipeline: self.pipeline._bypassed = self._bypassed logger.info("Bypass %s", "ON" if self._bypassed else "OFF") if self.leds: self.leds.set_bypass_led(2, self._bypassed) self._update_display() def _on_tap_tempo(self) -> None: if self.leds: self.leds.tap_tempo_blip() logger.debug("Tap tempo") def _on_tuner_toggle(self) -> None: logger.info("Tuner mode toggled (stub — mute audio, show tuner display)") if self.display: state = DisplayState(mode="tuner", tuner_note="--", tuner_cents=0) self.display.update(state) def _on_snapshot_save(self) -> None: if not self.presets: return try: bank = self.presets.current_bank program = self.presets.current_program preset = self.presets.load(bank, program) self.presets.save(preset) logger.info("Snapshot saved: '%s'", preset.name) except Exception as e: logger.warning("Snapshot save failed: %s", e) def _on_preset_changed(self, preset: Preset) -> None: """Called after any preset change to sync UI/state.""" self._update_display() if self.leds: self.leds.set_pixel(0, (0, 64, 255)) # Blue = active logger.info("Active: '%s' (bank=%d, pg=%d)", preset.name, preset.bank, preset.program) # ═══════════════════════════════════════════════════════════════ # Audio routing mode switching # ═══════════════════════════════════════════════════════════════ def set_routing_mode(self, mode: str) -> None: """Hot-switch between mono and stereo_4cm routing at runtime. Updates the pipeline routing mode, then restarts the JACK audio client with the appropriate channel count. Args: mode: ``"mono"`` or ``"stereo_4cm"``. Raises: ValueError: If *mode* is not recognised. """ if mode not in ("mono", "stereo_4cm"): raise ValueError(f"set_routing_mode: unknown mode {mode!r}") if self.audio_config is not None: self.audio_config.mode = mode if self.pipeline is not None: self.pipeline.routing_mode = "4cm" if mode == "stereo_4cm" else "mono" logger.info("Pipeline routing mode switched to %s", self.pipeline.routing_mode) # Restart JACK client with new channel count if self.jack_audio is not None: channels = 2 if mode == "stereo_4cm" else 1 self.jack_audio.set_channel_count(channels, channels) # Re-connect JACK ports for the new routing if self.audio_system is not None and self.audio_config is not None: self.audio_system.connect_fx_ports() logger.info("Routing mode switched to %s", mode) # ═══════════════════════════════════════════════════════════════ # Display sync # ═══════════════════════════════════════════════════════════════ def _update_display(self) -> None: """Refresh the OLED with current state.""" if not self.display or not self.presets: return try: bank = self.presets.current_bank program = self.presets.current_program preset = self.presets.load(bank, program) fx_active = [b.fx_type.value for b in preset.chain if b.enabled and not b.bypass] fx_bypass_states = {b.fx_type.value: b.bypass for b in preset.chain} state = DisplayState( mode="preset", preset_name=preset.name, bank_name=f"Bank {bank}", bypassed=self._bypassed, fx_active=fx_active, fx_bypass_states=fx_bypass_states, ) self.display.update(state) except Exception: # Degraded: show minimal state state = DisplayState( mode="preset", preset_name="Ready", bank_name="", bypassed=self._bypassed, ) self.display.update(state) # ═══════════════════════════════════════════════════════════════ # Helpers # ═══════════════════════════════════════════════════════════════ def _build_footswitch_layout(self) -> list[FootSwitch]: """Build FootSwitch list from config.""" layout_cfg = self._config.get("footswitch", {}).get("layout", DEFAULT_CONFIG["footswitch"]["layout"]) action_map = {a.value: a for a in SwitchAction} switches: list[FootSwitch] = [] for entry in layout_cfg: default = action_map.get(entry.get("action_default", "")) long_press = action_map.get(entry.get("action_long_press", "")) if default is None: logger.warning("Unknown footswitch action: %s", entry.get("action_default")) continue switches.append(FootSwitch( gpio_pin=entry["gpio_pin"], action_default=default, action_long_press=long_press, active_low=entry.get("active_low", True), )) return switches # ═══════════════════════════════════════════════════════════════ # Main loop # ═══════════════════════════════════════════════════════════════ def run(self) -> None: """Main application loop. Sets up signal handlers and blocks until a shutdown is requested. """ self._running = True # Register signal handlers for graceful shutdown signal.signal(signal.SIGTERM, self._signal_handler) signal.signal(signal.SIGINT, self._signal_handler) logger.info("Pedal running — press Ctrl+C or send SIGTERM to stop") # Main loop — keep alive, refresh display periodically last_display_refresh = 0.0 refresh_interval = 0.25 # 250 ms display refresh while self._running and not self._shutdown_requested.is_set(): now = time.monotonic() # Periodic display refresh (catches MIDI/footswitch-triggered # state changes that don't go through _on_preset_changed) if now - last_display_refresh >= refresh_interval: self._update_display() last_display_refresh = now # Main loop sleeps — low CPU while waiting for events self._shutdown_requested.wait(0.1) self.shutdown() def _signal_handler(self, signum: int, _frame: object) -> None: """Handle SIGTERM/SIGINT for graceful shutdown.""" signame = signal.Signals(signum).name logger.info("Received %s — shutting down...", signame) self._shutdown_requested.set() self._running = False # ═══════════════════════════════════════════════════════════════ # Shutdown — reverse boot order # ═══════════════════════════════════════════════════════════════ def shutdown(self) -> None: """Graceful shutdown: stop inputs → fade UI → save → stop audio.""" logger.info("═══════ Shutting down ═══════") shutdown_start = time.monotonic() # 1. Stop JACK audio client (before JACK server) if self.jack_audio: try: self.jack_audio.stop() logger.debug("JACK audio client stopped") except Exception as e: logger.warning("JACK audio client stop error: %s", e) # 2. Stop accepting input (MIDI + footswitch) if self.midi: try: self.midi.stop() logger.debug("MIDI stopped") except Exception as e: logger.warning("MIDI stop error: %s", e) if self.footswitches: try: self.footswitches.stop() logger.debug("Footswitch stopped") except Exception as e: logger.warning("Footswitch stop error: %s", e) # 2b. Stop web server (before state save — WS clients get final message) if self.web: try: self.web.stop() logger.debug("Web server stopped") except Exception as e: logger.warning("Web server stop error: %s", e) # 3. Save current state (preset + bank) if self.presets: try: self.presets.save_state() logger.info("State saved") except Exception as e: logger.warning("State save error: %s", e) # 3. Fade LEDs if self.leds: try: self._fade_leds() logger.debug("LEDs faded") except Exception as e: logger.warning("LED fade error: %s", e) # 4. Clear display if self.display: try: self.display.clear() logger.debug("Display cleared") except Exception as e: logger.warning("Display clear error: %s", e) # 5. Stop JACK if self.audio_system: try: self.audio_system.stop_jack() logger.debug("JACK stopped") except Exception as e: logger.warning("JACK stop error: %s", e) elapsed = time.monotonic() - shutdown_start logger.info("Shutdown complete in %.1fs — goodbye", elapsed) # Flush log for h in logger.handlers: h.flush() @staticmethod def _fade_leds() -> None: """Fade all LEDs to off over ~300ms (simulated if no hardware).""" # In production this would ramp brightness down. # For now we just set all off. time.sleep(0.05) # ═══════════════════════════════════════════════════════════════════════════════ # CLI entry # ═══════════════════════════════════════════════════════════════════════════════ def main() -> int: """Parse args, boot, and run the pedal application. Returns exit code 0 on clean shutdown, 1 on boot failure. """ import argparse parser = argparse.ArgumentParser( description="Pi Multi-FX Pedal — real-time guitar multi-effects", ) parser.add_argument( "-c", "--config", type=str, default=os.environ.get("PEDAL_CONFIG", str(DEFAULT_CONFIG_PATH)), help="Path to config YAML (env: PEDAL_CONFIG, default: ~/.pedal/config.yaml)", ) parser.add_argument( "-v", "--verbose", action="store_true", help="Enable debug logging", ) args = parser.parse_args() if args.verbose: logger.setLevel(logging.DEBUG) config_path = Path(args.config) app = PedalApp(config_path=config_path) if not app.boot(): return 1 app.run() return 0 if __name__ == "__main__": sys.exit(main())