]> Repositories - Adafruit_Blinka-hackapet.git/commitdiff
add pico u2if
authorcaternuson <caternuson@gmail.com>
Sun, 25 Apr 2021 00:13:43 +0000 (17:13 -0700)
committercaternuson <caternuson@gmail.com>
Sun, 25 Apr 2021 00:13:43 +0000 (17:13 -0700)
16 files changed:
src/adafruit_blinka/board/pico_u2if.py [new file with mode: 0644]
src/adafruit_blinka/microcontroller/pico_u2if/__init__.py [new file with mode: 0644]
src/adafruit_blinka/microcontroller/pico_u2if/analogio.py [new file with mode: 0644]
src/adafruit_blinka/microcontroller/pico_u2if/i2c.py [new file with mode: 0644]
src/adafruit_blinka/microcontroller/pico_u2if/neopixel.py [new file with mode: 0644]
src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py [new file with mode: 0644]
src/adafruit_blinka/microcontroller/pico_u2if/pin.py [new file with mode: 0644]
src/adafruit_blinka/microcontroller/pico_u2if/pwmio.py [new file with mode: 0644]
src/adafruit_blinka/microcontroller/pico_u2if/spi.py [new file with mode: 0644]
src/analogio.py
src/board.py
src/busio.py
src/digitalio.py
src/microcontroller/pin.py
src/neopixel_write.py
src/pwmio.py

