cristhian
bde8582de3
The reader owns the pad on a thread of its own, and nothing outside that file may touch pygame: everything else sees a plain immutable Sample. That is what lets the rest of the node be tested with no hardware, and what keeps SDL's quirks out of the arm logic. Sample.present is the field that matters. A pad that has been unplugged reads as absent rather than as centred, so "the operator let go" and "the pad fell off the desk" stay distinguishable — they call for the same stop, but only one of them is worth logging. Only the display and joystick subsystems are initialised. pygame.init() also opens audio, which on a headless robot fills stderr with ALSA errors for a device this node never uses; the display is needed despite there being no window, because SDL refreshes joystick state during the event pump. Settings are parsed once at startup so a deadzone that would silently make the pad inert stops the node before it joins the dataflow.
136 lines
4.9 KiB
Python
136 lines
4.9 KiB
Python
"""The gamepad, and the thread that reads it.
|
|
|
|
**Nothing outside this file may touch pygame.** The rest of the node sees only
|
|
:class:`Sample`, which is a plain immutable reading — that is what lets every
|
|
other module be tested without a pad plugged in, and what keeps SDL's quirks
|
|
from leaking into the arm logic.
|
|
|
|
The reader is a thread because SDL wants to be pumped at its own cadence and
|
|
because a pad being unplugged should be noticed within milliseconds rather than
|
|
on whenever the dataflow next ticks. It never publishes anything; it fills a
|
|
slot, and the event loop drains it.
|
|
"""
|
|
|
|
import os
|
|
import threading
|
|
import time
|
|
from dataclasses import dataclass, field
|
|
|
|
from dora_teleop.log import log
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Sample:
|
|
"""One reading of the pad.
|
|
|
|
``present`` is the one that matters for safety: a pad that has been
|
|
unplugged reads as absent rather than as centred, so the difference between
|
|
"the operator let go" and "the operator's pad fell off the desk" is not lost.
|
|
"""
|
|
|
|
axes: list[float] = field(default_factory=list)
|
|
buttons: list[bool] = field(default_factory=list)
|
|
present: bool = False
|
|
|
|
@property
|
|
def active(self) -> bool:
|
|
"""Whether anything is being touched at all.
|
|
|
|
Deliberately raw — no deadzone. This decides whether the pad is asking
|
|
for the arm, and :mod:`dora_teleop.mapping` decides how far to move it.
|
|
A separate, tighter threshold here would mean a stick could claim the
|
|
arm and then not move it.
|
|
"""
|
|
return self.present and (any(abs(a) > 0.3 for a in self.axes) or any(self.buttons))
|
|
|
|
|
|
ABSENT = Sample()
|
|
|
|
|
|
class Gamepad:
|
|
"""Reads one pad on a thread and keeps the latest reading available."""
|
|
|
|
def __init__(self, index: int, poll_interval_s: float = 0.01):
|
|
self.index = index
|
|
self.poll_interval_s = poll_interval_s
|
|
self._lock = threading.Lock()
|
|
self._sample = ABSENT
|
|
self._stopping = threading.Event()
|
|
self._thread = threading.Thread(target=self._run, name="teleop-gamepad", daemon=True)
|
|
|
|
def start(self) -> None:
|
|
"""Start reading. Never blocks, and never fails for want of a pad."""
|
|
self._thread.start()
|
|
|
|
def read(self) -> Sample:
|
|
"""Return the most recent reading."""
|
|
with self._lock:
|
|
return self._sample
|
|
|
|
def shutdown(self) -> None:
|
|
"""Stop reading and let the thread finish."""
|
|
self._stopping.set()
|
|
self._thread.join(timeout=1.0)
|
|
|
|
def _run(self) -> None:
|
|
"""Open the pad, read it, and keep trying if it is not there.
|
|
|
|
A missing pad is an ordinary state, not an error: the browser is a whole
|
|
control path of its own, and the node has no business refusing to start
|
|
because nobody has plugged a controller in yet.
|
|
"""
|
|
# Two subsystems, not all of them. `pygame.init()` also opens audio,
|
|
# which on a headless robot means a screenful of ALSA errors in the
|
|
# dataflow's stderr for a device this node never uses. The display is
|
|
# needed despite there being no window: SDL refreshes joystick state
|
|
# during the event pump, and the pump needs the video subsystem.
|
|
os.environ.setdefault("SDL_VIDEODRIVER", "dummy")
|
|
os.environ.setdefault("SDL_AUDIODRIVER", "dummy")
|
|
import pygame
|
|
|
|
pygame.display.init()
|
|
pygame.joystick.init()
|
|
joystick = None
|
|
|
|
while not self._stopping.is_set():
|
|
pygame.event.pump()
|
|
if joystick is None:
|
|
joystick = self._open(pygame)
|
|
if joystick is None:
|
|
self._store(ABSENT)
|
|
self._stopping.wait(1.0)
|
|
continue
|
|
try:
|
|
sample = Sample(
|
|
axes=[joystick.get_axis(i) for i in range(joystick.get_numaxes())],
|
|
buttons=[bool(joystick.get_button(i)) for i in range(joystick.get_numbuttons())],
|
|
present=True,
|
|
)
|
|
except Exception as error: # noqa: BLE001 — an unplugged pad raises from inside SDL
|
|
log(f"the gamepad went away ({error})")
|
|
joystick = None
|
|
self._store(ABSENT)
|
|
continue
|
|
self._store(sample)
|
|
self._stopping.wait(self.poll_interval_s)
|
|
|
|
def _open(self, pygame):
|
|
"""Open the configured pad, or return None if it is not there."""
|
|
pygame.joystick.quit()
|
|
pygame.joystick.init()
|
|
if pygame.joystick.get_count() <= self.index:
|
|
return None
|
|
joystick = pygame.joystick.Joystick(self.index)
|
|
joystick.init()
|
|
log(f"gamepad {self.index}: {joystick.get_name()}")
|
|
return joystick
|
|
|
|
def _store(self, sample: Sample) -> None:
|
|
with self._lock:
|
|
self._sample = sample
|
|
|
|
|
|
def now() -> float:
|
|
"""Return the clock the deadman runs on: monotonic, so it cannot step backwards."""
|
|
return time.monotonic()
|