You've already forked MicroPythonOS
mirror of
https://github.com/m5stack/MicroPythonOS.git
synced 2026-05-20 11:51:27 -07:00
Bert Outtier
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
import struct
|
|
|
|
from micropython import const
|
|
from machine import I2C, Pin
|
|
|
|
from .device import Device
|
|
|
|
# registers
|
|
_EXPANDER_REG_INPUTS = const(0x04)
|
|
_EXPANDER_REG_ANALOG = const(0x06)
|
|
_EXPANDER_REG_LCD_BRIGHTNESS = const(0x12)
|
|
_EXPANDER_REG_DEBUG_LED = const(0x14)
|
|
_EXPANDER_REG_CONFIG = const(0x16)
|
|
|
|
_EXPANDER_I2CADDR_DEFAULT = const(0x50)
|
|
|
|
|
|
class Expander(Device):
|
|
"""Fri3d Badge 2026 expander MCU."""
|
|
|
|
def __init__(
|
|
self,
|
|
i2c_bus: I2C,
|
|
address: int = _EXPANDER_I2CADDR_DEFAULT,
|
|
int_pin: Pin = None,
|
|
):
|
|
"""Read from a sensor on the given I2C bus, at the given address."""
|
|
Device.__init__(self, i2c_bus, address)
|
|
self.use_interrupt = False
|
|
if int_pin:
|
|
self.use_interrupt = True
|
|
self._rx_buf = bytearray(2)
|
|
self._rx_mv = memoryview(self._rx_buf)
|
|
self.int_pin = int_pin
|
|
self.i2c.readfrom_mem_into(self.address, _EXPANDER_REG_INPUTS, self._rx_mv)
|
|
self.int_pin.irq(trigger=Pin.IRQ_RISING, handler=self.int_callback)
|
|
|
|
def int_callback(self, p):
|
|
self.i2c.readfrom_mem_into(self.address, _EXPANDER_REG_INPUTS, self._rx_mv)
|
|
|
|
@property
|
|
def analog(self) -> tuple[int, int, int, int, int, int]:
|
|
"""Read the analog inputs: ain1, ain0, battery_monitor, usb_monitor, joystick_y, joystick_x"""
|
|
return self._read("<HHHHHH", _EXPANDER_REG_ANALOG, 12)
|
|
|
|
@property
|
|
def digital(
|
|
self,
|
|
) -> tuple[bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool]:
|
|
"""Read the digital inputs: usb_plugged, joy_right, joy_left, joy_down, joy_up, button_menu, button_b, button_a, button_y, button_x, charger_standby, charger_charging"""
|
|
if self.use_interrupt:
|
|
inputs = struct.unpack("<H", self._rx_buf)[0]
|
|
else:
|
|
inputs = self._read("<H", _EXPANDER_REG_INPUTS, 2)[0]
|
|
return tuple([bool(int(digit)) for digit in "{:016b}".format(inputs)[4:]])
|
|
|
|
@property
|
|
def lcd_brightness(self) -> int:
|
|
"""Read the LCD brightness state (0-100)"""
|
|
return self._read("<H", _EXPANDER_REG_LCD_BRIGHTNESS, 2)[0]
|
|
|
|
@lcd_brightness.setter
|
|
def lcd_brightness(self, value: int):
|
|
"""Set the LCD brightness (0-100)"""
|
|
if value >= 0 and value <= 100:
|
|
self._write(_EXPANDER_REG_LCD_BRIGHTNESS, struct.pack("<H", value))
|
|
|
|
@property
|
|
def debug_led(self) -> int:
|
|
"""Read the Debug LED state (0-100)"""
|
|
return self._read("<H", _EXPANDER_REG_DEBUG_LED, 2)[0]
|
|
|
|
@debug_led.setter
|
|
def debug_led(self, value: int):
|
|
"""Set the Debug LED (0-100)"""
|
|
if value >= 0 and value <= 100:
|
|
self._write(_EXPANDER_REG_DEBUG_LED, struct.pack("<H", value))
|
|
|
|
@property
|
|
def config(self) -> tuple[bool, bool, bool]:
|
|
"""Read the configuration bits: reboot, lcd_reset, aux_power"""
|
|
config = self._read("B", _EXPANDER_REG_CONFIG, 1)[0]
|
|
return tuple([bool(int(digit)) for digit in "{:08b}".format(config)[5:]])
|
|
|
|
@config.setter
|
|
def config(self, value: int):
|
|
"""set the configuration byte"""
|
|
if value >= 0 and value <= 0xFF:
|
|
self._write(_EXPANDER_REG_CONFIG, struct.pack("B", value))
|