X-Git-Url: https://git.ayoreis.com/Adafruit_Blinka-hackapet.git/blobdiff_plain/dbfeb3b7e1dad621914a5fc580492a4d877ce24b..50ee4bd432cd68a53e351dc45d43ce1513e852f0:/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 9c99496..f5f3945 100644
--- a/src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py
+++ b/src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py
@@ -1,6 +1,7 @@
 """Chip Definition for Pico with u2if firmware"""
 # https://github.com/execuc/u2if
 
+import time
 import hid
 
 # pylint: disable=import-outside-toplevel,too-many-branches,too-many-statements
@@ -15,6 +16,7 @@ class Pico_u2if:
 
     # MISC
     RESP_OK = 0x01
+    SYS_RESET = 0x10
 
     # GPIO
     GPIO_INIT_PIN = 0x20
@@ -64,20 +66,13 @@ class Pico_u2if:
     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)
         self._i2c_index = None
         self._spi_index = None
         self._serial = None
         self._neopixel_initialized = False
         self._uart_rx_buffer = None
+        self._reset()
 
     def _hid_xfer(self, report, response=True):
         """Perform HID Transfer"""
@@ -90,6 +85,28 @@ class Pico_u2if:
             return self._hid.read(64)
         return None
 
+    def _reset(self):
+        # get a HID device
+        self._hid = hid.device()
+        # open and reset
+        self._hid.open(Pico_u2if.VID, Pico_u2if.PID)
+        resp = self._hid_xfer(bytes([self.SYS_RESET]), True)
+        if resp[1] != self.RESP_OK:
+            raise RuntimeError("Reset error.")
+        # reopen
+        max_retry = 10
+        retries = 0
+        while True:
+            try:
+                self._hid.open(Pico_u2if.VID, Pico_u2if.PID)
+                return True
+            except OSError:
+                if retries >= max_retry:
+                    break
+                time.sleep(0.1)
+                retries += 1
+        return False
+
     # ----------------------------------------------------------------
     # GPIO
     # ----------------------------------------------------------------
@@ -363,6 +380,7 @@ class Pico_u2if:
         # 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(
@@ -403,7 +421,7 @@ class Pico_u2if:
     # ----------------------------------------------------------------
     # PWM
     # ----------------------------------------------------------------
-    # pylint: disable=unused_argument
+    # pylint: disable=unused-argument
     def pwm_configure(self, pin, frequency=500, duty_cycle=0, variable_frequency=False):
         """Configure PWM."""
         self.pwm_deinit(pin)