"""Complete Ruth theme, procedural instruments only, unchanged accepted models; no audio input."""
from pathlib import Path
import json, hashlib, shutil, subprocess
import numpy as np
from scipy.signal import resample_poly, butter, sosfilt
from scipy.io.wavfile import write, read
import guitar_model, bass_model
import sys
HERE=Path(__file__).resolve().parent
OUT=Path(sys.argv[1]).resolve() if len(sys.argv)>1 else HERE/'output'
OUT.mkdir(exist_ok=True)
score=json.loads((HERE/'score.json').read_text())
frames=round(score['duration']*44100);sr=44100
gains=json.loads((HERE/'gains.json').read_text());parts={};stats=[]
for part in ['guitar','bass']:
 mono=np.zeros(frames);notes=[e for e in score['events'] if e['part']==part]
 model=guitar_model if part=='guitar' else bass_model
 for i,e in enumerate(notes):
  duration=e['duration'];release=.16
  following=next((n for n in notes[i+1:] if n['string']==e['string']),None)
  if following and following['time']<e['time']+duration:
   duration=following['time']-e['time'];release=.085
  amp=.0003+.00095*e['velocity'] if part=='guitar' else .0004+.005*e['velocity']**2
  rt=.0015 if part=='guitar' else .0035-.002*e['velocity']
  x,energy,*_=model.simulate(e['pitch'],amp,part=='bass',duration+release,rt)
  energy=energy[energy>0]
  assert np.isfinite(x).all() and max(energy)<energy[0]*1.03
  x=resample_poly(x,1,2);t=np.arange(len(x))/sr
  x*=np.minimum(1,np.maximum(0,(duration+release-t)/release))**2
  at=round(e['time']*sr);n=min(len(x),frames-at);mono[at:at+n]+=x[:n]
  stats.append(dict(part=part,pitch=e['pitch'],bar=e['bar'],energyRatio=float(max(energy)/energy[0])))
 mono=sosfilt(butter(2,25,fs=sr,btype='highpass',output='sos'),mono)
 pan=[.82,.64] if part=='guitar' else [.64,.82]
 stereo=mono[:,None]*np.array(pan)*gains[part]
 stereo*=np.minimum(1,np.maximum(0,(score['duration']-np.arange(frames)/sr)/1.5))[:,None]
 parts[part]=stereo
 print(part,'done peak',float(np.max(abs(stereo))),flush=True)
parts['mix']=parts['guitar']+parts['bass']
assert np.max(abs(parts['mix']))<.99, 'Clipping: review voicing balance rather than change accepted opening gain'
for part,x in parts.items():
 write(OUT/(part+'.wav'),sr,np.round(x*32767).astype(np.int16))
 subprocess.run(['ffmpeg','-hide_banner','-loglevel','error','-y','-i',str(OUT/(part+'.wav')),'-codec:a','libmp3lame','-b:a','192k',str(OUT/(part+'.mp3'))],check=True)
for name in ['guitar_model.py','bass_model.py','gains.json','render.py']:
 shutil.copyfile(HERE/name,OUT/name)
(OUT/'score.json').write_text(json.dumps(score,indent=2))
report=dict(duration=score['duration'],peak=float(np.max(abs(parts['mix']))),
 notes=stats,
 models={n:hashlib.sha256((HERE/n).read_bytes()).hexdigest() for n in ['guitar_model.py','bass_model.py']},
 limitations='Per-note string/body voices; simplified bass accompaniment, not an exact performance transcription.')
(OUT/'report.json').write_text(json.dumps(report,indent=2))
print(json.dumps({k:v for k,v in report.items() if k!='notes'},indent=2))