diff --git a/src/adafruit_blinka/board/pico_u2if.py b/src/adafruit_blinka/board/pico_u2if.py
new file mode 100644 (file)
index 0000000..bcdfe47
--- /dev/null
@@ -0,0 +1,46 @@
+"""Pin definitions for the MicroChip MCP2221"""
+from adafruit_blinka.microcontroller.pico_u2if import pin
+
+GP0 = pin.GP0
+GP1 = pin.GP1
+GP2 = pin.GP2
+GP3 = pin.GP3
+GP4 = pin.GP4
+GP5 = pin.GP5
+GP6 = pin.GP6
+GP7 = pin.GP7
+GP8 = pin.GP8
+GP9 = pin.GP9
+GP10 = pin.GP10
+GP11 = pin.GP11
+GP12 = pin.GP12
+GP13 = pin.GP13
+GP14 = pin.GP14
+GP15 = pin.GP15
+GP16 = pin.GP16
+GP17 = pin.GP17
+GP18 = pin.GP18
+GP19 = pin.GP19
+GP20 = pin.GP20
+GP21 = pin.GP21
+GP22 = pin.GP22
+GP26 = pin.GP26
+GP27 = pin.GP27
+GP28 = pin.GP28
+
+ADC0 = GP26
+ADC1 = GP27
+
+SCL = SCL0 = GP5
+SDA = SDA0 = GP4
+
+SCL1 = GP15
+SDA1 = GP14
+
+SCLK = SCK = SCLK0 = SCK0 = GP18
+MOSI = MOSI0 = GP19
+MISO = MISO0 = GP12
+
+SCLK1 = SCK1 = GP10
+MOSI1 = GP11
+MISO1 = GP12
diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/__init__.py b/src/adafruit_blinka/microcontroller/pico_u2if/__init__.py
new file mode 100644 (file)
index 0000000..e69de29
diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/analogio.py b/src/adafruit_blinka/microcontroller/pico_u2if/analogio.py
new file mode 100644 (file)
index 0000000..6eb7ead
--- /dev/null
@@ -0,0 +1,35 @@
+"""
+`analogio` - Analog input and output control
+=================================================
+See `CircuitPython:analogio` in CircuitPython for more details.
+* Author(s): Carter Nelson
+"""
+from adafruit_blinka import ContextManaged
+from .pico_u2if import pico_u2if
+
+
+class AnalogIn(ContextManaged):
+    """Analog Input Class"""
+
+    def __init__(self, pin):
+        # per their pinout, why only two?
+        if pin.id not in (26, 27):
+            raise ValueError("Pin does not support ADC.")
+        self.pin_id = pin.id
+        pico_u2if.adc_init_pin(self.pin_id)
+
+    @property
+    def value(self):
+        """Read the ADC and return the value"""
+        return pico_u2if.adc_get_value(self.pin_id)
+
+    # pylint: disable=no-self-use
+    @value.setter
+    def value(self, value):
+        # emulate what CircuitPython does
+        raise AttributeError("'AnalogIn' object has no attribute 'value'")
+
+    # pylint: enable=no-self-use
+
+    def deinit(self):
+        pass
diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/i2c.py b/src/adafruit_blinka/microcontroller/pico_u2if/i2c.py
new file mode 100644 (file)
index 0000000..ee393bb
--- /dev/null
@@ -0,0 +1,62 @@
+"""I2C Class for Pico u2if"""
+from .pico_u2if import pico_u2if
+
+
+class I2C:
+    """Custom I2C Class for Pico u2if"""
+
+    def __init__(self, scl, sda, *, frequency=100000):
+        index = None
+        if scl.id == 5 and sda.id == 4:
+            index = 0
+        if scl.id == 15 and sda.id == 14:
+            index = 1
+        if index is None:
+            raise ValueError("I2C not found on specified pins.")
+        self._index = index
+        pico_u2if.i2c_set_port(self._index)
+        pico_u2if.i2c_configure(frequency)
+
+    def scan(self):
+        """Perform an I2C Device Scan"""
+        pico_u2if.i2c_set_port(self._index)
+        return pico_u2if.i2c_scan()
+
+    # pylint: disable=unused-argument
+    def writeto(self, address, buffer, *, start=0, end=None, stop=True):
+        """Write data from the buffer to an address"""
+        pico_u2if.i2c_set_port(self._index)
+        pico_u2if.i2c_writeto(address, buffer, start=start, end=end)
+
+    def readfrom_into(self, address, buffer, *, start=0, end=None, stop=True):
+        """Read data from an address and into the buffer"""
+        pico_u2if.i2c_set_port(self._index)
+        pico_u2if.i2c_readfrom_into(address, buffer, start=start, end=end)
+
+    def writeto_then_readfrom(
+        self,
+        address,
+        buffer_out,
+        buffer_in,
+        *,
+        out_start=0,
+        out_end=None,
+        in_start=0,
+        in_end=None,
+        stop=False
+    ):
+        """Write data from buffer_out to an address and then
+        read data from an address and into buffer_in
+        """
+        pico_u2if.i2c_set_port(self._index)
+        pico_u2if.i2c_writeto_then_readfrom(
+            address,
+            buffer_out,
+            buffer_in,
+            out_start=out_start,
+            out_end=out_end,
+            in_start=in_start,
+            in_end=in_end,
+        )
+
+    # pylint: enable=unused-argument
diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/neopixel.py b/src/adafruit_blinka/microcontroller/pico_u2if/neopixel.py
new file mode 100644 (file)
index 0000000..281bec4
--- /dev/null
@@ -0,0 +1,15 @@
+from .pico_u2if import pico_u2if
+
+
+def neopixel_write(gpio, buf):
+    """NeoPixel Writing Function"""
+
+    # pad output buffer from 3 bpp to 4 bpp
+    buffer = []
+    for i in range(0, len(buf), 3):
+        buffer.append(0)
+        buffer.append(buf[i + 2])
+        buffer.append(buf[i])
+        buffer.append(buf[i + 1])
+
+    pico_u2if.neopixel_write(gpio, buffer)
diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py b/src/adafruit_blinka/microcontroller/pico_u2if/pico_u2if.py
new file mode 100644 (file)
index 0000000..25ce962
--- /dev/null
@@ -0,0 +1,448 @@
+"""Chip Definition for Pico with u2if firmware"""
+# https://github.com/execuc/u2if
+
+import hid
+
+
+class Pico_u2if:
+    """MCP2221 Device Class Definition"""
+
+    VID = 0xCAFE
+    PID = 0x4005
+
+    # MISC
+    RESP_OK = 0x01
+
+    # GPIO
+    GPIO_INIT_PIN = 0x20
+    GPIO_SET_VALUE = 0x21
+    GPIO_GET_VALUE = 0x22
+
+    # ADC
+    ADC_INIT_PIN = 0x40
+    ADC_GET_VALUE = 0x41
+
+    # I2C
+    I2C0_INIT = 0x80
+    I2C0_DEINIT = 0x81
+    I2C0_WRITE = 0x82
+    I2C0_READ = 0x83
+    I2C0_WRITE_FROM_UART = 0x84
+    I2C1_INIT = I2C0_INIT + 0x10
+    I2C1_DEINIT = I2C0_DEINIT + 0x10
+    I2C1_WRITE = I2C0_WRITE + 0x10
+    I2C1_READ = I2C0_READ + 0x10
+    I2C1_WRITE_FROM_UART = I2C0_WRITE_FROM_UART + 0x10
+
+    # SPI
+    SPI0_INIT = 0x60
+    SPI0_DEINIT = 0x61
+    SPI0_WRITE = 0x62
+    SPI0_READ = 0x63
+    SPI0_WRITE_FROM_UART = 0x64
+    SPI1_INIT = SPI0_INIT + 0x10
+    SPI1_DEINIT = SPI0_DEINIT + 0x10
+    SPI1_WRITE = SPI0_WRITE + 0x10
+    SPI1_READ = SPI0_READ + 0x10
+    SPI1_WRITE_FROM_UART = SPI0_WRITE_FROM_UART + 0x10
+
+    # WS2812B (LED)
+    WS2812B_INIT = 0xA0
+    WS2812B_DEINIT = 0xA1
+    WS2812B_WRITE = 0xA2
+
+    # PWM
+    PWM_INIT_PIN = 0x30
+    PWM_DEINIT_PIN = 0x31
+    PWM_SET_FREQ = 0x32
+    PWM_GET_FREQ = 0x33
+    PWM_SET_DUTY_U16 = 0x34
+    PWM_GET_DUTY_U16 = 0x35
+    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
+
+    def _hid_xfer(self, report, response=True):
+        """Perform HID Transfer"""
+        # first byte is report ID, which =0
+        # remaing bytes = 64 byte report data
+        # https://github.com/libusb/hidapi/blob/083223e77952e1ef57e6b77796536a3359c1b2a3/hidapi/hidapi.h#L185
+        self._hid.write(b"\0" + report + b"\0" * (64 - len(report)))
+        if response:
+            # return is 64 byte response report
+            return self._hid.read(64)
+        return None
+
+    # ----------------------------------------------------------------
+    # GPIO
+    # ----------------------------------------------------------------
+    def gpio_init_pin(self, pin_id, direction, pull):
+        """Configure GPIO Pin."""
+        self._hid_xfer(
+            bytes(
+                [
+                    self.GPIO_INIT_PIN,
+                    pin_id,
+                    direction,
+                    pull,
+                ]
+            )
+        )
+
+    def gpio_set_pin(self, pin_id, value):
+        """Set Current GPIO Pin Value"""
+        self._hid_xfer(
+            bytes(
+                [
+                    self.GPIO_SET_VALUE,
+                    pin_id,
+                    int(value),
+                ]
+            )
+        )
+
+    def gpio_get_pin(self, pin_id):
+        """Get Current GPIO Pin Value"""
+        resp = self._hid_xfer(
+            bytes(
+                [
+                    self.GPIO_GET_VALUE,
+                    pin_id,
+                ]
+            ),
+            True,
+        )
+        return resp[3] != 0x00
+
+    # ----------------------------------------------------------------
+    # ADC
+    # ----------------------------------------------------------------
+    def adc_init_pin(self, pin_id):
+        """Configure ADC Pin."""
+        self._hid_xfer(
+            bytes(
+                [
+                    self.ADC_INIT_PIN,
+                    pin_id,
+                ]
+            )
+        )
+
+    def adc_get_value(self, pin_id):
+        """Get ADC value for pin."""
+        resp = self._hid_xfer(
+            bytes(
+                [
+                    self.ADC_GET_VALUE,
+                    pin_id,
+                ]
+            ),
+            True,
+        )
+        return int.from_bytes(resp[3 : 3 + 2], byteorder="little")
+
+    # ----------------------------------------------------------------
+    # I2C
+    # ----------------------------------------------------------------
+    def i2c_configure(self, baudrate, pullup=False):
+        if self._i2c_index is None:
+            raise RuntimeError("I2C bus not initialized.")
+
+        resp = self._hid_xfer(
+            bytes(
+                [
+                    self.I2C0_INIT if self._i2c_index == 0 else self.I2C1_INIT,
+                    0x00 if not pullup else 0x01,
+                ]
+            )
+            + baudrate.to_bytes(4, byteorder="little"),
+            True,
+        )
+        if resp[1] != self.RESP_OK:
+            raise RuntimeError("I2C init error.")
+
+    def i2c_set_port(self, index):
+        if index not in (0, 1):
+            raise ValueError("I2C index must be 0 or 1.")
+        self._i2c_index = index
+
+    def _i2c_write(self, address, buffer, start=0, end=None, stop=True):
+        """Write data from the buffer to an address"""
+        if self._i2c_index is None:
+            raise RuntimeError("I2C bus not initialized.")
+
+        end = end if end else len(buffer)
+
+        write_cmd = self.I2C0_WRITE if self._i2c_index == 0 else self.I2C1_WRITE
+        stop_flag = 0x01 if stop else 0x00
+
+        while (end - start) > 0:
+            remain_bytes = end - start
+            chunk = min(remain_bytes, 64 - 7)
+            resp = self._hid_xfer(
+                bytes([write_cmd, address, stop_flag])
+                + remain_bytes.to_bytes(4, byteorder="little")
+                + buffer[start : (start + chunk)],
+                True,
+            )
+            if resp[1] != self.RESP_OK:
+                raise RuntimeError("I2C write error")
+            start += chunk
+
+    def _i2c_read(self, address, buffer, start=0, end=None):
+        """Read data from an address and into the buffer"""
+        # TODO: support chunkified reads
+        if self._i2c_index is None:
+            raise RuntimeError("I2C bus not initialized.")
+
+        end = end if end else len(buffer)
+
+        read_cmd = self.I2C0_READ if self._i2c_index == 0 else self.I2C1_READ
+        stop_flag = 0x01  # always stop
+        read_size = end - start
+
+        resp = self._hid_xfer(bytes([read_cmd, address, stop_flag, read_size]), True)
+        if resp[1] != self.RESP_OK:
+            raise RuntimeError("I2C write error")
+        # move into buffer
+        for i in range(read_size):
+            buffer[start + i] = resp[i + 2]
+
+    def i2c_writeto(self, address, buffer, *, start=0, end=None):
+        """Write data from the buffer to an address"""
+        self._i2c_write(address, buffer, start, end)
+
+    def i2c_readfrom_into(self, address, buffer, *, start=0, end=None):
+        """Read data from an address and into the buffer"""
+        self._i2c_read(address, buffer, start, end)
+
+    def i2c_writeto_then_readfrom(
+        self,
+        address,
+        out_buffer,
+        in_buffer,
+        *,
+        out_start=0,
+        out_end=None,
+        in_start=0,
+        in_end=None
+    ):
+        """Write data from buffer_out to an address and then
+        read data from an address and into buffer_in
+        """
+        self._i2c_write(address, out_buffer, out_start, out_end, False)
+        self._i2c_read(address, in_buffer, in_start, in_end)
+
+    def i2c_scan(self, *, start=0, end=0x79):
+        """Perform an I2C Device Scan"""
+        if self._i2c_index is None:
+            raise RuntimeError("I2C bus not initialized.")
+        found = []
+        for addr in range(start, end + 1):
+            # try a write
+            try:
+                self.i2c_writeto(addr, b"\x00\x00\x00")
+            except RuntimeError:  # no reply!
+                continue
+            # store if success
+            found.append(addr)
+        return found
+
+    # ----------------------------------------------------------------
+    # SPI
+    # ----------------------------------------------------------------
+    def spi_configure(self, baudrate):
+        if self._spi_index is None:
+            raise RuntimeError("SPI bus not initialized.")
+
+        resp = self._hid_xfer(
+            bytes(
+                [
+                    self.SPI0_INIT if self._spi_index == 0 else self.SPI1_INIT,
+                    0x00,  # mode, not yet implemented
+                ]
+            )
+            + baudrate.to_bytes(4, byteorder="little"),
+            True,
+        )
+        if resp[1] != self.RESP_OK:
+            raise RuntimeError("SPI init error.")
+
+    def spi_set_port(self, index):
+        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):
+        if self._spi_index is None:
+            raise RuntimeError("SPI bus not initialized.")
+
+        end = end if end else len(buffer)
+
+        write_cmd = self.SPI0_WRITE if self._spi_index == 0 else self.SPI1_WRITE
+
+        while (end - start) > 0:
+            remain_bytes = end - start
+            chunk = min(remain_bytes, 64 - 3)
+            resp = self._hid_xfer(
+                bytes([write_cmd, chunk]) + buffer[start : (start + chunk)], True
+            )
+            if resp[1] != self.RESP_OK:
+                raise RuntimeError("SPI write error")
+            start += chunk
+
+    def spi_readinto(self, buffer, *, start=0, end=None, write_value=0):
+        if self._spi_index is None:
+            raise RuntimeError("SPI bus not initialized.")
+
+        end = end if end else len(buffer)
+        read_cmd = self.SPI0_READ if self._spi_index == 0 else self.SPI1_READ
+        read_size = end - start
+
+        resp = self._hid_xfer(bytes([read_cmd, write_value, read_size]), True)
+        if resp[1] != self.RESP_OK:
+            raise RuntimeError("SPI write error")
+        # move into buffer
+        for i in range(read_size):
+            buffer[start + i] = resp[i + 2]
+
+    def spi_write_readinto(
+        self,
+        buffer_out,
+        buffer_in,
+        *,
+        out_start=0,
+        out_end=None,
+        in_start=0,
+        in_end=None
+    ):
+        raise NotImplementedError("SPI write_readinto Not implemented")
+
+    # ----------------------------------------------------------------
+    # NEOPIXEL
+    # ----------------------------------------------------------------
+    def neopixel_write(self, gpio, buf):
+        # open serial (data is sent over this)
+        if self._serial is None:
+            import serial
+            import serial.tools.list_ports
+
+            ports = serial.tools.list_ports.comports()
+            for port in ports:
+                if port.vid == self.VID and port.pid == self.PID:
+                    self._serial = serial.Serial(port.device)
+                    break
+        if self._serial is None:
+            raise RuntimeError("Could not find Pico com port.")
+
+        # init
+        if not self._neopixel_initialized:
+            # deinit any current setup
+            self._hid_xfer(bytes([self.WS2812B_DEINIT]))
+            resp = self._hid_xfer(
+                bytes(
+                    [
+                        self.WS2812B_INIT,
+                        gpio._pin.id,
+                    ]
+                ),
+                True,
+            )
+            if resp[1] != self.RESP_OK:
+                raise RuntimeError("Neopixel init error")
+            self._neopixel_initialized = True
+
+        # write
+        # command is done over HID
+        remain_bytes = len(buf)
+        resp = self._hid_xfer(
+            bytes([self.WS2812B_WRITE]) + remain_bytes.to_bytes(4, byteorder="little"),
+            True,
+        )
+        if resp[1] != self.RESP_OK:
+            if resp[2] == 0x01:
+                raise RuntimeError(
+                    "Neopixel write error : too many pixel for the firmware."
+                )
+            elif resp[2] == 0x02:
+                raise RuntimeError(
+                    "Neopixel write error : transfer already in progress."
+                )
+            else:
+                raise RuntimeError("Neopixel write error")
+        # buffer is sent over serial
+        self._serial.write(buf)
+        self._serial.flush()
+
+    # ----------------------------------------------------------------
+    # PWM
+    # ----------------------------------------------------------------
+    def pwm_configure(self, pin, frequency=500, duty_cycle=0, variable_frequency=False):
+        self.pwm_deinit(pin)
+        resp = self._hid_xfer(bytes([self.PWM_INIT_PIN, pin.id]), True)
+        if resp[1] != self.RESP_OK:
+            raise RuntimeError("PWM init error.")
+
+        self.pwm_set_frequency(pin, frequency)
+        self.pwm_set_duty_cycle(pin, duty_cycle)
+
+    def pwm_deinit(self, pin):
+        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,
+        )
+        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):
+        resp = self._hid_xfer(
+            bytes([self.PWM_SET_FREQ, pin.id])
+            + frequency.to_bytes(4, byteorder="little"),
+            True,
+        )
+        if resp[1] != self.RESP_OK:
+            if resp[3] == 0x01:
+                raise RuntimeError("PWM different frequency on same slice.")
+            elif resp[3] == 0x02:
+                raise RuntimeError("PWM frequency too low.")
+            elif resp[3] == 0x03:
+                raise RuntimeError("PWM frequency too high.")
+            else:
+                raise RuntimeError("PWM frequency error.")
+
+    def pwm_get_duty_cycle(self, pin):
+        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):
+        resp = self._hid_xfer(
+            bytes([self.PWM_SET_DUTY_U16, pin.id])
+            + duty_cycle.to_bytes(2, byteorder="little"),
+            True,
+        )
+        if resp[1] != self.RESP_OK:
+            raise RuntimeError("PWM set duty cycle error.")
+
+
+pico_u2if = Pico_u2if()
diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/pin.py b/src/adafruit_blinka/microcontroller/pico_u2if/pin.py
new file mode 100644 (file)
index 0000000..ef22587
--- /dev/null
@@ -0,0 +1,84 @@
+"""PICO pin names"""
+from .pico_u2if import pico_u2if
+
+
+class Pin:
+    """A basic Pin class for use with MCP2221."""
+
+    # pin modes
+    IN = 0
+    OUT = 1
+    # pin values
+    LOW = 0
+    HIGH = 1
+    # pin pulls
+    PULL_NONE = 0
+    PULL_UP = 1
+    PULL_DOWN = 2
+
+    def __init__(self, pin_id=None):
+        self.id = pin_id
+        self._mode = None
+        self._pull = None
+
+    def init(self, mode=IN, pull=PULL_NONE):
+        """Initialize the Pin"""
+        pull = Pin.PULL_NONE if pull is None else pull
+        if self.id is None:
+            raise RuntimeError("Can not init a None type pin.")
+        if mode not in (Pin.IN, Pin.OUT):
+            raise ValueError("Incorrect mode value.")
+        if pull not in (Pin.PULL_NONE, Pin.PULL_UP, Pin.PULL_DOWN):
+            raise ValueError("Incorrect pull value.")
+
+        pico_u2if.gpio_init_pin(self.id, mode, pull)
+
+        self._mode = mode
+        self._pull = pull
+
+    def value(self, val=None):
+        """Set or return the Pin Value"""
+        # Digital In / Out
+        if self._mode in (Pin.IN, Pin.OUT):
+            # digital read
+            if val is None:
+                return pico_u2if.gpio_get_pin(self.id)
+            # digital write
+            if val in (Pin.LOW, Pin.HIGH):
+                pico_u2if.gpio_set_pin(self.id, val)
+                return None
+            # nope
+            raise ValueError("Invalid value for pin.")
+
+        raise RuntimeError(
+            "No action for mode {} with value {}".format(self._mode, val)
+        )
+
+
+# create pin instances for each pin
+GP0 = Pin(0)
+GP1 = Pin(1)
+GP2 = Pin(2)
+GP3 = Pin(3)
+GP4 = Pin(4)
+GP5 = Pin(5)
+GP6 = Pin(6)
+GP7 = Pin(7)
+GP8 = Pin(8)
+GP9 = Pin(9)
+GP10 = Pin(10)
+GP11 = Pin(11)
+GP12 = Pin(12)
+GP13 = Pin(13)
+GP14 = Pin(14)
+GP15 = Pin(15)
+GP16 = Pin(16)
+GP17 = Pin(17)
+GP18 = Pin(18)
+GP19 = Pin(19)
+GP20 = Pin(20)
+GP21 = Pin(21)
+GP22 = Pin(22)
+GP26 = Pin(26)
+GP27 = Pin(27)
+GP28 = Pin(28)
diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/pwmio.py b/src/adafruit_blinka/microcontroller/pico_u2if/pwmio.py
new file mode 100644 (file)
index 0000000..f849ae0
--- /dev/null
@@ -0,0 +1,46 @@
+"""PWMOut Class for Pico u2if"""
+from .pico_u2if import pico_u2if
+
+
+class PWMOut:
+    """Pulse Width Modulation Output Class"""
+
+    def __init__(self, pin, *, frequency=500, duty_cycle=0, variable_frequency=False):
+        pico_u2if.pwm_configure(
+            pin,
+            frequency=frequency,
+            duty_cycle=duty_cycle,
+            variable_frequency=variable_frequency,
+        )
+
+        self._pin = pin
+
+    def __del__(self):
+        self.deinit()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, t, value, traceback):
+        self.deinit()
+
+    def deinit(self):
+        pico_u2if.pwm_deinit(self._pin)
+
+    @property
+    def duty_cycle(self):
+        """The PWM's output duty cycle, 16-bit."""
+        return pico_u2if.pwm_get_duty_cycle(self._pin)
+
+    @duty_cycle.setter
+    def duty_cycle(self, duty_cycle):
+        pico_u2if.pwm_set_duty_cycle(self._pin, duty_cycle)
+
+    @property
+    def frequency(self):
+        """The PWM's output frequency in Hertz."""
+        return pico_u2if.pwm_get_frequency(self._pin)
+
+    @frequency.setter
+    def frequency(self, frequency):
+        pico_u2if.pwm_set_frequency(self._pin, frequency)
diff --git a/src/adafruit_blinka/microcontroller/pico_u2if/spi.py b/src/adafruit_blinka/microcontroller/pico_u2if/spi.py
new file mode 100644 (file)
index 0000000..7ce2158
--- /dev/null
@@ -0,0 +1,71 @@
+"""SPI Class for Pico u2if"""
+from .pico_u2if import pico_u2if
+
+# pylint: disable=protected-access
+class SPI:
+    """Custom SPI Class for Pico u2if"""
+
+    MSB = 0
+
+    def __init__(self, clock, *, baudrate=100000):
+        index = None
+        if clock.id == 18:
+            index = 0
+        if clock.id == 10:
+            index = 1
+        if index is None:
+            raise ValueError("No SPI port on specified pin.")
+        self._index = index
+        self._frequency = baudrate
+        pico_u2if.spi_set_port(self._index)
+        pico_u2if.spi_configure(self._frequency)
+
+    # pylint: disable=too-many-arguments,unused-argument
+    def init(
+        self,
+        baudrate=1000000,
+        polarity=0,
+        phase=0,
+        bits=8,
+        firstbit=MSB,
+        sck=None,
+        mosi=None,
+        miso=None,
+    ):
+        """Initialize the Port"""
+        self._frequency = baudrate
+        pico_u2if.spi_set_port(self._index)
+        pico_u2if.spi_configure(self._frequency)
+
+    # pylint: enable=too-many-arguments
+
+    @property
+    def frequency(self):
+        """Return the current frequency"""
+        return self._frequency
+
+    def write(self, buf, start=0, end=None):
+        """Write data from the buffer to SPI"""
+        pico_u2if.spi_write(buf, start=start, end=end)
+
+    def readinto(self, buf, start=0, end=None, write_value=0):
+        """Read data from SPI and into the buffer"""
+        pico_u2if.spi_readinto(buf, start=start, end=end, write_value=write_value)
+
+    # pylint: disable=too-many-arguments
+    def write_readinto(
+        self, buffer_out, buffer_in, out_start=0, out_end=None, in_start=0, in_end=None
+    ):
+        """Perform a half-duplex write from buffer_out and then
+        read data into buffer_in
+        """
+        pico_u2if.spi_write_readinto(
+            buffer_out,
+            buffer_in,
+            out_start=out_start,
+            out_end=out_end,
+            in_start=in_start,
+            in_end=in_end,
+        )
+
+    # pylint: enable=too-many-arguments
index 7c393c3709526b1bca67c0a7e050e7fcd37c817a..04d3afce8494afb0b38f47a45a54cfc964acb8d1 100644 (file)
@@ -29,5 +29,7 @@ elif detector.chip.STM32MP157:
     from adafruit_blinka.microcontroller.generic_linux.sysfs_analogin import AnalogIn
 elif "sphinx" in sys.modules:
     pass
