import hid
+# pylint: disable=import-outside-toplevel,too-many-branches,too-many-statements
+# pylint: disable=too-many-arguments,too-many-function-args, too-many-public-methods
+
class Pico_u2if:
"""MCP2221 Device Class Definition"""
PWM_SET_DUTY_NS = 0x36
PWM_GET_DUTY_NS = 0x37
- # UART
- UART0_INIT = 0x50
- UART0_DEINIT = 0x51
- UART0_WRITE = 0x52
- UART0_READ = 0x53
-
def __init__(self):
self._hid = hid.device()
self._hid.open(Pico_u2if.VID, Pico_u2if.PID)
# I2C
# ----------------------------------------------------------------
def i2c_configure(self, baudrate, pullup=False):
+ """Configure I2C."""
if self._i2c_index is None:
raise RuntimeError("I2C bus not initialized.")
raise RuntimeError("I2C init error.")
def i2c_set_port(self, index):
+ """Set I2C port."""
if index not in (0, 1):
raise ValueError("I2C index must be 0 or 1.")
self._i2c_index = index
# SPI
# ----------------------------------------------------------------
def spi_configure(self, baudrate):
+ """Configure SPI."""
if self._spi_index is None:
raise RuntimeError("SPI bus not initialized.")
raise RuntimeError("SPI init error.")
def spi_set_port(self, index):
+ """Set SPI port."""
if index not in (0, 1):
raise ValueError("SPI index must be 0 or 1.")
self._spi_index = index
def spi_write(self, buffer, *, start=0, end=None):
+ """SPI write."""
if self._spi_index is None:
raise RuntimeError("SPI bus not initialized.")
start += chunk
def spi_readinto(self, buffer, *, start=0, end=None, write_value=0):
+ """SPI readinto."""
if self._spi_index is None:
raise RuntimeError("SPI bus not initialized.")
in_start=0,
in_end=None
):
+ """SPI write and readinto."""
raise NotImplementedError("SPI write_readinto Not implemented")
# ----------------------------------------------------------------
# NEOPIXEL
# ----------------------------------------------------------------
def neopixel_write(self, gpio, buf):
+ """NeoPixel write."""
# open serial (data is sent over this)
if self._serial is None:
import serial
# init
if not self._neopixel_initialized:
# deinit any current setup
+ # pylint: disable=protected-access
self._hid_xfer(bytes([self.WS2812B_DEINIT]))
resp = self._hid_xfer(
bytes(
True,
)
if resp[1] != self.RESP_OK:
+ # pylint: disable=no-else-raise
if resp[2] == 0x01:
raise RuntimeError(
"Neopixel write error : too many pixel for the firmware."
# ----------------------------------------------------------------
# PWM
# ----------------------------------------------------------------
+ # pylint: disable=unused-argument
def pwm_configure(self, pin, frequency=500, duty_cycle=0, variable_frequency=False):
+ """Configure PWM."""
self.pwm_deinit(pin)
resp = self._hid_xfer(bytes([self.PWM_INIT_PIN, pin.id]), True)
if resp[1] != self.RESP_OK:
self.pwm_set_duty_cycle(pin, duty_cycle)
def pwm_deinit(self, pin):
+ """Deinit PWM."""
self._hid_xfer(bytes([self.PWM_DEINIT_PIN, pin.id]))
def pwm_get_frequency(self, pin):
- resp = self._hid_xfer(
- bytes([self.PWM_GET_FREQ, pin.id])
- + frequency.to_bytes(4, byteorder="little"),
- True,
- )
+ """PWM get freq."""
+ resp = self._hid_xfer(bytes([self.PWM_GET_FREQ, pin.id]), True)
if resp[1] != self.RESP_OK:
raise RuntimeError("PWM get frequency error.")
return int.from_bytes(resp[3 : 3 + 4], byteorder="little")
def pwm_set_frequency(self, pin, frequency):
+ """PWM set freq."""
resp = self._hid_xfer(
bytes([self.PWM_SET_FREQ, pin.id])
+ frequency.to_bytes(4, byteorder="little"),
True,
)
if resp[1] != self.RESP_OK:
+ # pylint: disable=no-else-raise
if resp[3] == 0x01:
raise RuntimeError("PWM different frequency on same slice.")
elif resp[3] == 0x02:
raise RuntimeError("PWM frequency error.")
def pwm_get_duty_cycle(self, pin):
+ """PWM get duty cycle."""
resp = self._hid_xfer(bytes([self.PWM_GET_DUTY_U16, pin.id]), True)
if resp[1] != self.RESP_OK:
raise RuntimeError("PWM get duty cycle error.")
return int.from_bytes(resp[3 : 3 + 4], byteorder="little")
def pwm_set_duty_cycle(self, pin, duty_cycle):
+ """PWM set duty cycle."""
resp = self._hid_xfer(
bytes([self.PWM_SET_DUTY_U16, pin.id])
+ duty_cycle.to_bytes(2, byteorder="little"),