"""Gera os 7 efeitos sonoros do Xeonjia que o autor nao credita (collision, damage, dialog, explosion, money,
punch, snowball); os tres que o README dele credita ao Yubatake (gameover, item, win) ficam como estao.

Sintese propria, deterministica (PRNG com semente fixa), dedicada ao dominio publico (CC0). Escreve WAV estereo de
16 bits a 44,1 kHz e converte em .oga (Ogg Vorbis) com o wav2ogg do projeto (Builds/pygame-music/wav2ogg.c, numero de
serie fixo: mesma entrada, mesmos bytes). Mesmos nomes de arquivo do jogo.

  python3 -I sounds.py <pasta-de-saida> <caminho-do-wav2ogg>
"""
import math, os, random, struct, subprocess, sys

SR = 44100


def buf(seconds):
    return [0.0] * int(SR * seconds)


def add(dst, src, at=0.0, gain=1.0):
    o = int(at * SR)
    for i, v in enumerate(src):
        if o + i < len(dst):
            dst[o + i] += v * gain


def env(t, attack, decay):
    return min(1.0, t / attack) * math.exp(-t / decay) if attack else math.exp(-t / decay)


def tone(freq, dur, decay, attack=0.004, shape='sine', f_end=None):
    out, ph = [], 0.0
    n = int(SR * dur)
    for i in range(n):
        t = i / SR
        f = freq if f_end is None else freq + (f_end - freq) * (t / dur)
        ph += 2 * math.pi * f / SR
        s = math.sin(ph)
        if shape == 'tri':
            s = 2 / math.pi * math.asin(s)
        elif shape == 'soft':  # seno com um pouco de harmonicos
            s = s + 0.25 * math.sin(2 * ph) + 0.1 * math.sin(3 * ph)
        out.append(s * env(t, attack, decay))
    return out


def noise(dur, decay, rng, lp=1.0, attack=0.0005):
    out, y = [], 0.0
    for i in range(int(SR * dur)):
        t = i / SR
        y += lp * (rng.uniform(-1, 1) - y)
        out.append(y * env(t, attack, decay))
    return out


def sweep_noise(dur, rng, f0, f1, decay):
    out, y = [], 0.0
    n = int(SR * dur)
    for i in range(n):
        t = i / SR
        a = 1 - math.exp(-2 * math.pi * (f0 + (f1 - f0) * t / dur) / SR)
        y += a * (rng.uniform(-1, 1) - y)
        out.append(y * env(t, 0.003, decay))
    return out


def clap(rng, gain=1.0):
    n = noise(0.09, 0.018, rng, lp=0.55, attack=0.0008)
    # passa-altas simples para tirar o grave
    out, prev = [], 0.0
    for v in n:
        out.append((v - prev * 0.6) * gain)
        prev = v
    return out


def finish(mono, path, pan=None):
    peak = max(1e-9, max(abs(v) for v in mono))
    scale = 0.85 / peak
    n = len(mono)
    fade = int(0.005 * SR)
    data = bytearray()
    for i, v in enumerate(mono):
        if i > n - fade:
            v *= (n - i) / fade
        s = max(-32767, min(32767, int(v * scale * 32767)))
        data += struct.pack('<hh', s, s)
    with open(path, 'wb') as f:
        f.write(b'RIFF' + struct.pack('<I', 36 + len(data)) + b'WAVEfmt ' + struct.pack('<IHHIIHH', 16, 1, 2, SR, SR * 4, 4, 16)
                + b'data' + struct.pack('<I', len(data)) + bytes(data))


def make_all():
    S = {}
    r = random.Random(20261105)

    # Batida em obstaculo: "tump" grave e curto.
    b = buf(0.3)
    add(b, noise(0.015, 0.004, r, lp=0.7), 0, 0.6)
    add(b, tone(170, 0.2, 0.045, f_end=75), 0, 1.0)
    S['collision'] = b

    # Dano: queda de tom com chiado.
    b = buf(0.5)
    add(b, tone(520, 0.3, 0.09, f_end=130, shape='tri'), 0, 0.9)
    add(b, noise(0.25, 0.07, r, lp=0.35), 0, 0.5)
    S['damage'] = b

    # Fala de dialogo: dois bipes curtos.
    b = buf(0.25)
    add(b, tone(620, 0.07, 0.022, shape='tri'), 0.0, 0.8)
    add(b, tone(820, 0.09, 0.026, shape='tri'), 0.07, 0.8)
    S['dialog'] = b

    # Explosao: estouro que escurece e um grave.
    b = buf(1.2)
    add(b, sweep_noise(1.1, r, 4200, 180, 0.28), 0, 1.0)
    add(b, tone(95, 0.9, 0.22, f_end=40), 0, 0.9)
    S['explosion'] = b

    # Moeda: dois tons subindo.
    b = buf(0.6)
    add(b, tone(988, 0.12, 0.05, shape='soft'), 0.0, 0.7)
    add(b, tone(1319, 0.5, 0.11, shape='soft'), 0.08, 0.7)
    S['money'] = b

    # Soco: golpe seco.
    b = buf(0.3)
    add(b, noise(0.03, 0.007, r, lp=0.6), 0, 0.9)
    add(b, tone(150, 0.2, 0.04, f_end=65), 0, 1.0)
    add(b, tone(900, 0.04, 0.01), 0, 0.15)
    S['punch'] = b

    # Bola de neve: assobio de ar e um "puf".
    b = buf(0.6)
    add(b, sweep_noise(0.35, r, 600, 3400, 0.16), 0, 0.7)
    add(b, noise(0.12, 0.03, r, lp=0.25), 0.32, 0.8)
    S['snowball'] = b
    return S


if __name__ == '__main__':
    out = sys.argv[1]
    enc = sys.argv[2]
    os.makedirs(out, exist_ok=True)
    for name, mono in make_all().items():
        wav = os.path.join(out, name + '.wav')
        finish(mono, wav)
        subprocess.run([enc, wav, os.path.join(out, name + '.oga'), '0.4'], check=True)
        os.remove(wav)