import math, wave, struct
SR=48000
HARM=(1.00, 0.85, 0.10, 0.02)

def note(f, dur, tau, amp=1.0):
    n=int(SR*dur); o=[]
    for i in range(n):
        t=i/SR
        s=sum(a*math.sin(2*math.pi*f*(k+1)*t) for k,a in enumerate(HARM))
        o.append(s*amp*min(1.0,t/0.004)*math.exp(-t/tau))
    return o

def thump(f, dur, amp):
    n=int(SR*dur); o=[]; ph=0.0
    for i in range(n):
        t=i/SR
        ph += 2*math.pi*(f*(1+1.2*math.exp(-t/0.02)))/SR
        o.append(math.sin(ph)*amp*math.exp(-t/(dur*0.30)))
    return o

def mix(layers,total):
    b=[0.0]*total
    for off,s in layers:
        for i,v in enumerate(s):
            if off+i<total: b[off+i]+=v
    return b

def peak_rms(x, win=int(SR*0.01)):
    return max(math.sqrt(sum(v*v for v in x[i:i+win])/win) for i in range(0,len(x)-win,win))

def write(name, buf, target):
    g=target/max(1e-9,peak_rms(buf))
    pk=max(abs(s*g) for s in buf)
    if pk>0.97: g*=0.97/pk
    with wave.open(name,'w') as w:
        w.setnchannels(1); w.setsampwidth(2); w.setframerate(SR)
        w.writeframes(b''.join(struct.pack('<h',int(max(-1,min(1,s*g))*32767)) for s in buf))
    print(f"{name}: pico RMS {peak_rms([s*g for s in buf]):.3f}, {len(buf)/SR:.2f}s")

C3,C4,E4,G4,C5 = 130.81,261.63,329.63,392.00,523.25
step=int(SR*0.033)

# Espelho do inicio: o arpejo DESCE e resolve na raiz grave, que entra por
# ultimo. Sem o sopro de ar — brilho sugere chegada, nao encerramento.
# Mais curto e mais baixo: encerrar merece menos destaque que comecar.
write('stream_ended.wav', mix([
    (0,      note(C5, 0.42, 0.15, 0.70)),
    (step,   note(G4, 0.44, 0.16)),
    (2*step, note(E4, 0.46, 0.17)),
    (3*step, note(C4, 0.48, 0.19)),
    (3*step, note(C3, 0.50, 0.21, 0.80)),
    (3*step, thump(85, 0.13, 0.40)),
], int(SR*0.55)), 0.34)