More hw synth refinement.
This commit is contained in:
@@ -6,6 +6,7 @@
|
||||
|
||||
import glob
|
||||
import os
|
||||
import select
|
||||
import termios
|
||||
import threading
|
||||
import tty
|
||||
@@ -28,6 +29,8 @@ class SpeakQueue(Queue):
|
||||
class hardware_serial_driver(speech_driver):
|
||||
cancel_command = b""
|
||||
default_baud_rate = 9600
|
||||
hardware_probe_command = b""
|
||||
hardware_probe_timeout = 0.2
|
||||
|
||||
def __init__(self):
|
||||
speech_driver.__init__(self)
|
||||
@@ -147,14 +150,50 @@ class hardware_serial_driver(speech_driver):
|
||||
)
|
||||
|
||||
def _open_serial_port(self):
|
||||
device = self._resolve_device(self.device)
|
||||
if not device:
|
||||
auto_detect = self.device == "auto"
|
||||
devices = self._resolve_devices(self.device)
|
||||
if not devices:
|
||||
self._debug(
|
||||
"Hardware speech device not found",
|
||||
debug.DebugLevel.ERROR,
|
||||
on_any_level=True,
|
||||
)
|
||||
return
|
||||
|
||||
fallback_device = None
|
||||
fallback_port = None
|
||||
for device in devices:
|
||||
port = self._open_configured_serial_port(device)
|
||||
if port is None:
|
||||
continue
|
||||
if not auto_detect:
|
||||
self._activate_serial_port(
|
||||
device, port, probe_matched=False
|
||||
)
|
||||
return
|
||||
if self._probe_serial_port(port):
|
||||
if fallback_port is not None:
|
||||
self._close_port(fallback_port)
|
||||
self._activate_serial_port(device, port, probe_matched=True)
|
||||
return
|
||||
if fallback_port is None:
|
||||
fallback_device = device
|
||||
fallback_port = port
|
||||
else:
|
||||
self._close_port(port)
|
||||
|
||||
if fallback_port is not None:
|
||||
self._debug(
|
||||
"Hardware speech probe did not identify a synth; "
|
||||
f"falling back to {fallback_device}",
|
||||
debug.DebugLevel.WARNING,
|
||||
on_any_level=True,
|
||||
)
|
||||
self._activate_serial_port(
|
||||
fallback_device, fallback_port, probe_matched=False
|
||||
)
|
||||
|
||||
def _open_configured_serial_port(self, device):
|
||||
try:
|
||||
port = os.open(device, os.O_RDWR | os.O_NOCTTY)
|
||||
tty.setraw(port)
|
||||
@@ -167,35 +206,86 @@ class hardware_serial_driver(speech_driver):
|
||||
attrs[6][termios.VTIME] = 0
|
||||
attrs[0] &= ~(termios.IXON | termios.IXOFF | termios.IXANY)
|
||||
termios.tcsetattr(port, termios.TCSANOW, attrs)
|
||||
self.serial_port = port
|
||||
self.device = device
|
||||
self._debug(
|
||||
"Hardware speech device opened: "
|
||||
f"{device}, baud_rate={self.baud_rate}",
|
||||
debug.DebugLevel.INFO,
|
||||
on_any_level=True,
|
||||
)
|
||||
return port
|
||||
except OSError as error:
|
||||
self._debug(
|
||||
f"Hardware speech device open failed: {device}: {error}",
|
||||
debug.DebugLevel.ERROR,
|
||||
on_any_level=True,
|
||||
)
|
||||
self.serial_port = None
|
||||
return None
|
||||
|
||||
def _activate_serial_port(self, device, port, probe_matched=False):
|
||||
self.serial_port = port
|
||||
self.device = device
|
||||
probe_status = "matched" if probe_matched else "not matched"
|
||||
self._debug(
|
||||
"Hardware speech device opened: "
|
||||
f"{device}, baud_rate={self.baud_rate}, probe={probe_status}",
|
||||
debug.DebugLevel.INFO,
|
||||
on_any_level=True,
|
||||
)
|
||||
|
||||
def _probe_serial_port(self, port):
|
||||
if not self.hardware_probe_command:
|
||||
return True
|
||||
try:
|
||||
self._drain_serial_input(port)
|
||||
os.write(port, self.hardware_probe_command)
|
||||
readable, _, _ = select.select(
|
||||
[port], [], [], self.hardware_probe_timeout
|
||||
)
|
||||
if not readable:
|
||||
self._debug(
|
||||
"Hardware speech probe got no response",
|
||||
debug.DebugLevel.INFO,
|
||||
on_any_level=True,
|
||||
)
|
||||
return False
|
||||
response = os.read(port, 256)
|
||||
self._debug(
|
||||
"Hardware speech probe response: "
|
||||
f"{self._format_bytes_preview(response)}",
|
||||
debug.DebugLevel.INFO,
|
||||
on_any_level=True,
|
||||
)
|
||||
return self._is_hardware_probe_response(response)
|
||||
except OSError as error:
|
||||
self._debug(
|
||||
f"Hardware speech probe failed: {error}",
|
||||
debug.DebugLevel.WARNING,
|
||||
on_any_level=True,
|
||||
)
|
||||
return False
|
||||
|
||||
def _drain_serial_input(self, port):
|
||||
while True:
|
||||
readable, _, _ = select.select([port], [], [], 0)
|
||||
if not readable:
|
||||
return
|
||||
if not os.read(port, 256):
|
||||
return
|
||||
|
||||
def _is_hardware_probe_response(self, response):
|
||||
return bool(response)
|
||||
|
||||
def _close_serial_port(self):
|
||||
with self.lock:
|
||||
if self.serial_port is None:
|
||||
return
|
||||
self._close_port(self.serial_port)
|
||||
self.serial_port = None
|
||||
|
||||
def _close_port(self, port):
|
||||
if port is None:
|
||||
return
|
||||
try:
|
||||
os.close(self.serial_port)
|
||||
os.close(port)
|
||||
except OSError as error:
|
||||
self._debug(
|
||||
f"Hardware speech device close failed: {error}",
|
||||
debug.DebugLevel.WARNING,
|
||||
)
|
||||
finally:
|
||||
self.serial_port = None
|
||||
|
||||
def _write_bytes(self, data, description="data"):
|
||||
if not data:
|
||||
@@ -226,14 +316,15 @@ class hardware_serial_driver(speech_driver):
|
||||
on_any_level=True,
|
||||
)
|
||||
|
||||
def _resolve_device(self, device):
|
||||
def _resolve_devices(self, device):
|
||||
if device and device != "auto":
|
||||
self._debug(
|
||||
f"Hardware speech using configured device: {device}",
|
||||
debug.DebugLevel.INFO,
|
||||
on_any_level=True,
|
||||
)
|
||||
return device
|
||||
return [device]
|
||||
devices = []
|
||||
for pattern in ("/dev/ttyACM*", "/dev/ttyUSB*", "/dev/ttyS*"):
|
||||
matches = sorted(glob.glob(pattern))
|
||||
self._debug(
|
||||
@@ -242,13 +333,20 @@ class hardware_serial_driver(speech_driver):
|
||||
on_any_level=True,
|
||||
)
|
||||
if matches:
|
||||
if len(matches) > 1:
|
||||
self._debug(
|
||||
f"Hardware speech auto selected device: {matches[0]}",
|
||||
"Hardware speech auto found multiple devices for "
|
||||
f"{pattern}: {matches}; probing in order",
|
||||
debug.DebugLevel.WARNING,
|
||||
on_any_level=True,
|
||||
)
|
||||
self._debug(
|
||||
f"Hardware speech auto candidate devices: {matches}",
|
||||
debug.DebugLevel.INFO,
|
||||
on_any_level=True,
|
||||
)
|
||||
return matches[0]
|
||||
return ""
|
||||
devices.extend(matches)
|
||||
return devices
|
||||
|
||||
def _termios_baud_rate(self, baud_rate):
|
||||
baud_name = f"B{baud_rate}"
|
||||
|
||||
@@ -11,6 +11,7 @@ from fenrirscreenreader.speechDriver.hardwareSerialDriver import (
|
||||
|
||||
class driver(hardware_serial_driver):
|
||||
cancel_command = b"\x18"
|
||||
hardware_probe_command = b"\x01I"
|
||||
|
||||
def _speak_bytes(self, text):
|
||||
return self._clean_text(text).encode("ascii", errors="replace") + b"\r"
|
||||
@@ -26,3 +27,10 @@ class driver(hardware_serial_driver):
|
||||
|
||||
def _setting_command(self, value, command):
|
||||
return b"\x01" + str(value).encode("ascii") + command
|
||||
|
||||
def _is_hardware_probe_response(self, response):
|
||||
response_text = response.decode("ascii", errors="ignore").lower()
|
||||
return any(
|
||||
token in response_text
|
||||
for token in ("litetalk", "lite talk", "rpitalk", "doubletalk")
|
||||
)
|
||||
|
||||
@@ -123,7 +123,139 @@ def test_auto_device_detection_includes_classic_serial(
|
||||
try:
|
||||
assert speech_driver.device == slave_name
|
||||
speech_driver.speak("Serial")
|
||||
assert read_available(master_fd, 7) == b"Serial\r"
|
||||
assert read_available(master_fd, 9) == b"\x01ISerial\r"
|
||||
finally:
|
||||
speech_driver.shutdown()
|
||||
|
||||
|
||||
def test_auto_device_detection_prefers_probe_response(monkeypatch):
|
||||
opened_ports = []
|
||||
closed_ports = []
|
||||
writes = {}
|
||||
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.glob.glob",
|
||||
lambda pattern: ["/dev/ttyUSB0", "/dev/ttyUSB1"]
|
||||
if pattern == "/dev/ttyUSB*"
|
||||
else [],
|
||||
)
|
||||
|
||||
def fake_open(device, flags):
|
||||
port = 100 + len(opened_ports)
|
||||
opened_ports.append((device, port))
|
||||
return port
|
||||
|
||||
def fake_write(port, data):
|
||||
writes.setdefault(port, b"")
|
||||
writes[port] += data
|
||||
return len(data)
|
||||
|
||||
def fake_select(readable, writable, exceptional, timeout):
|
||||
port = readable[0]
|
||||
if port == 101 and writes.get(port) == b"\x01I":
|
||||
return readable, writable, exceptional
|
||||
return [], writable, exceptional
|
||||
|
||||
def fake_read(port, size):
|
||||
if port == 101:
|
||||
return b"RPItalk 1.3\r"
|
||||
return b""
|
||||
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.os.open",
|
||||
fake_open,
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.os.close",
|
||||
lambda port: closed_ports.append(port),
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.os.write",
|
||||
fake_write,
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.os.read",
|
||||
fake_read,
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.select.select",
|
||||
fake_select,
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.termios.tcgetattr",
|
||||
lambda port: [0, 0, 0, 0, 0, 0, [0] * 32],
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.termios.tcsetattr",
|
||||
lambda port, when, attrs: None,
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.tty.setraw",
|
||||
lambda port: None,
|
||||
)
|
||||
|
||||
speech_driver = litetalkDriver.driver()
|
||||
speech_driver.initialize(build_environment("auto"))
|
||||
try:
|
||||
assert opened_ports == [("/dev/ttyUSB0", 100), ("/dev/ttyUSB1", 101)]
|
||||
assert writes == {100: b"\x01I", 101: b"\x01I"}
|
||||
assert closed_ports == [100]
|
||||
assert speech_driver.device == "/dev/ttyUSB1"
|
||||
finally:
|
||||
speech_driver.shutdown()
|
||||
|
||||
|
||||
def test_auto_device_detection_falls_back_without_probe_response(
|
||||
monkeypatch
|
||||
):
|
||||
opened_ports = []
|
||||
closed_ports = []
|
||||
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.glob.glob",
|
||||
lambda pattern: ["/dev/ttyUSB0", "/dev/ttyUSB1"]
|
||||
if pattern == "/dev/ttyUSB*"
|
||||
else [],
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.os.open",
|
||||
lambda device, flags: opened_ports.append(device) or len(opened_ports),
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.os.close",
|
||||
lambda port: closed_ports.append(port),
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.os.write",
|
||||
lambda port, data: len(data),
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.select.select",
|
||||
lambda readable, writable, exceptional, timeout: (
|
||||
[],
|
||||
writable,
|
||||
exceptional,
|
||||
),
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.termios.tcgetattr",
|
||||
lambda port: [0, 0, 0, 0, 0, 0, [0] * 32],
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.termios.tcsetattr",
|
||||
lambda port, when, attrs: None,
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"fenrirscreenreader.speechDriver.hardwareSerialDriver.tty.setraw",
|
||||
lambda port: None,
|
||||
)
|
||||
|
||||
speech_driver = litetalkDriver.driver()
|
||||
speech_driver.initialize(build_environment("auto"))
|
||||
try:
|
||||
assert opened_ports == ["/dev/ttyUSB0", "/dev/ttyUSB1"]
|
||||
assert closed_ports == [2]
|
||||
assert speech_driver.device == "/dev/ttyUSB0"
|
||||
finally:
|
||||
speech_driver.shutdown()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user