+elif detector.board.pico_u2if:
+    from adafruit_blinka.microcontroller.pico_u2if.analogio import AnalogIn
 else:
     raise NotImplementedError("analogio not supported for this board.")
index 3303bb012d85bd392d774caaf98080999f9abb3c..45eaed6c3ac79596cd4b053ac555ebd191c493b2 100755 (executable)
@@ -215,6 +215,9 @@ elif board_id == ap_board.NANOPI_NEO_AIR:
 elif board_id == ap_board.NANOPI_DUO2:
     from adafruit_blinka.board.nanopi.duo2 import *
 
+elif board_id == ap_board.PICO_U2IF:
+    from adafruit_blinka.board.pico_u2if import *
+
 elif "sphinx" in sys.modules:
     pass
 
index 67494592e6592ea7928289fe94f93e726e47f02c..a11957482fd7135804e70cb2deb0b5cd04d7d489 100755 (executable)
@@ -53,6 +53,11 @@ class I2C(Lockable):
 
             self._i2c = _I2C(frequency=frequency)
             return
+        if detector.board.pico_u2if:
+            from adafruit_blinka.microcontroller.pico_u2if.i2c import I2C as _I2C
+
+            self._i2c = _I2C(scl, sda, frequency=frequency)
+            return
         if detector.board.any_embedded_linux:
             from adafruit_blinka.microcontroller.generic_linux.i2c import I2C as _I2C
         else:
