Moar pins

[Adafruit_Blinka-hackapet.git] / src / adafruit_blinka / microcontroller / pico_u2if / pico_u2if.py

diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py b/src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py
index 0a1b596d7df4e42b8c3b5327bc0e381b3ee6f9bc..687bfc033585b4212b5bd44945fee65c4140461f 100644 (file)
--- a/src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py
+++ b/src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py
@@ -1,8 +1,16 @@
 """Chip Definition for Pico with u2if firmware"""
 # https://github.com/execuc/u2if
 
 """Chip Definition for Pico with u2if firmware"""
 # https://github.com/execuc/u2if
 

+import os
+import time

 import hid
 
 import hid
 

+# Use to set delay between reset and device reopen. if negative, don't reset at all
+PICO_U2IF_RESET_DELAY = float(os.environ.get("PICO_U2IF_RESET_DELAY", 1))
+
+# 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"""
 
 class Pico_u2if:
     """MCP2221 Device Class Definition"""


@@ -12,6 +20,7 @@ class Pico_u2if:
 
     # MISC
     RESP_OK = 0x01
 
     # MISC
     RESP_OK = 0x01

+    SYS_RESET = 0x10

 
     # GPIO
     GPIO_INIT_PIN = 0x20
 
     # GPIO
     GPIO_INIT_PIN = 0x20


@@ -61,15 +70,11 @@ class Pico_u2if:
     PWM_SET_DUTY_NS = 0x36
     PWM_GET_DUTY_NS = 0x37
 
     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)
     def __init__(self):
         self._hid = hid.device()
         self._hid.open(Pico_u2if.VID, Pico_u2if.PID)

+        if PICO_U2IF_RESET_DELAY >= 0:
+            self._reset()

         self._i2c_index = None
         self._spi_index = None
         self._serial = None
         self._i2c_index = None
         self._spi_index = None
         self._serial = None


@@ -87,6 +92,19 @@ class Pico_u2if:
             return self._hid.read(64)
         return None
 
             return self._hid.read(64)
         return None
 

+    def _reset(self):
+        self._hid_xfer(bytes([self.SYS_RESET]), False)
+        time.sleep(PICO_U2IF_RESET_DELAY)
+        start = time.monotonic()
+        while time.monotonic() - start < 5:
+            try:
+                self._hid.open(Pico_u2if.VID, Pico_u2if.PID)
+            except OSError:
+                time.sleep(0.1)
+                continue
+            return
+        raise OSError("Pico open error.")
+

     # ----------------------------------------------------------------
     # GPIO
     # ----------------------------------------------------------------
     # ----------------------------------------------------------------
     # GPIO
     # ----------------------------------------------------------------


@@ -360,6 +378,7 @@ class Pico_u2if:
         # init
         if not self._neopixel_initialized:
             # deinit any current setup
         # 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(
             self._hid_xfer(bytes([self.WS2812B_DEINIT]))
             resp = self._hid_xfer(
                 bytes(


@@ -374,6 +393,8 @@ class Pico_u2if:
                 raise RuntimeError("Neopixel init error")
             self._neopixel_initialized = True
 
                 raise RuntimeError("Neopixel init error")
             self._neopixel_initialized = True
 

+        self._serial.reset_output_buffer()
+

         # write
         # command is done over HID
         remain_bytes = len(buf)
         # write
         # command is done over HID
         remain_bytes = len(buf)


@@ -382,23 +403,35 @@ class Pico_u2if:
             True,
         )
         if resp[1] != self.RESP_OK:
             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."
                 )
             elif resp[2] == 0x02:
             if resp[2] == 0x01:
                 raise RuntimeError(
                     "Neopixel write error : too many pixel for the firmware."
                 )
             elif resp[2] == 0x02:

+                print(resp[0:10])

                 raise RuntimeError(
                     "Neopixel write error : transfer already in progress."
                 )
             else:
                 raise RuntimeError(
                     "Neopixel write error : transfer already in progress."
                 )
             else:

-                raise RuntimeError("Neopixel write error")
+                raise RuntimeError("Neopixel write error.")

         # buffer is sent over serial
         self._serial.write(buf)
         # buffer is sent over serial
         self._serial.write(buf)

+        # hack (see u2if)
+        if len(buf) % 64 == 0:
+            self._serial.write([0])

         self._serial.flush()
         self._serial.flush()

+        # polling loop to wait for write complete?
+        resp = self._hid.read(64)
+        while resp[0] != self.WS2812B_WRITE:
+            resp = self._hid.read(64)
+        if resp[1] != self.RESP_OK:
+            raise RuntimeError("Neopixel write (flush) error.")

 
     # ----------------------------------------------------------------
     # PWM
     # ----------------------------------------------------------------
 
     # ----------------------------------------------------------------
     # PWM
     # ----------------------------------------------------------------

+    # pylint: disable=unused-argument

     def pwm_configure(self, pin, frequency=500, duty_cycle=0, variable_frequency=False):
         """Configure PWM."""
         self.pwm_deinit(pin)
     def pwm_configure(self, pin, frequency=500, duty_cycle=0, variable_frequency=False):
         """Configure PWM."""
         self.pwm_deinit(pin)


@@ -415,11 +448,7 @@ class Pico_u2if:
 
     def pwm_get_frequency(self, pin):
         """PWM get freq."""
 
     def pwm_get_frequency(self, pin):
         """PWM get freq."""

-        resp = self._hid_xfer(
-            bytes([self.PWM_GET_FREQ, pin.id])
-            + frequency.to_bytes(4, byteorder="little"),
-            True,
-        )
+        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")
         if resp[1] != self.RESP_OK:
             raise RuntimeError("PWM get frequency error.")
         return int.from_bytes(resp[3 : 3 + 4], byteorder="little")


@@ -432,6 +461,7 @@ class Pico_u2if:
             True,
         )
         if resp[1] != self.RESP_OK:
             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:
             if resp[3] == 0x01:
                 raise RuntimeError("PWM different frequency on same slice.")
             elif resp[3] == 0x02: