m60.py

import analogio
import microcontroller
from .is32fl3733 import IS31FL3733

try:
    # usebuilt-in matrix if it is available
    from matrix import Matrix
except ImportError:
    from ..matrix import Matrix
    from board import R1, R2, R3, R4, R5, R6, R7, R8, C1, C2, C3, C4, C5, C6, C7, C8

    Matrix.ROWS = (R1, R2, R3, R4, R5, R6, R7, R8)
    Matrix.COLS = (C1, C2, C3, C4, C5, C6, C7, C8)
    Matrix.ROW2COL = False

# fmt: off
# ESC   1   2   3   4   5   6   7   8   9   0   -   =  BACKSPACE
# TAB   Q   W   E   R   T   Y   U   I   O   P   [   ]   |
# CAPS  A   S   D   F   G   H   J   K   L   ;   "      ENTER
#LSHIFT Z   X   C   V   B   N   M   ,   .   /         RSHIFT
# LCTRL LGUI LALT          SPACE         RALT MENU  L1 RCTRL
COORDS = (
    0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13,
    27,26, 25, 24, 23, 22, 21, 20, 19, 18, 17, 16, 15, 14,
    28,29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39,     40,
    52,51, 50, 49, 48, 47, 46, 45, 44, 43, 42,         41,
    53,  54, 55,             56,           57, 58, 59, 60
)


BATTERY_LIMIT = 3100  # Cutoff voltage [mV].
BATTERY_FULLLIMIT = 4190  # Full charge definition [mV].
BATTERY_DELTA = 10  # mV between each element in the SoC vector.

BATTERY_VOLTAGE = (
    0,  0,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  1,  2,
    2,  2,  2,  2,  2,  2,  2,  2,  3,  3,  3,  3,  3,  3,  3,  4,  4,  4,  4,  4,
    4,  5,  5,  5,  6,  6,  7,  7,  8,  8,  9,  9, 10, 11, 12, 13, 13, 14, 15, 16,
    18, 19, 22, 25, 28, 32, 36, 40, 44, 47, 51, 53, 56, 58, 60, 62, 64, 66, 67, 69,
    71, 72, 74, 76, 77, 79, 81, 82, 84, 85, 85, 86, 86, 86, 87, 88, 88, 89, 90, 91,
    91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 100
)

battery_in = analogio.AnalogIn(microcontroller.pin.P0_02)


def battery_level():
    # (3300 * 2 * battery.value) >> 16
    voltage = (3300 * battery_in.value) >> 15
    i = (voltage - BATTERY_LIMIT) // BATTERY_DELTA
    if i >= len(BATTERY_VOLTAGE):
        i = len(BATTERY_VOLTAGE) - 1
    elif i < 0:
        i = 0
    return BATTERY_VOLTAGE[i]


leds_x = bytearray((
    0, 16, 32, 48, 64, 80, 96, 112, 128, 144, 160, 176, 192, 216,
    220,200, 184, 168, 152, 136, 120, 104, 88, 72, 56, 40, 24, 4,
    6, 27, 43, 59, 75, 91, 107, 123, 139, 155, 171, 187, 214,
    210, 180, 164, 148, 132, 116, 100, 84, 68, 52, 36, 10,
    2, 22, 42, 102, 162, 182, 202, 222,
    123, 82
))

leds_y = bytearray((
    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
    16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16,
    32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32,
    48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48, 48,
    64, 64, 64, 64, 64, 64, 64, 64,
    64, 64
))

angle = bytearray((
    180, 178, 176, 173, 168, 160, 146, 128, 109, 96, 87, 82, 79, 76,
    69, 71, 72, 75, 79, 87, 109, 146, 168, 176, 180, 183, 184, 186,
    170, 170, 170, 170, 170, 170, 170, 64, 64, 64, 64, 64, 64,
    57, 54, 51, 46, 36, 9, 229, 213, 206, 202, 200, 198,
    203, 205, 209, 243, 40, 46, 50, 52,
    13, 225
))

distance = bytearray((
    116, 101, 86, 71, 57, 45, 35, 32, 35, 45, 57, 71, 86, 108, 109, 89, 73,
    58, 43, 28, 17, 17, 28, 43, 58, 73, 89, 109, 106, 85, 69, 53, 37, 21, 5,
    11, 27, 43, 59, 75, 102, 99, 69, 54, 39, 25, 16, 20, 32, 46, 62, 77, 103,
    114, 95, 76, 33, 59, 76, 95, 114,
    33, 43
))
# fmt: on


