LEDMATRIXライブラリ: NeoPixel マトリックスライブラリ (MicroPython)

"""
LEDMATRIX: WS2812B (NeoPixel) Matrix Display driver
for MicroPython
Version 1.0

Copyright 2023 K.Kakizaki
LEDMATRIXライブラリ: NeoPixel マトリックスライブラリ (MicroPython)

"""

from machine import Pin
import framebuf
from micropython import const
from neopixel import NeoPixel
import time

def LEDColor(r, g, b) :
    return ((r & 0xF8) << 8) | ((g & 0xFC) << 3) | (b >> 3)

class LEDMATRIX(framebuf.FrameBuffer):

    RED = LEDColor(255, 0, 0)
    GREEN = LEDColor(0, 255, 0)
    BLUE = LEDColor(0, 0, 255)
    CYAN = LEDColor(0, 255, 255)
    YELLOW = LEDColor(255, 255, 0)
    PURPLE = LEDColor(255, 0, 255)
    WHITE = LEDColor(255, 255, 255)
    BLACK = LEDColor(0, 0, 0)

    SIMPLE=const(0x01)
    ZIGZAG=const(0x02)
    TOP=const(0x10)
    BOTTOM=const(0x20)
    RIGHT=const(0x40)
    LEFT=const(0x50)

    def __init__(self, pin, width, height, skip=0, link=SIMPLE):
        self._pin = Pin(pin, Pin.OUT)
        self._width = width
        self._height = height
        self._skip = skip
        self._link = link
        self._np = NeoPixel(self._pin, width * height + skip)
        self._brt = 10
        self.buffer = bytearray(self._height * self._width * 2)
        super().__init__(self.buffer, self._width, self._height, framebuf.RGB565)
        self.fill(0)
        for i in range(0, skip):
            self._np[i] = (0,0,0)
        self.show()

    def color(self, r, g, b):
        return ((r & 0xF8) << 8) | ((g & 0xFC) << 3) | (b >> 3)

    def setPixelColor(self, n, r, g, b):
        self._np[n] = (r, g, b)

    def setBrightness(self, brt):
        if (brt > 255):
            brt = 255
        if (brt < 0):
            brt = 0
        self._brt = brt
        self.show()

    def getBrightness(self):
        return self._brt

    def showScrollText(self, text, color, bgcolor=0, y=0, dir=0, ms=50):
        text = str(text)
        for x in range(self._width,-8*len(text)-1,-1):
            self.fill(bgcolor)
            self.text(text,x,y,color)
            self.show()
            time.sleep_ms(ms)

    def setScrollText(self, text, color, bgcolor=0, y=0, dir=0, ms=50):
        self._text = str(text)
        self._color = color
        self._bgcolor = bgcolor
        self._sty = y
        self._stx = self._width
        self._stxe = -8*len(self._text)-1
        self._dir = dir
        self._ms = ms

    def runScrollText(self):
        if self._stx > self._stxe:
            self.fill(self._bgcolor)
            self.text(self._text, self._stx, self._sty, self._color)
            self.show()
            self._stx = self._stx - 1
            return True
        else:
            return False

    def size(self):
        return (self._width, self._height)

    def show(self):
        for x in range(0, self._width):
            for y in range(0, self._height):
                idx = x + y * self._width
                v1 = self.buffer[idx*2]
                v0 = self.buffer[idx*2+1]
                vr = (v0 >> 3) << 3
                vg = (((v0 & 0x7) << 3) | (v1 >> 5)) << 2
                vb = (v1 & 0x1F) << 3
                rate = self._brt/255
                if self._link == SIMPLE:
                    lidx = x + y * self._width
                else:
                    lidx = x * self._height
                    if x % 2 == 0:
                        lidx += y
                    else:
                        lidx += (self._height - y - 1)
                self._np[lidx + self._skip] = (int(vr*rate), int(vg*rate), int(vb*rate))
        self._np.write()

from ledmatrix import LEDMATRIX
import time

# 8x8のカラーLEDマトリックスを使用
lmx = LEDMATRIX(2,8,8) # ESP8266-LEAF
# lmx = LEDMATRIX('RGB',8,8,3) # ESP32-S3-KEY-R2など開発基板上に３個のカラーLEDがあるもの

# lmx.setBrightness(50) # 暗いと思う場合にはコメントを外す

