"""2024-12-22
Vendo o Som 3 - Somebody to Shove (Soul Asylum)
Exercício de visualização de som com espirais
png
Sketch,py5,CreativeCoding
"""

import math

import numpy as np
import py5
import sounddevice as sd

from utils import helpers

sketch = helpers.info_for_sketch(__file__, __doc__)

MARGEM_X = 200
MARGEM_Y = 200


ESPIRAIS = 4
ALL_DATA = []
DEVICE = 0
GAIN = 30
BLOCK_DURATION = 10

CAPTURAS = 720


def capture_sound(total_points: int):
    data = []
    samplerate = sd.query_devices(DEVICE, "input")["default_samplerate"]
    low, high = [100, 2000]
    delta_f = (high - low) / (80 - 1)
    fftsize = math.ceil(samplerate / delta_f)

    def callback(indata, frames, time, status):
        if any(indata):
            magnitude = np.abs(np.fft.rfft(indata[:, 0], n=fftsize))
            magnitude *= 10 / fftsize
            data.append(magnitude)

    with sd.InputStream(
        device=DEVICE,
        channels=1,
        callback=callback,
        blocksize=int(samplerate * BLOCK_DURATION / 1000),
        samplerate=samplerate,
    ):
        while len(data) < total_points:
            print(f"{len(data)}/{total_points} ({samplerate})")
    return data


def espiral(
    xb: float,
    yb: float,
    offset: float,
    data: np.array,
    frame_max: float,
    frame_min: float,
):
    pontos = []
    total_pontos = len(data)
    raio_max = py5.width * 2
    for idx in range(total_pontos):
        direcao = 1
        angulo = py5.radians(idx * 5 + offset)
        r = idx * raio_max / total_pontos
        pb = data[idx]
        diff = py5.remap(pb, frame_min, frame_max, 1, 2) * direcao
        x0 = xb + r * np.cos(angulo)
        y0 = yb + r * np.sin(angulo)
        x = x0 * diff
        y = y0 * diff
        pontos.append((x0, y0, x, y, diff))
    return pontos


def setup():
    global ALL_DATA
    py5.size(helpers.LARGURA, helpers.ALTURA, py5.P3D)
    py5.color_mode(py5.HSB, 360, 100, 100)
    py5.ellipse_mode(py5.CENTER)
    ALL_DATA = capture_sound(CAPTURAS)


def draw():
    py5.background(0)
    numero = py5.frame_count
    offset_b = 360 / ESPIRAIS
    offset = 0
    with py5.push_style():
        with py5.push_matrix():
            py5.translate(py5.width // 2, py5.height // 2, -30)
            for i in range(0, ESPIRAIS):
                idc = ((numero * ESPIRAIS) + i) % CAPTURAS
                data = ALL_DATA[idc]
                frame_min = np.min(data)
                frame_max = np.max(data)
                pontos = espiral(0, 0, offset, data, frame_max, frame_min)
                for idx, (x0, y0, x, y, diff) in enumerate(pontos):
                    h = (diff * 180) % 360
                    py5.stroke(py5.color(h, 90, 90))
                    weight = (idx // 100) + 2
                    py5.stroke_weight(weight)
                    py5.line(x0, y0, x, y)
                offset += offset_b

    helpers.write_legend(sketch=sketch, frame="#000")


def key_pressed():
    key = py5.key
    if key == " ":
        save_and_close()


def save_and_close():
    py5.no_loop()
    helpers.save_sketch_image(sketch)
    py5.exit_sketch()


if __name__ == "__main__":
    py5.run_sketch()