Initial shot at is31fl3731

drivers/CMakeLists.txt

@@ -1,2 +1,3 @@
 add_subdirectory(st7789)
-add_subdirectory(msa301)
+add_subdirectory(msa301)
+add_subdirectory(is31fl3731)

drivers/is31fl3731/CMakeLists.txt

@@ -0,0 +1,10 @@
+add_library(is31fl3731 INTERFACE)
+
+target_sources(is31fl3731 INTERFACE
+  ${CMAKE_CURRENT_LIST_DIR}/is31fl3731.cpp
+)
+
+target_include_directories(is31fl3731 INTERFACE ${CMAKE_CURRENT_LIST_DIR})
+
+# Pull in pico libraries that we need
+target_link_libraries(is31fl3731 INTERFACE pico_stdlib hardware_i2c)

drivers/is31fl3731/is31fl3731.cpp

@@ -0,0 +1,115 @@
+#include <cstdlib>
+#include <math.h>
+#include <map>
+#include <vector>
+
+#include "is31fl3731.hpp"
+
+namespace pimoroni {
+
+  constexpr uint8_t CONFIG_BANK = 0x0b;
+  constexpr uint8_t NUM_PIXELS = 144;
+  constexpr uint8_t NUM_FRAMES = 8;
+
+  constexpr uint8_t ENABLE_OFFSET = 0x00;
+  constexpr uint8_t COLOR_OFFSET = 0x24;
+
+  constexpr uint8_t GAMMA[256] = {
+    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
+    0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2,
+    2, 2, 2, 3, 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5,
+    6, 6, 6, 7, 7, 7, 8, 8, 8, 9, 9, 9, 10, 10, 11, 11,
+    11, 12, 12, 13, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18,
+    19, 19, 20, 21, 21, 22, 22, 23, 23, 24, 25, 25, 26, 27, 27, 28,
+    29, 29, 30, 31, 31, 32, 33, 34, 34, 35, 36, 37, 37, 38, 39, 40,
+    40, 41, 42, 43, 44, 45, 46, 46, 47, 48, 49, 50, 51, 52, 53, 54,
+    55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70,
+    71, 72, 73, 74, 76, 77, 78, 79, 80, 81, 83, 84, 85, 86, 88, 89,
+    90, 91, 93, 94, 95, 96, 98, 99, 100, 102, 103, 104, 106, 107, 109, 110,
+    111, 113, 114, 116, 117, 119, 120, 121, 123, 124, 126, 128, 129, 131, 132, 134,
+    135, 137, 138, 140, 142, 143, 145, 146, 148, 150, 151, 153, 155, 157, 158, 160,
+    162, 163, 165, 167, 169, 170, 172, 174, 176, 178, 179, 181, 183, 185, 187, 189,
+    191, 193, 194, 196, 198, 200, 202, 204, 206, 208, 210, 212, 214, 216, 218, 220,
+    222, 224, 227, 229, 231, 233, 235, 237, 239, 241, 244, 246, 248, 250, 252, 255};
+  
+  enum mode {
+      PICTURE   = 0x00,
+      AUTOPLAY  = 0x08,
+      AUDIOPLAY = 0x18
+  };
+
+  enum reg {
+    MODE      = 0x00,
+    FRAME     = 0x01,
+    AUTPLAY1  = 0x02,
+    AUTOPLAY2 = 0x03,
+    BLINK     = 0x05,
+    AUDIOSYNC = 0x06,
+    BREATH1   = 0x07,
+    BREATH2   = 0x08,
+    SHUTDOWN  = 0x0a,
+    GAIN      = 0x0b,
+    ADC       = 0x0c,
+    BANK      = 0xfd
+  };
+
+  void IS31FL3731::init() {
+    i2c_init(i2c, 100000);
+
+    gpio_set_function(sda, GPIO_FUNC_I2C); gpio_pull_up(sda);
+    gpio_set_function(scl, GPIO_FUNC_I2C); gpio_pull_up(scl);
+
+    i2c_reg_write_uint8(reg::SHUTDOWN, 0b00000001);
+
+    i2c_reg_write_uint8(reg::BANK, CONFIG_BANK);
+
+    i2c_reg_write_uint8(reg::MODE, mode::PICTURE);
+
+    i2c_reg_write_uint8(reg::AUDIOSYNC, 0);
+
+    clear();
+  }
+
+  void IS31FL3731::clear() {
+    for(auto i = 0u; i < sizeof(buf); i++) {
+        buf[i] = 0;
+    }
+  }
+
+  void IS31FL3731::i2c_reg_write_uint8(uint8_t reg, uint8_t value) {
+    uint8_t buffer[2] = {reg, value};
+    i2c_write_blocking(i2c, address, buffer, 2, false);
+  }
+
+  int16_t IS31FL3731::i2c_reg_read_int16(uint8_t reg) {
+    int16_t value;
+    i2c_write_blocking(i2c, address, &reg, 1, true);
+    i2c_read_blocking(i2c, address, (uint8_t *)&value, 2, false);
+    return value;
+  }
+
+  void IS31FL3731::enable(std::initializer_list<uint8_t> pattern, uint8_t frame) {
+    i2c_reg_write_uint8(reg::BANK, frame);
+    uint8_t enable_buf[19];
+    uint8_t *ptr = enable_buf;
+    enable_buf[0] = ENABLE_OFFSET;
+    ptr++;
+    for(auto byte : pattern) {
+        *ptr++ = byte;
+    }
+    i2c_write_blocking(i2c, address, enable_buf, sizeof(enable_buf), false);
+  }
+
+  void IS31FL3731::set(uint8_t index, uint8_t brightness) {
+    buf[index + 1] = GAMMA[brightness];
+  }
+
+  void IS31FL3731::update(uint8_t frame) {
+    i2c_reg_write_uint8(reg::BANK, frame);
+    buf[0] = COLOR_OFFSET;
+    i2c_write_blocking(i2c, address, buf, sizeof(buf), false);
+    i2c_reg_write_uint8(reg::BANK, CONFIG_BANK); // Switch back to config bank
+    i2c_reg_write_uint8(reg::FRAME, frame); // Set the desired frame as active
+  }
+
+}