lmx.line(0,0,7,7,LEDMATRIX.BLUE)
lmx.line(0,7,7,0,LEDMATRIX.BLUE)
lmx.show()
time.sleep_ms(1000)
lmx.rect(0,0,8,8,LEDMATRIX.GREEN)
lmx.show()
time.sleep_ms(1000)
lmx.text('A',0,0,LEDMATRIX.RED)
lmx.show()

from ledmatrix import LEDMATRIX
import time

# 8x8のカラーLEDマトリックスを使用
# lmx = LEDMATRIX(2,8,8) # ESP8266-LEAF
lmx = LEDMATRIX('RGB',8,8,3) # ESP32-S3-KEY-R2

# lmx.setBrightness(50) # 暗いと思う場合にはコメントを外す
lmx.showScrollText('LEDMATRIX', LEDMATRIX.GREEN, ms=100)
lmx.showScrollText(12.34, LEDMATRIX.BLUE) # 数値も出力可

lmx.setScrollText('MicroFan', LEDMATRIX.RED)
while lmx.runScrollText():
# スクロール中にブロックされると困る処理が書ける
    time.sleep_ms(50)

from ledmatrix import LEDMATRIX
import time

color = [LEDMATRIX.RED, LEDMATRIX.GREEN, LEDMATRIX.BLUE, LEDMATRIX.CYAN, LEDMATRIX.YELLOW, LEDMATRIX.PURPLE, LEDMATRIX.WHITE]
brt = [0,1,2,3,5,10,20,30]

# lmx = LEDMATRIX(15,18,8) # ESP32-PIXEL
# lmx = LEDMATRIX(23,8,8) # LEDマトリックスパネル 8x8
# lmx = LEDMATRIX(23,32,8,link=LEDMATRIX.ZIGZAG) # LEDパネルマトリックスパネル 32x8
# lmx = LEDMATRIX(23,16,16,link=LEDMATRIX.ZIGZAG) # LEDパネルマトリックスパネル 16x16

if lmx.size()[0] <= 8:
    scrms = 80
elif lmx.size()[0] <= 20:
    scrms = 50
else:
    scrms = 20

lmx.setBrightness(0)
lmx.fill(LEDMATRIX.GREEN)
for n in brt:
    lmx.setBrightness(n)
    time.sleep_ms(200)
time.sleep_ms(1000)

lmx.fill(0)
lmx.setBrightness(10)
time.sleep_ms(1000)

if lmx.size()[1] == 18:
    str = 'ESP32-PIXEL LED LED MATRIX'
else:
    str = 'MicroFan, LED MATRIX'
    
lmx.showScrollText(str, LEDMATRIX.YELLOW, ms=scrms)
time.sleep_ms(1000)

for y in range(lmx.size()[1]+1,0,-1):
    lmx.fill_rect(0, 0, y*2, y, color[y%len(color)])
    lmx.show()
    time.sleep_ms(100)
time.sleep_ms(1000)

lmx.showScrollText('Temp,Hum,Press,Accel,Bright', LEDMATRIX.CYAN, ms=scrms)
time.sleep_ms(1000)

lmx.fill(0)
for y in range(lmx.size()[1],-1,-1):
    lmx.line(0, y, lmx.size()[0]-1, 0, color[y%len(color)])
    lmx.show()
    time.sleep_ms(100)
time.sleep_ms(1000)

lmx.showScrollText('MicroFan', LEDMATRIX.RED, ms=scrms)
time.sleep_ms(1000)

# ellipseはESP8266やESP32-C3では実装されていないようなので、このブロックは削除
for r in range(1,lmx.size()[0]//2):
    lmx.ellipse(lmx.size()[0]//2,lmx.size()[1]//2,r,r,LEDMATRIX.BLUE,True)
    lmx.show()
    time.sleep_ms(300)
time.sleep_ms(1000)

lmx.fill(LEDMATRIX.BLACK)
str = '> <'
x = lmx.size()[0]//2 - (8*len(str))//2
y = (lmx.size()[1] - 8) // 2
if lmx.size()[0] <= 8:
    lmx.text('#',0,0,LEDMATRIX.GREEN)
else:
    lmx.text(str,x,y,LEDMATRIX.CYAN)
    lmx.text('#',x+8,y,LEDMATRIX.GREEN)
for n in [0,1,1,1,2,2,3,5,7,10]:
    lmx.setBrightness(n)
    lmx.show()
    time.sleep_ms(300)