picotouch

# picotouch_grid test2_leds -- testing touch pads w/ touchio and LEDs
#
# 21 Feb 2023 - @todbot / Tod Kurt
#
#

import time
import board
import touchio
import neopixel
import rainbowio

touch_col_pins = (
    board.GP0, board.GP1, board.GP2, board.GP3, board.GP4, board.GP5,
    board.GP6, board.GP7, board.GP8, board.GP9
)
touch_row_pins = ( board.GP10, board.GP11, board.GP12, board.GP13, )
neopixel_pin = board.GP28

class TouchMatrix:
    def __init__(self, col_pins, row_pins, threshold_adjust_col=10, threshold_adjust_row=10):
        self.touch_cols = []
        self.touch_rows = []
        for pin in col_pins:
            touchin = touchio.TouchIn(pin)
            touchin.threshold += threshold_adjust_col
            self.touch_cols.append(touchin)
        for pin in row_pins:
            touchin = touchio.TouchIn(pin)
            touchin.threshold += threshold_adjust_row
            self.touch_rows.append(touchin)
        self.num_touch_cols = len(self.touch_cols)
        self.num_touch_rows = len(self.touch_rows)

    def update(self):
        col_pressed, row_pressed = None,None
        for i in range(self.num_touch_rows):
            if self.touch_rows[i].value:
                row_pressed = i
        for i in range(self.num_touch_cols):
            if self.touch_cols[i].value:
                col_pressed = i
        return (col_pressed,row_pressed)


print("picotouch_grid test2_leds hello")
print("\n----------")

touch_matrix = TouchMatrix( col_pins=touch_col_pins, row_pins=touch_row_pins )
leds = neopixel.NeoPixel(neopixel_pin, 40, brightness=0.1)
leds.fill(0x999999)

dim_by = 5

while True:
    leds[:] = [[max(i-dim_by,0) for i in l] for l in leds] # dim all by (dim_by,dim_by,dim_by)

    (col,row) = touch_matrix.update()
    if col is not None and row is not None:
        p = (row*10) + col # create number 0-39
        print("touch! col=",col,"row=",row, "led:",p)
        h = int(p * (256/40))  # map to 0-255
        leds[p] = rainbowio.colorwheel( time.monotonic() * 50 )