# noise_square_code.py -- playing with simplex noise in CircuitPython
# 9 Feb 2023 - @todbot / Tod Kurt
# https://github.com/todbot/CircuitPython_Noise
#
# 21 Mar 2025 - @pagong
# Uses Raspberry-Pi Pico with a 16x16 NeoPixel matrix

import time
import board
import random
import neopixel
import rainbowio

import matrix16

#####################

# for RPi-Pico with 16x16 NeoPixel-Matrix
BRIGHTNESS = 0.1   # A number between 0.0 and 1.0, where 0.0 is off, and 1.0 is max.
PIN = board.GP28   # This is the default pin on RPi-Pico with 16x16 NeoPixel matrix
matrix = matrix16.MatrixSetup(PIN, "hsquare", 1.0)
grid = matrix._grid

NUM_COLS  = matrix._width
NUM_CELLS = matrix._height

NUM_PIXELS = (NUM_COLS * NUM_CELLS)  # Update this to match the number of LEDs.
SPEED = 0.01       # Increase to slow down the animation. Decrease to speed it up.

#####################

# use Todbot's noise module from community bundle
import noise
noise_scale = 0.07   # noise_scale * max(width, height) should be smaller than 1.0
noise_incr = 0.01
noise_x = 0.0
noise_y = 0.0
xsign = -1.0
ysign = -1.0

#####################

# prepare rainbow palette
def make_palette(palette, bright):
    for hue in range(256):
        color = rainbowio.colorwheel(hue)
        r = int(bright * float((color >> 16) & 255))
        g = int(bright * float((color >> 8) & 255))
        b = int(bright * float(color & 255))
        col = (r, g, b)
        palette.append(col)

# change direction of movement
def change_direction():
    xs, ys = 0, 0
    while (abs(xs) + abs(ys) == 0):
        xs = random.randint(-1, 1)
        ys = random.randint(-1, 1)
    return float(xs), float(ys)

def do_frame():
    for i in range(NUM_COLS):           # for each pixel column
        nsx = noise_x + noise_scale*i
        pxl = grid[i]
        for j in range(NUM_CELLS):      # for each pixel row
            # get a noise value in 2D noise space
            n = noise.noise(nsx, noise_y + noise_scale*j)
            c = int((n+1.0) * 127.5)    # scale it from -1 - +1 -> 0 - 255
            pxl[j] = palette[c]         # convert hue to rainbow color

#####################

palette = []
make_palette(palette, BRIGHTNESS)

Debug = True

while True:
    t1 = time.monotonic_ns()
    do_frame()
    t2 = time.monotonic_ns()

    grid.show()
    t3 = time.monotonic_ns()

    if Debug:
        d1 = (t2 - t1) / 1000000.0
        print(f"Compute {d1} ms", end=" +\t")
        d2 = (t3 - t2) / 1000000.0
        print(f"Display {d2} ms", end=" =\t")
        print(f"Total {d1+d2} ms", end=" -->\t")
        print(f"{1000.0/(d1+d2)} fps")

    # move around in noise space
    noise_x += noise_incr * xsign
    noise_y += noise_incr * ysign
    if (random.randint(0, 99) == 8):
        xsign, ysign = change_direction()

    time.sleep(SPEED)