@@ -171,6 +176,12 @@ class SPI(Lockable):
             self._spi = _SPI()
             self._pins = (SCK, MOSI, MISO)
             return
+        if detector.board.pico_u2if:
+            from adafruit_blinka.microcontroller.pico_u2if.spi import SPI as _SPI
+
+            self._spi = _SPI(clock)            # this is really all that's needed
+            self._pins = (clock, clock, clock) # will determine MOSI/MISO from clock
+            return
         if detector.board.any_embedded_linux:
             from adafruit_blinka.microcontroller.generic_linux.spi import SPI as _SPI
         else:
@@ -286,6 +297,9 @@ class SPI(Lockable):
         elif detector.board.any_lubancat and detector.chip.id == ap_chip.IMX6ULL:
             from adafruit_blinka.microcontroller.nxp_imx6ull.pin import Pin
             from adafruit_blinka.microcontroller.generic_linux.spi import SPI as _SPI
+        elif detector.board.pico_u2if:
+            from adafruit_blinka.microcontroller.pico_u2if.spi import SPI as _SPI
+            from adafruit_blinka.microcontroller.pico_u2if.pin import Pin
         else:
             from machine import SPI as _SPI
             from machine import Pin
index a59866b9a74eb10270f775983156f22da5ae9a73..361663b04a3ab8c3890c187245d9d6459933cfd7 100755 (executable)
@@ -73,6 +73,8 @@ elif detector.chip.H5:
     from adafruit_blinka.microcontroller.allwinner.h5.pin import Pin
 elif detector.chip.H616:
     from adafruit_blinka.microcontroller.allwinner.h616.pin import Pin