def hsv_to_rgb(h, s, v):
    i = (h * 6) >> 8
    f = (h * 6) & 0xFF

    p = (v * (256 - s)) >> 8
    q = (v * (65536 - s * f)) >> 16
    t = (v * (65536 - s * (256 - f))) >> 16

    if i == 0:
        return (v, t, p)
    if i == 1:
        return (q, v, p)
    if i == 2:
        return (p, v, t)
    if i == 3:
        return (p, q, v)
    if i == 4:
        return (t, p, v)
    return (v, p, q)


def wheel(h):
    i = (h * 3) >> 8
    a = (h * 3) & 0xFF
    b = 255 - a

    if i == 0:
        return (b, a, 0)
    if i == 1:
        return (0, b, a)

    return (a, 0, b)


def wheel2(h, v):
    i = (h * 3) >> 8
    a = (h * 3) & 0xFF
    b = 255 - a

    a = a * v // 255
    b = b * v // 255

    if i == 0:
        return (b, a, 0)
    if i == 1:
        return (0, b, a)

    return (a, 0, b)


class Backlight:
    def __init__(self):
        self.dev = IS31FL3733()
        self._hid_leds = 0
        self._bt_led = None
        self.pixel = self.dev.pixel

        self.enabled = True
        self.hsv = [0, 255, 255]
        self.keys = {}
        self.n = 0

        self.modes = (
            self.off,
            self.mono,
            self.gradient,
            self.spectrum,
            self.spectrum_x,
            self.spectrum_y,
            self.elapse,
            self.broadcast,
            self.blackhole,
            self.pinwheel,
            self.beacon,
            self.beacon2,
            self.flow
        )
        self.set_mode(6)
        self.enabled = False

    @property
    def hue(self):
        return self.hsv[0]

    @hue.setter
    def hue(self, h):
        self.hsv[0] = h & 0xFF
        self.refresh()

    @property
    def sat(self):
        return self.hsv[1]

    @sat.setter
    def sat(self, s):
        self.hsv[1] = 0 if s < 0 else (255 if s > 255 else s)
        self.refresh()

    @property
    def val(self):
        return self.dev.brightness

    @val.setter
    def val(self, v):
        self.set_brightness(v)

    def set_brightness(self, v):
        if v <= 0:
            self.enabled = False
            self.off()
        else:
            self.enabled = True
            self.dev.brightness = v if v < 0xFF else 0xFF
            if not self.dynamic:
                self.mode_function()

    def on(self, r=0xFF, g=0xFF, b=0xFF):
        for i in range(63):
            self.pixel(i, r, g, b)
        self.update()

    def off(self):
        self.dev.clear()
        self.update()

    def toggle(self):
        self.enabled = not self.enabled
        if self.enabled:
            self.set_mode(self.mode)
        else:
            self.off()