drivers/is31fl3731/is31fl3731.hpp

@@ -0,0 +1,34 @@
+#pragma once
+
+#include "hardware/i2c.h"
+#include "hardware/gpio.h"
+
+namespace pimoroni {
+
+  class IS31FL3731 {
+    i2c_inst_t *i2c = i2c0;
+
+    // interface pins with our standard defaults where appropriate
+    int8_t address   = 0x74;
+    int8_t sda       = 4;
+    int8_t scl       = 5;
+
+    uint8_t buf[145];
+
+  public:
+    IS31FL3731() {}
+
+    IS31FL3731(i2c_inst_t *i2c, uint8_t sda, uint8_t scl) :
+      i2c(i2c), sda(sda), scl(scl) {}
+
+    void init();
+    void enable(std::initializer_list<uint8_t> pattern, uint8_t frame = 0);
+    void set(uint8_t index, uint8_t brightness);
+    void update(uint8_t frame = 0);
+    void clear();
+
+    void i2c_reg_write_uint8(uint8_t reg, uint8_t value);
+    int16_t i2c_reg_read_int16(uint8_t reg);
+  };
+
+}

examples/CMakeLists.txt

@@ -2,4 +2,5 @@ add_subdirectory(pico_display)
 add_subdirectory(pico_unicorn)
 add_subdirectory(pico_scroll)
 add_subdirectory(pico_explorer)
-add_subdirectory(pico_rgb_keypad)
+add_subdirectory(pico_rgb_keypad)
+add_subdirectory(breakout_rgbmatrix5x5)

examples/breakout_rgbmatrix5x5/CMakeLists.txt

@@ -0,0 +1,10 @@
+add_executable(
+  rgbmatrix5x5
+  demo.cpp
+)
+
+# Pull in pico libraries that we need
+target_link_libraries(rgbmatrix5x5 pico_stdlib hardware_spi hardware_pwm hardware_dma is31fl3731)
+
+# create map/bin/hex file etc.
+pico_add_extra_outputs(rgbmatrix5x5)

examples/breakout_rgbmatrix5x5/demo.cpp

@@ -0,0 +1,93 @@
+#include <string.h>
+#include <math.h>
+#include <vector>
+#include <cstdlib>
+
+#include "is31fl3731.hpp"
+
+using namespace pimoroni;
+
+IS31FL3731 rgbmatrix5x5;
+
+struct RGB {
+    uint8_t r;
+    uint8_t g;
+    uint8_t b;
+};
+
+// This wonderful lookup table maps the LEDs on RGB Matrix 5x5
+// from their 3x5x5 (remember, they're RGB) configuration to
+// their specific location in the 144 pixel buffer.
+const RGB lookup_table[] = {
+    (118, 69, 85),
+    (117, 68, 101),
+    (116, 84, 100),
+    (115, 83, 99),
+    (114, 82, 98),
+    (113, 81, 97),
+    (112, 80, 96),
+    (134, 21, 37),
+    (133, 20, 36),
+    (132, 19, 35),
+    (131, 18, 34),
+    (130, 17, 50),
+    (129, 33, 49),
+    (128, 32, 48),
+    (127, 47, 63),
+    (121, 41, 57),
+    (122, 25, 58),
+    (123, 26, 42),
+    (124, 27, 43),
+    (125, 28, 44),
+    (126, 29, 45),
+    (15, 95, 111),
+    (8, 89, 105),
+    (9, 90, 106),
+    (10, 91, 107),
+    (11, 92, 108),
+    (12, 76, 109),
+    (13, 77, 93),
+};
+
+RGB lookup_pixel(uint8_t index) {
+    return lookup_table[index];
+}
+
+void set_pixel(uint8_t x, uint8_t y, uint8_t r, uint8_t g, uint8_t b) {
+    if (x == 1 || x == 3) {
+        y = 4 - y;
+    }
+    uint8_t index = y + (x * 5);
+    RGB rgb = lookup_pixel(index);
+    rgbmatrix5x5.set(rgb.r, r);
+    rgbmatrix5x5.set(rgb.g, g);
+    rgbmatrix5x5.set(rgb.b, b);
+}
+
+int main() {
+    rgbmatrix5x5.init();
+    rgbmatrix5x5.enable({
+        0b00000000, 0b10111111,
+        0b00111110, 0b00111110,
+        0b00111111, 0b10111110,
+        0b00000111, 0b10000110,
+        0b00110000, 0b00110000,
+        0b00111111, 0b10111110,
+        0b00111111, 0b10111110,
+        0b01111111, 0b11111110,
+        0b01111111, 0b00000000
+    }, 0);
+
+    uint8_t col = 0;
+    while(1) {
+        for(auto x = 0u; x < 5; x++) {
+            for(auto y = 0u; y < 5; y++) {
+                set_pixel(x, y, col, col, col);
+            }
+        }
+        rgbmatrix5x5.update(0);
+        sleep_ms(10);
+        col++;
+    }
+    return 0;
+}