#!/usr/bin/env python3
# Gera os tres efeitos sonoros que trocam, no Asteroid's Revenge, arquivos sem origem declarada no repositorio
# (o README e o credits.md nao os citam e o historico do git so diz "add menu sound" etc.):
#   art-and-sound/menu_click_sound.ogg       clique dos botoes do menu
#   art-and-sound/missile_target_locked.ogg  aviso de missil travado no alvo
#   art-and-sound/transition_sound2.ogg      som da abertura do menu
# Os sons daqui sao nossos, sintetizados do zero (seno e ruido com semente fixa), e ficam em CC0 (dominio
# publico). Saida: .wav estereo 16 bits 44100 Hz ao lado deste script; o wav2ogg (Builds/pygame-music/wav2ogg.c,
# serie fixa) os passa para Ogg Vorbis, e os .ogg ficam versionados ao lado (o build-asteroids-revenge.sh so copia).
#
#   python3 -I gen-sounds.py && for n in menu_click_sound missile_target_locked transition_sound2; do
#     wav2ogg $n.wav $n.ogg 0.4; done
import math, os, random, struct, wave

RATE = 44100
HERE = os.path.dirname(os.path.abspath(__file__))


def write(name, samples):
    peak = max(abs(s) for s in samples) or 1.0
    with wave.open(os.path.join(HERE, name), 'wb') as w:
        w.setnchannels(2); w.setsampwidth(2); w.setframerate(RATE)
        frames = bytearray()
        for s in samples:
            v = int(max(-1.0, min(1.0, s / peak * 0.7)) * 32767)
            frames += struct.pack('<hh', v, v)
        w.writeframes(bytes(frames))


def env(i, n, attack=0.004, power=3.0):
    return min(1.0, i / RATE / attack) * (1.0 - i / n) ** power


# Clique: um "tic" curto de onda quadrada suavizada que cai de 1400 para 900 Hz.
def click():
    n = int(RATE * 0.09)
    out, ph = [], 0.0
    for i in range(n):
        f = 1400 - 500 * i / n
        ph += 2 * math.pi * f / RATE
        out.append(math.tanh(3 * math.sin(ph)) * env(i, n, 0.002, 2.0))
    return out


# Travado: dois bipes agudos e rapidos (2 kHz), como um radar confirmando o alvo.
def locked():
    n = int(RATE * 0.40)
    out = [0.0] * n
    for start in (0.0, 0.16):
        s0, m = int(start * RATE), int(RATE * 0.12)
        for i in range(m):
            t = i / RATE
            v = math.sin(2 * math.pi * 2000 * t) + 0.3 * math.sin(2 * math.pi * 4000 * t)
            out[s0 + i] += v * env(i, m, 0.003, 1.2)
    return out


# Transicao: um sopro de ruido filtrado que sobe e desce, com um tom grave que desliza por baixo.
def whoosh():
    n = int(RATE * 1.6)
    rnd = random.Random(20261111)
    out, lp, ph = [], 0.0, 0.0
    for i in range(n):
        x = i / n
        a = math.sin(math.pi * x) ** 2
        cut = 0.02 + 0.25 * math.sin(math.pi * x)
        lp += cut * (rnd.uniform(-1, 1) - lp)
        ph += 2 * math.pi * (90 + 160 * x) / RATE
        out.append(a * (lp * 2.5 + 0.35 * math.sin(ph)))
    return out


write('menu_click_sound.wav', click())
write('missile_target_locked.wav', locked())
write('transition_sound2.wav', whoosh())