# ESC(0)    1(1)   2(2)   3(3)   4(4)   5(5)   6(6)   7(7)   8(8)   9(9)   0(10)  -(11)  =(12)  BACKSPACE(13)
# TAB(27)   Q(26)  W(25)  E(24)  R(23)  T(22)  Y(21)  U(20)  I(19)  O(18)  P(17)  [(16)  ](15)   \(14)
# CAPS(28)  A(29)  S(30)  D(31)  F(32)  G(33)  H(34)  J(35)  K(36)  L(37)  ;(38)  "(39)      ENTER(40)
#LSHIFT(52) Z(51)  X(50)  C(49)  V(48)  B(47)  N(46)  M(45)  ,(44)  .(43)  /(42)            RSHIFT(41)
# LCTRL(53)  LGUI(54)  LALT(55)               SPACE(56)          RALT(57)  MENU(58)  Fn(59)  RCTRL(60)
#No.61 and No.62 are under the space key. No.63 is at the back of keyboard.


    def flow(self):
        stepby = 10
        #print(self.n)
        i = self.n
        j = int(i/stepby)
        self.pixel(j, 0xff, 0xff, 0xff)
        self.pixel(j-1 if j-1>=0 else 60, 0, 0, 0)
        if j == 56:
            self.pixel(61, 0xff, 0xff, 0xff)
            self.pixel(62, 0xff, 0xff, 0xff)
        else:
            self.pixel(61, 0, 0, 0)
            self.pixel(62, 0, 0, 0)
                

        self.update()
        #self.n = (i + 1) & (60*stepby+stepby-1)
        self.n = i + 1
        if self.n > (60*stepby+stepby-1):
            self.n = 0

        return True

    def mono(self):
        self.on(*hsv_to_rgb(*self.hsv))

    def gradient(self):
        h0, s0, v0 = self.hsv
        for i in range(63):
            h = (leds_y[i] + h0) & 0xFF
            self.pixel(i, *hsv_to_rgb(h, s0, v0))
        self.update()

    def spectrum(self, offset=0):
        self.n = (self.n + 1) & 0xFF
        r, g, b = wheel(self.n)
        for i in range(63):
            self.pixel(i, r, g, b)
        self.update()
        return True

    def spectrum_x(self):
        n = self.n
        for i in range(63):
            h = (leds_x[i] + n) & 0xFF
            self.pixel(i, *wheel(h))
        self.update()
        self.n = (n + 1) & 0xFF
        return True

    def spectrum_y(self):
        n = self.n
        for i in range(63):
            h = (leds_y[i] + n) & 0xFF
            self.pixel(i, *wheel(h))
        self.update()
        self.n = (n + 1) & 0xFF
        return True

    def broadcast(self):
        n = self.n
        for i in range(63):
            self.pixel(i, *wheel((distance[i] - n) & 0xFF))
        self.update()
        self.n = (n + 2) & 0xFF
        return True

    def blackhole(self):
        n = self.n
        for i in range(63):
            self.pixel(i, *wheel2((distance[i] + n) & 0xFF, distance[i] * 2 - 10))
        self.update()
        self.n = (n + 2) & 0xFF
        return True

    def pinwheel(self):
        n = self.n
        for i in range(63):
            self.pixel(i, *wheel((angle[i] + n) & 0xFF))
        self.update()
        self.n = (n + 2) & 0xFF
        return True

    def beacon(self):
        n = self.n
        for i in range(63):
            offset = (angle[i] + n) & 0xFF
            if offset < 64:
                offset <<= 2
            else:
                offset = 0
            self.pixel(i, *wheel(offset))
        self.update()
        self.n = (n + 2) & 0xFF
        return True

    def beacon2(self):
        n = self.n
        for i in range(63):
            offset = (angle[i] + n) & 0xFF
            if offset < 64:
                offset <<= 2
            elif 128 < offset and offset < 192:
                offset = (offset - 128) << 2
            else:
                offset = 0
            self.pixel(i, *wheel(offset))
        self.update()
        self.n = (n + 2) & 0xFF
        return True

    def handle_key(self, key, pressed):
        if pressed and self.enabled and self.mode == 6:
            self.keys[key] = 255

    def elapse(self):
        if 0 == len(self.keys):
            return False
        for i in self.keys.keys():
            t = self.keys[i]
            self.pixel(i, *wheel2(255 - t, t))
            if t >= 4:
                self.keys[i] = t - 4
            elif t > 0:
                self.keys[i] = t - 1
            else:
                self.keys.pop(i)
        self.update()
        return True

    def set_hid_leds(self, v):
        if self._hid_leds != v:
            self._hid_leds = v
            g = 128 if (self._hid_leds & 2) else 0
            self.dev.update_pixel(28, 0, g, 0)
            self.refresh()

    def set_bt_led(self, v):
        if v == 0:
            v = 10
        if v is not None:
            self.dev.set_mode(v, 2)
        if self._bt_led is not None:
            self.dev.set_mode(self._bt_led, 0)
        self._bt_led = v

    def update(self):
        if self._hid_leds & 2:
            self.pixel(28, 0, 0x80, 0)
        self.dev.update()

    def check(self):
        if self.enabled and self.dynamic:
            return self.mode_function()
        return False

    def refresh(self):
        if self.enabled and not self.dynamic:
            self.mode_function()

    def next(self):
        self.set_mode(self.mode + 1)

    def set_mode(self, mode):
        self.enabled = True
        self.dev.clear()
        self.mode = mode if mode < len(self.modes) else 0
        self.mode_function = self.modes[self.mode]
        if self.mode == 6:
            self.keys.clear()
        if self.mode >= 3:
            self.dynamic = True
        else:
            self.dynamic = False
            self.mode_function()