+elif detector.board.pico_u2if:
+    from adafruit_blinka.microcontroller.pico_u2if.pin import Pin
 
 from adafruit_blinka import Enum, ContextManaged
 
index 88c2d00b33c786af09b517362ded3ca04c382664..f69d30e7aef0ddb76c33df7d9cab0b1aff3b6876 100755 (executable)
@@ -72,5 +72,7 @@ elif chip_id == ap_chip.STM32MP157:
     from adafruit_blinka.microcontroller.stm32.stm32mp157.pin import *
 elif chip_id == ap_chip.MT8167:
     from adafruit_blinka.microcontroller.mt8167.pin import *
+elif chip_id == ap_chip.PICO_U2IF:
+    from adafruit_blinka.microcontroller.pico_u2if.pin import *
 else:
     raise NotImplementedError("Microcontroller not supported: ", chip_id)
index 2c85f0166ff1b79555e6127563b6c7f78155bb16..00ffa0b8df09026c32605ddf4f78d2a089832ba1 100644 (file)
@@ -14,6 +14,8 @@ from adafruit_blinka.agnostic import detector
 
 if detector.board.any_raspberry_pi:
     from adafruit_blinka.microcontroller.bcm283x import neopixel as _neopixel
+elif detector.board.pico_u2if:
+    from adafruit_blinka.microcontroller.pico_u2if import neopixel as _neopixel
 elif "sphinx" in sys.modules:
     pass
 else:
index 3a6b5bdc01e7a7d4291f9d55fe8a405302b82b35..dbb8bb79777c6ff3a4602c2fd2bd0bb1fffb2037 100644 (file)
@@ -29,6 +29,8 @@ elif detector.board.greatfet_one:
     from adafruit_blinka.microcontroller.nxp_lpc4330.pwmout import PWMOut
 elif detector.board.any_lubancat:
     from adafruit_blinka.microcontroller.generic_linux.sysfs_pwmout import PWMOut
+elif detector.board.pico_u2if:
+    from adafruit_blinka.microcontroller.pico_u2if.pwmio import PWMOut
 elif "sphinx" in sys.modules:
     pass
 else: